Mathematical Programming in JAX
☆173Jun 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☆165Oct 11, 2026Updated 2 weeks ago
- Julia interface for cuPDLPx.☆15Jul 10, 2026Updated 3 months ago
- Differentiable QP solver in JAX☆174Jul 24, 2026Updated 3 months ago
- Fully First-Order Differentiable Optimization☆41Mar 28, 2026Updated 7 months ago
- HPR-LP-C: A C implementation of the Halpern Peaceman--Rachford (HPR) method for solving linear programming (LP) problems on GPUs.☆18Sep 26, 2026Updated last month
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆34Updated this week
- First-order solver for quadratric programming on CPU and GPU.☆20Jun 6, 2024Updated 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆630Aug 11, 2026Updated 2 months ago
- The GPU-based primal-dual conic solver☆14Oct 1, 2026Updated 3 weeks ago
- Fast QP Solver in JAX☆28Aug 29, 2024Updated 2 years ago
- ☆28Mar 25, 2026Updated 7 months ago
- SParse lINear solvers in JAX☆59Aug 4, 2026Updated 2 months ago
- ☆38Updated this week
- GPU-based first-order solver for linear programming.☆108Feb 18, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,067Oct 16, 2026Updated 2 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆587Sep 26, 2026Updated last month
- A Learning-to-Optimize (L2O) package for Mixed-Integer Nonlinear Programming (MINLP)☆12Mar 15, 2026Updated 7 months ago
- Convex Optimization under Uncertainty☆35Oct 12, 2026Updated 2 weeks ago
- A solver for linear programming with GPU support☆42Sep 21, 2026Updated last month
- ☆59Jun 5, 2026Updated 4 months ago
- A solver for nonlinear programming with GPU support☆252Updated this week
- A small library of paramaterizations and parameter constraints for PyTrees.☆68Jan 23, 2026Updated 9 months ago
- Experimental first-order solvers for linear and quadratic programming.☆128Nov 3, 2023Updated 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.
- 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
- Learning parametric convex functions☆15Jul 5, 2026Updated 3 months ago
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- The Moreau solver.☆38Updated this week
- Structured Optimization using CUDA☆24Oct 23, 2026Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆327Oct 14, 2026Updated 2 weeks ago
- LP presolver☆42Updated this week
- Cone-Aligned Vector Estimation for End-to-End Binary Linear Programming☆10May 25, 2026Updated 5 months ago
- ☆26Oct 10, 2026Updated 2 weeks 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.
- Combinatorial optimization layers for machine learning pipelines☆133Aug 20, 2026Updated 2 months ago
- A PyTorch-based End-to-End Predict-then-Optimize Library for Linear and Integer Programming☆694Oct 7, 2026Updated 3 weeks ago
- pytransform3d in JAX (experimental)☆12Oct 4, 2026Updated 3 weeks ago
- Efficient modeling interface for mathematical optimization in Python☆319Oct 14, 2026Updated 2 weeks ago
- Augments popular QP solvers with differentiability☆24May 5, 2026Updated 5 months ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆770Sep 18, 2026Updated last month
- GPU-friendly, auto-differentiable LQR solver with JAX.☆225Aug 8, 2025Updated last year