dgasmith/opt_einsum: v2.1.0
Creators
- 1. The Molecular Sciences Software Insititute
- 2. University of Wisconsin
- 3. D.E. Shaw Research
- 4. @spacetelescope
Description
opt_einsum continues to improve its support for additional backends beyond NumPy with PyTorch
We have also published the opt_einsum package in the Journal of Open Source Software. If you use this package in your work, please consider citing us!
New features:
PyTorch backend support Tensorflow eager-mode execution backend support Enhancements:
Intermediate tensordot-like expressions are now ordered to avoid transposes. CI now uses conda backend to better support GPU and tensor libraries. Now accepts arbitrary unicode indices rather than a subset. New auto path option which switches between optimal and greedy at four tensors. Bug fixes:
Fixed issue where broadcast indices were incorrectly locked out of tensordot-like evaluations even after their dimension was broadcast.
Files
dgasmith/opt_einsum-v2.1.0.zip
Files
(182.5 kB)
| Name | Size | Download all |
|---|---|---|
|
md5:9ed50d0872458a50e862c5cfcabc9df4
|
182.5 kB | Preview Download |
Additional details
Related works
- Is supplement to
- https://github.com/dgasmith/opt_einsum/tree/v2.1.0 (URL)