Add JAX support - Githubissues

parrt / tensor-sensor

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

https://github.com/parrt/tensor-sensor

MIT License

771 stars 39 forks source link

Add JAX support #20

Closed parrt closed 3 years ago