Repository navigation
Compatibility with JAX 0.3.25
Pre-release
Pre-release
Ensure compatibility with JAX 0.3.25 by explicitly casting mask count to integer. Also limit install_requires to <0.4.0.
Ensure compatibility with JAX 0.3.25 by explicitly casting mask count to integer. Also limit install_requires to <0.4.0.