PyCallChainRules v0.4.0
- Add integration and example usage with Lux.jl and jax.
Merged pull requests:
- Add training examples for mixing Flux and Torch layers (#22) (@rejuvyesh)
- use BatchNorm removal function from
functorch
(#25) (@rejuvyesh) - CompatHelper: bump compat for Functors to 0.3, (keep existing compat) (#26) (@github-actions[bot])
- Add Lux.jl integration with Jax (#27) (@rejuvyesh)