3 ms·
Hopefully, people are also looking at https://github.com/dgasmith/opt_einsum https://github.com/dgasmith/opt_einsum for ideas of what could go into a julia impl
by Blammar 7y ago
Hopefully, people are also looking at https://github.com/dgasmith/opt_einsum https://github.com/dgasmith/opt_einsum for ideas of what could go into a julia implementation.
- eigenspace 7y agoThat'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.
- under-Peter 7y agoThis 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).