Update
scansparsity propagation for JAX 0.11's new jaxpr representation, which replaces thenum_consts/num_carryparameters withft_in/ft_out.
Following upstream, asdex now requiresjax>=0.11.0and Python>=3.12(#180)
Merged pull requests
- chore: fix DOI badge (#179) (@gdalle)
- fix: update scan propagation for JAX 0.11's num_consts/num_carry removal (#180) (@gdalle)
Closed issues
- Trace through linearization (#184)
New Contributors
Full Changelog: v0.5.1...v0.5.2