Lineax: unified linear solves and linear least-squares in JAX and Equinox. - Jason Rader; Terry J. Lyons; Patrick Kidger (CoRR, 2023)

Optimistix: modular optimisation in JAX and Equinox. - Jason Rader; Terry J. Lyons; Patrick Kidger (CoRR, 2024)