Diffrax v0.1.1
Diffrax uses some JAX-internal functionality that will shortly be deprecated in JAX. This release adds the appropriate support for both older and newer versions of JAX.
Autogenerated release notes as follows:
What's Changed
- [JAX] Add MHLO lowerings in preparation for xla.lower_fun() removal by @hawkinsp in #91
- Bump version by @patrick-kidger in #92
New Contributors
Full Changelog: v0.1.0...v0.1.1