dgasmith / opt_einsum

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
https://dgasmith.github.io/opt_einsum/
MIT License
822 stars 67 forks source link

add jaxlib alias #228

Closed jcmgray closed 1 month ago

jcmgray commented 2 months ago

Description

Alias jaxlib to jax.numpy namespace.

Status

codecov[bot] commented 1 month ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 89.44%. Comparing base (1a984b7) to head (82eeae8). Report is 2 commits behind head on master.

Additional details and impacted files