Skip to content

JAX Update
Compare
Choose a tag to compare
@zombie-einstein zombie-einstein released this 26 Apr 10:44
· 7 commits to main since this release
0.2.0
43d5fda
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.
  • Use jax.debug.callback instead of deprecated jax.experimental.host_callback (thanks @BirkhoffG )
  • Update min JAX version to 0.4.12

Contributors

BirkhoffG
Assets 2
Loading