REVIEW 9 cited by
Optimistix: modular optimisation in JAX and Equinox
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
read the original abstract
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. This modularity relies on new practical abstractions for optimisation which we call search and descent, and which generalise classical notions of line search, trust-region, and learning-rate algorithms. It provides high-level APIs and solvers for minimisation, nonlinear least-squares, root-finding, and fixed-point iteration. Optimistix is available at https://github.com/patrick-kidger/optimistix.
Forward citations
Cited by 9 Pith papers
-
Numerical polology: towards next-generation model-building for cosmology
Numerical polology framework samples coupling space to discover ghost-free tensor field theories up to rank three for cosmology, then applies resulting priors to black hole superradiance, dynamical dark energy, and GW data.
-
Complex surface patterning in homo- and heteroepitaxial contexts: (simultaneous) step bunching and step meandering
A linked discrete-continuum model demonstrates that step bunching and meandering can coexist as common growth modes on vicinal surfaces, yielding diverse patterns beyond separate limiting cases.
-
jaxdae: A JAX-native Differentiable Solver for Differential-Algebraic Equations in Coupled Multi-physics
jaxdae provides the first JAX-native differentiable DAE solver, using a frozen-grid BDF-2 replay adjoint for reverse-mode gradients and XLA-fused batched sweeps.
-
Thermo-Structural Topology Optimization Considering Nonlinear Creep
A differentiable JAX-based topology-optimization framework minimizes accumulated creep deformation under thermo-mechanical loading using Norton's power law with Arrhenius temperature dependence, demonstrated on 2D ben...
-
A guided residual search for nonlinear state-space identification
A three-stage identification method (guided residual search, neural-network residual learning, multiple-shooting refinement) makes nonlinear state-space model fitting more reliable.
-
linrax: A JAX Compatible, Simplex Method Linear Program Solver
linrax implements the simplex method in JAX, enabling JIT-compiled, differentiable LP solving with support for degenerate constraints.
-
Towards real-time surrogate-free Bayesian inversion for neutron reflectometry
Automatic differentiation through the Abeles reflectivity model gives exact gradients for neutron reflectometry, making Hamiltonian Monte Carlo and variational inference fast, practical alternatives to slow MCMC for t...
-
Tolerancing the PIAA-ZWFS: a practical and robust wavefront sensor that approaches the fundamental sensitivity limit
Typical manufacturing and alignment errors do not significantly degrade the PIAA-ZWFS's wavefront sensing performance, preserving a large margin over a standard ZWFS.
-
A parallel-in-time Newton's method-based ODE solver
Newton's method on the fully discretized ODE, accelerated by parallel prefix sums, solves nonlinear IVPs with logarithmic span per iteration and beats Parareal in GPU runtime comparisons.
Discussion (0). Continue with ORCID to comment.