Mathematical Programming in JAX
☆166Jun 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☆155Updated this week
- Julia interface for cuPDLPx.☆15Jul 10, 2026Updated last month
- Differentiable QP solver in JAX☆165Jul 24, 2026Updated 2 weeks ago
- Fully First-Order Differentiable Optimization☆37Mar 28, 2026Updated 4 months ago
- First-order solver for quadratric programming on CPU and GPU.☆19Jun 6, 2024Updated 2 years ago
- 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.
- SParse lINear solvers in JAX☆53Updated this week
- Fast QP Solver in JAX☆30Aug 29, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆611May 13, 2026Updated 2 months ago
- GPU-based first-order solver for linear programming.☆107Feb 18, 2025Updated last year
- ☆28Jul 22, 2026Updated 2 weeks ago
- A solver for linear programming with GPU support☆41Jul 23, 2026Updated 2 weeks ago
- ☆27Mar 25, 2026Updated 4 months ago
- ☆35Jul 23, 2026Updated 2 weeks ago
- 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
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,056Jun 4, 2026Updated 2 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆65Jan 23, 2026Updated 6 months ago
- LP presolver☆37Updated this week
- Convex Optimization under Uncertainty☆33Mar 31, 2026Updated 4 months ago
- A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programming☆682Jul 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☆248Aug 3, 2026Updated last week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆25Apr 27, 2026Updated 3 months ago
- ☆17Jul 15, 2026Updated 3 weeks ago
- 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.☆321Updated this week
- ☆10Sep 16, 2020Updated 5 years ago
- Implements vanilla PDHG algorithm☆10Mar 13, 2023Updated 3 years ago
- Efficient modeling interface for mathematical optimization in Python☆314Jul 28, 2026Updated last week
- Learning-to-Optimize for Mixed-Integer Non-Linear Programming☆69Mar 10, 2026Updated 5 months ago
- GPU-friendly, auto-differentiable LQR solver with JAX.☆222Aug 8, 2025Updated last year
- 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.
- Combinatorial optimization layers for machine learning pipelines☆132Updated this week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆570Updated this week
- ☆239Jun 21, 2026Updated last month
- ☆361Apr 1, 2026Updated 4 months ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆202Aug 1, 2026Updated last week
- Matrix-free linear algebra in JAX.☆179Jul 1, 2026Updated last month
- Visualize linear programming at https://lpviz.net☆46Jul 12, 2026Updated 3 weeks ago