jaxipm is the first GPU-batched IPOPT solver in JAX using heterogeneous iteration fusion and iteration-level batching, delivering up to 32.85x higher throughput than standard IPOPT on quadrotor NMPC benchmarks.
Efficient and Modular Im- plicit Differentiation
12 Pith papers cite this work, alongside 37 external citations. Polarity classification is still indexing.
representative citing papers
A differentiable neural framework for learning state- and time-dependent parameters of finite-state mean field games from population trajectories via implicit differentiation.
RLNS regularizes LNS to perform block Gibbs sampling under entropy, interpolating between pseudolikelihood and exact MLE for differentiable combinatorial optimization.
A learning approach trains neural networks to approximate solutions of multiparametric GNEPs using NI gap loss with value surrogates, achieving large speedups and providing new existence conditions for continuous selections.
Proves H^{-1} norm equivalence to expectation over random test functions and introduces SV-PINNs that outperform standard PINNs on eight elliptic problems.
Introduces an inverse-design framework showing that energy barriers and binding-site separations in bistable nanostructures are readily programmable while transition-state location and full profile shape require more design freedom.
A data-driven model for periodic ADC integral nonlinearity in JWST/NIRISS is fitted to ramp residuals and applied to correct the ERS1366 WASP-39b transmission spectrum, reducing systematics at the 30ppm level.
Combined five-PTA dataset yields posterior on SGWB power-law amplitude and index consistent with nonzero signal but below 5-sigma significance, with reconstructed angular correlations matching the Hellings-Downs prediction.
AFSAT realizes FastFourierSAT as a production GPU solver for heterogeneous symmetric pseudo-Boolean SAT via JAX-compiled continuous local search, with tailored DFT for stability and near-linear multi-accelerator scaling.
IQPopt is a JAX-based software tool enabling classical optimization of IQP circuits with thousands of qubits via efficient simulation of Pauli-Z expectation values, plus a module for quantum generative model training.
Mass of 13.7 Earth masses and density 0.4 g cm^{-3} measured for TOI-1883 b, a super-Neptune in the ridge regime around an early-M dwarf, with implications for disk migration and photoevaporation.
Empirical study of parallel continuous local search for SAT finds redundant constraints can slow convergence, CLS works as a hybrid sub-solver, and search stabilizes quickly due to saddle-dense objectives.
citing papers explorer
-
Scaling Nonlinear Optimization: Many Problems One GPU
jaxipm is the first GPU-batched IPOPT solver in JAX using heterogeneous iteration fusion and iteration-level batching, delivering up to 32.85x higher throughput than standard IPOPT on quadrotor NMPC benchmarks.
-
Neural Parameter Calibration for Finite-State Mean Field Games
A differentiable neural framework for learning state- and time-dependent parameters of finite-state mean field games from population trajectories via implicit differentiation.
-
Regularized Large Neighborhood Search
RLNS regularizes LNS to perform block Gibbs sampling under entropy, interpolating between pseudolikelihood and exact MLE for differentiable combinatorial optimization.
-
Learning Approximate Solutions to Multiparametric Generalized Nash Equilibrium Problems
A learning approach trains neural networks to approximate solutions of multiparametric GNEPs using NI gap loss with value surrogates, achieving large speedups and providing new existence conditions for continuous selections.
-
Random test functions, $H^{-1}$ norm equivalence, and stochastic variational physics-informed neural networks
Proves H^{-1} norm equivalence to expectation over random test functions and introduces SV-PINNs that outperform standard PINNs on eight elliptic problems.
-
Designing bistable nanostructures for target behavior
Introduces an inverse-design framework showing that energy barriers and binding-site separations in bistable nanostructures are readily programmable while transition-state location and full profile shape require more design freedom.
-
Calibration of an Analog-to-Digital Conversion Nonlinearity in JWST/NIRISS
A data-driven model for periodic ADC integral nonlinearity in JWST/NIRISS is fitted to ramp residuals and applied to correct the ERS1366 WASP-39b transmission spectrum, reducing systematics at the 30ppm level.
-
Stochastic gravitational-wave background search using data from five pulsar timing arrays
Combined five-PTA dataset yields posterior on SGWB power-law amplitude and index consistent with nonzero signal but below 5-sigma significance, with reconstructed angular correlations matching the Hellings-Downs prediction.
-
Accelerated Fourier SAT (AFSAT): Fully Realising a GPU-based Symmetric Pseudo-Boolean SAT Solver
AFSAT realizes FastFourierSAT as a production GPU solver for heterogeneous symmetric pseudo-Boolean SAT via JAX-compiled continuous local search, with tailored DFT for stability and near-linear multi-accelerator scaling.
-
IQPopt: Fast optimization of instantaneous quantum polynomial circuits in JAX
IQPopt is a JAX-based software tool enabling classical optimization of IQP circuits with thousands of qubits via efficient simulation of Pauli-Z expectation values, plus a module for quantum generative model training.
-
The mass of TOI-1883 b: A low density super-Neptune in the ridge regime transiting an early-M dwarf
Mass of 13.7 Earth masses and density 0.4 g cm^{-3} measured for TOI-1883 b, a super-Neptune in the ridge regime around an early-M dwarf, with implications for disk migration and photoevaporation.
-
A Study of Parallel Continuous Local Search
Empirical study of parallel continuous local search for SAT finds redundant constraints can slow convergence, CLS works as a hybrid sub-solver, and search stabilizes quickly due to saddle-dense objectives.