Mathematical Programming in JAX
☆146Jun 2, 2025Updated 8 months ago
Alternatives and similar repositories for MPAX
Users that are interested in MPAX are comparing it to the libraries listed below
Sorting:
- ☆27Dec 14, 2023Updated 2 years ago
- Convex Optimization under Uncertainty☆29Feb 16, 2026Updated last week
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆176Feb 23, 2025Updated last year
- Differentiable QP solver in JAX☆132Oct 3, 2024Updated last year
- GPU-based first-order solver for linear programming.☆101Feb 18, 2025Updated last year
- ☆33Nov 27, 2024Updated last year
- ☆18May 14, 2024Updated last year
- Fast QP Solver in JAX☆29Aug 29, 2024Updated last year
- ☆20Dec 7, 2025Updated 2 months ago
- First-order solver for quadratric programming on CPU and GPU.☆19Jun 6, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆544Feb 16, 2026Updated last week
- Experimental first-order solvers for linear and quadratic programming.☆126Nov 3, 2023Updated 2 years ago
- Code for "Explainable Data-Driven Optimization" (ICML 2023)☆14Jul 21, 2023Updated 2 years ago
- ☆50Oct 12, 2025Updated 4 months ago
- A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programming☆635Feb 11, 2026Updated 2 weeks ago
- Visualize linear programming at https://lpviz.net☆33Jan 20, 2026Updated last month
- Learning-to-Optimize for Mixed-Integer Non-Linear Programming☆61Feb 18, 2026Updated last week
- Differentiating optimization programs w.r.t. program parameters☆137Updated this week
- [AAMAS 2025 Oral] CAMP: Collaborative Attention Model with Profiles for Vehicle Routing Problems☆30Dec 3, 2025Updated 2 months ago
- TaskMet Task-driven Metric Learning for Model Learning☆20Feb 9, 2024Updated 2 years ago
- A solver for linear programming with GPU support☆40Updated this week
- ☆10Mar 25, 2024Updated last year
- ☆21Jun 20, 2025Updated 8 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- ☆22Mar 23, 2022Updated 3 years ago
- Julia Package for reading .tsp files☆14Jul 2, 2022Updated 3 years ago
- SParse lINear solvers in JAX☆26Feb 22, 2026Updated last week
- Implements vanilla PDHG algorithm☆10Mar 13, 2023Updated 2 years ago
- Numerical quadrature with JAX☆77Feb 16, 2026Updated last week
- Implementation of "Interior Point Solving for LP-based prediction+optimisation" paper in Neurips 2020.☆21May 16, 2024Updated last year
- ☆27Jan 8, 2024Updated 2 years ago
- Differentiation through cone programs☆117Jan 26, 2026Updated last month
- ☆14Jun 8, 2023Updated 2 years ago
- MISO: Learning Multiple Initial Solutions to Optimization Problems☆16Nov 8, 2024Updated last year
- A brief JAX tutorial with examples from control theory☆12Nov 17, 2022Updated 3 years ago
- Code the AAAI 2019 paper "Melding the Data-Decisions Pipeline: Decision-Focused Learning for Combinatorial Optimization"☆35Feb 12, 2021Updated 5 years ago
- A solver for nonlinear programming with GPU support☆238Updated this week
- Combinatorial optimization layers for machine learning pipelines☆129Nov 24, 2025Updated 3 months ago