Re:Julia Summer of Einsum
by Blammar on Wednesday June 05, 2019 @06:26PM
(#20109755)
Hopefully, people are also looking at https://github.com/dgasmith/opt_einsum for ideas of what could go into a julia implementation.
[ Parent
| on HN ]
Re:Julia Summer of Einsum
by under-Peter on Thursday June 06, 2019 @03:22AM
(#20112303)
This is pretty much what I'm going for :). At this point in the project we have an einsum that is slow but works for all cases and supports AD. Next steps are splitting up contractions into pairwise contractions, finding optimal orders, evaluating the contractions with the best algorithm (so we might dispatch matrix multiplication to BLAS) and supporting different backends (probably just GPU but if more general tensor types could be supported that'd be great).
Re:Julia Summer of Einsum
by eigenspace on Wednesday June 05, 2019 @06:46PM
(#20109889)
That's a good resource. I beleive a lot of these optimizations are in TensorOperations.jl but given the head-start, there's probably still some good ideas on the table in that library which people should be learning from.