Mathematical Programming in JAX
☆170Jun 2, 2025Updated last year
Alternatives and similar repositories for MPAX
Users that are interested in MPAX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A GPU-Accelerated First-Order LP Solver☆160Updated this week
- Julia interface for cuPDLPx.☆15Jul 10, 2026Updated last month
- Differentiable QP solver in JAX☆167Jul 24, 2026Updated last month
- Fully First-Order Differentiable Optimization☆39Mar 28, 2026Updated 5 months ago
- First-order solver for quadratric programming on CPU and GPU.☆20Jun 6, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- SParse lINear solvers in JAX☆53Aug 4, 2026Updated 3 weeks ago
- Fast QP Solver in JAX☆30Aug 29, 2024Updated 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆617Aug 11, 2026Updated 2 weeks ago
- GPU-based first-order solver for linear programming.☆107Feb 18, 2025Updated last year
- ☆31Updated this week
- A solver for linear programming with GPU support☆41Aug 16, 2026Updated 2 weeks ago
- ☆27Mar 25, 2026Updated 5 months ago
- ☆35Updated this week
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆185Feb 23, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,059Jun 4, 2026Updated 2 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆65Jan 23, 2026Updated 7 months ago
- LP presolver☆37Aug 18, 2026Updated last week
- Convex Optimization under Uncertainty☆33Aug 18, 2026Updated last week
- A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programming☆685Jul 10, 2026Updated last month
- Experimental first-order solvers for linear and quadratic programming.☆129Nov 3, 2023Updated 2 years ago
- ☆59Jun 5, 2026Updated 2 months ago
- Augments popular QP solvers with differentiability☆24May 5, 2026Updated 3 months ago
- A solver for nonlinear programming with GPU support☆250Updated this week
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆26Apr 27, 2026Updated 4 months ago
- ☆18Jul 15, 2026Updated last month
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆325Aug 6, 2026Updated 3 weeks ago
- ☆10Sep 16, 2020Updated 5 years ago
- Implements vanilla PDHG algorithm☆10Mar 13, 2023Updated 3 years ago
- Efficient modeling interface for mathematical optimization in Python☆315Aug 11, 2026Updated 2 weeks ago
- Learning-to-Optimize for Mixed-Integer Non-Linear Programming☆69Mar 10, 2026Updated 5 months ago
- GPU-friendly, auto-differentiable LQR solver with JAX.☆223Aug 8, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Combinatorial optimization layers for machine learning pipelines☆132Aug 20, 2026Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆576Updated this week
- ☆242Jun 21, 2026Updated 2 months ago
- ☆362Apr 1, 2026Updated 4 months ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆203Aug 1, 2026Updated 3 weeks ago
- Matrix-free linear algebra in JAX.☆179Jul 1, 2026Updated last month
- Visualize linear programming at https://lpviz.net☆48Aug 17, 2026Updated last week