mlsanityData to 7 Oct 2026

textttimmrax: A Parallelizable and Differentiable Toolbox for Interval Analysis and Mixed Monotone Reachability in JAX

arXiv 2401.11608 ↗First posted 21 Jan 2024eess.SY cs.LG cs.SYcited by 4

Not on the map. Why?

Abstract (arXiv)

We present an implementation of interval analysis and mixed monotone interval reachability analysis as function transforms in Python, fully composable with the computational framework JAX. The resulting toolbox inherits several key features from JAX, including computational efficiency through Just-In-Time Compilation, GPU acceleration for quick parallelized computations, and Automatic Differentiability. We demonstrate the toolbox's performance on several case studies, including a reachability problem on a vehicle model controlled by a neural network, and a robust closed-loop optimal control problem for a swinging pendulum.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 6 of its 16 references that are in the map. First posted on arXiv 21 Jan 2024.

Builds on

6 of 16 references.

References · 6

  1. 2022On Neural Differential Equationscited by 264
  2. 2021Equinox: neural networks in JAX via callable PyTrees and filtered transformationscited by 143
  3. 2019JuliaReach: a Toolbox for Set-Based Reachabilitycited by 28
  4. 2023POLAR-Express: Efficient and Precise Formal Reachability Analysis of Neural-Network Controlled Systemscited by 12
  5. 2023Efficient Interaction-Aware Interval Analysis of Neural Network Feedback Loopscited by 9
  6. 2023A Toolbox for Fast Interval Arithmetic in numpy with an Application to Formal Verification of Neural Network Controlled Systemscited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 1; 2025: 0; 2026: 3.

4 papers.

  1. 2024Certified Robust Invariant Polytope Training in Neural Controlled ODEscited by 3
  2. 2026Parallel Differentiable Reachability for Learning and Planning with Certified Neural Dynamics and Controllerscited by 2
  3. 2026Kernel-Based Learning of Safety Barrierscited by 0
  4. 2026GPU-Parallel Linearization Error Bounds for Real-Time Robust Optimal Control of Nonlinear and Neural Network Dynamicscited by 0