v0.2.4
What's Changed
- Reduced memory and debugged reverse AD by @cgiovanetti in #22
Three major fixes:
- Reduced memory usage dramatically.
- Added reverse AD option (adjoint option at initialization) and checkpointing. Use
RecursiveAdjointCheckpointinadjointinput toModelat initialization for this option. - Fixed double
jnp.wherebug leading to NaNs in gradients.
Full Changelog: v0.2.3...v0.2.4