Updated version compatible with jax v0.4.xx
Pre-release
Pre-release
Merge pull request #7 from DifferentiableUniverseInitiative/update-ja… …x-array-api Update jax array api
Merge pull request #7 from DifferentiableUniverseInitiative/update-ja… …x-array-api Update jax array api