4 ms·
I really like einops. This works for numpy, pytorch and keras/tensorflow and has easy named transpose, repeat, and eimsum operations.
by SimplyUnknown 1y ago
I really like einops. This works for numpy, pytorch and keras/tensorflow and has easy named transpose, repeat, and eimsum operations.
- rsfern 1y agoSame - I’ve been using einops and jaxtyping together pretty extensively recently and it helps a lot for reading/writing multidimensional array code. Also array_api_compat, the API coverage isn’t perfect but it’s pretty satisfying to write code that works for both PyTorch and numpy arrays https://docs.kidger.site/jaxtyping/ https://docs.kidger.site/jaxtyping/ https://data-apis.org/array-api-compat/ https://data-apis.org/array-api-compat/