Skip to content

Diffrax v0.6.2

Choose a tag to compare

View all tags
@github-actions github-actions released this 24 Dec 16:50
· 81 commits to main since this release
v0.6.2
7474749
  • Updated to silence warnings from JAX 0.4.38 (jax.core.Jaxpr -> jax.extend.core.Jaxpr) (Thanks @jakevdp! #544, #546)
  • Added diffrax.ForwardMode for efficient forward-mode autodifferentiation through diffeqsolve. Updated docs to reflect that there are now two main recommended choices, namely ForwardMode for forward-mode and RecursiveCheckpointAdjoint for reverse-mode. (Thanks @johannahaffner! #537)

Full Changelog: v0.6.1...v0.6.2

Contributors

jakevdp and johannahaffner
Assets 2
Loading