4 ms·
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 s
by under-Peter 7y ago
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).