2 citations · 2 across the 1 of their papers we have counts for
2 papers
math.OC2024★ 2 cited
Optimistix: modular optimisation in JAX and Equinox
Jason Rader, Terry Lyons, Patrick Kidger
We introduce Optimistix: a nonlinear optimisation library built in JAX and Equinox. Optimistix introduces a novel, modular approach for its minimisers and least-squares solvers. Th…
cs.MS2023
Lineax: unified linear solves and linear least-squares in JAX and Equinox
Jason Rader, Terry Lyons, Patrick Kidger
We introduce Lineax, a library bringing linear solves and linear least-squares to the JAX+Equinox scientific computing ecosystem. Lineax uses general linear operators, and unifies…