Release v0.2.2
What's Changed
Features
- Support simple sequential batching rule (support for
jax.vmapandjax.jacobian) (#47)
Bug Fixes
- Divergence free /= incompressibility (#21)
- Use has_side_effect=True in FFI calls (#44)
- Update version in vectoradd_jax example to not result in invalid docker tag (#71)
- (deps) Use jnp.moveaxis instead of jax.batching.moveaxis (#72)
- Broken link in docs (#76)
Documentation
Full diff: v0.2.1...v0.2.2