What's Changed
- Avoid usage of deprecated
jax.coreAPIs. by @copybara-service[bot] in #326 - [JAX] Replace uses of jnp.float_, jnp.int_, and jnp.complex_ with float, int, and complex in .astype, jnp.zeros, and jnp.ones. by @copybara-service[bot] in #327
- fix tests and drop python 3.11 by @copybara-service[bot] in #332
- Release v0.1.9 by @copybara-service[bot] in #333
Full Changelog: v0.1.8...v0.1.9
Contributors
copybara-service