Add support for JAX Arrays#1

Open

anshradh wants to merge 5 commits intoKiv:mainfrom

Commits on Nov 20, 2022

Add JAX backend and support jnp arrays
Ansh Radhakrishnan
committed
Handle all JAX Arrays
Ansh Radhakrishnan
committed
Fix autoformatting
Ansh Radhakrishnan
committed
Add jax to reqs
Ansh Radhakrishnan
committed

Commits on Nov 21, 2022

Modify README
Ansh Radhakrishnan
committed