Releases: eserie/wax-ml
Releases · eserie/wax-ml
v0.6.4
v0.6.3
v0.6.2
v0.6.1
v0.6.0
What's Changed
- replace jax.tree_multi_map (removed in JAX release v0.3.16) by jax.tr… by @eserie in #60
- Faster unroll: do not propagate params in scan state by @eserie in #61
- correct imports or tree utils to avoid FutureWarnings by @eserie in #63
- Remove dependency with eagerpy by @eserie in #62
- Docs by @eserie in #64
- add stateful.py module with vmap_lift_with_state and unroll_lift_with… by @eserie in #65
Full Changelog: v0.5.0...v0.6.0
v0.5.0
v0.4.0
v0.3.2
v0.3.1
Realease v0.3.0
What's Changed
- Renew the PRNG key in the unroll operations by @eserie in #45
- Rework notebook 05 by @eserie in #46
- Refactor online optimizer by @eserie in #47
- VMap module works in contexts without PRNG key by @eserie in #48
- OnlineOptimizer returns updated params by @eserie in #49
- Prepare release by @eserie in #50
Full Changelog: v0.2.0...v0.3.0