This is a compatibility release: JAX 0.4.34 changed how custom autodifferentiation rules work. As of this update we should now be compatible with this. (#87)
Full Changelog: v0.0.8...v0.0.9
This is a compatibility release: JAX 0.4.34 changed how custom autodifferentiation rules work. As of this update we should now be compatible with this. (#87)
Full Changelog: v0.0.8...v0.0.9