Skip to content

v0.2.4

Choose a tag to compare

@cgiovanetti cgiovanetti released this 04 May 20:12
· 37 commits to main since this release
4b0a9f0

What's Changed

Three major fixes:

  • Reduced memory usage dramatically.
  • Added reverse AD option (adjoint option at initialization) and checkpointing. Use RecursiveAdjointCheckpoint in adjoint input to Model at initialization for this option.
  • Fixed double jnp.where bug leading to NaNs in gradients.

Full Changelog: v0.2.3...v0.2.4