Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, JAXADi enables the creation of efficient code that runs smoothly on CPUs, GPUs, and TPUs.
☆176Feb 23, 2025Updated last year
Alternatives and similar repositories for jaxadi
Users that are interested in jaxadi are comparing it to the libraries listed below
Sorting:
- ☆326Jan 14, 2026Updated last month
- Sampling-based model predictive control on GPU with JAX/MJX☆251Feb 21, 2026Updated last week
- Differentiable QP solver in JAX☆132Oct 3, 2024Updated last year
- adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.☆212Feb 24, 2026Updated last week
- Use PyTorch Models with CasADi for data-driven optimization or learning-based optimal control. Supports Acados.☆557Jun 5, 2025Updated 8 months ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆179Updated this week
- Low-Complexity Inverse Kinematics☆102Updated this week
- Fatrop is a nonlinear optimal control problem solver that aims to be fast, support a broad class of optimal control problems and achieve …☆263Updated this week
- GPU-Accelerated Trajectory Optimization☆34Updated this week
- The Simple Simulator: Simulation Made Simple☆216Jan 7, 2026Updated last month
- A versatile and efficient C++ library for real-time constrained trajectory optimization☆277Feb 19, 2026Updated last week
- Numerical Inverse Kinematics solver based on JAX + MJX☆145Jul 31, 2025Updated 7 months ago
- Rigid transform using Lie groups and Dual Quaternions, written in CasADi!☆88Jul 15, 2025Updated 7 months ago
- A GPU-Accelerated Quadratic Programming Solver for Model-Predictive Control☆140Nov 18, 2025Updated 3 months ago
- [ARCHIVED] Development continues at https://github.com/Simple-Robotics/aligator. [Old description] A primal-dual augmented Lagrangian sol…☆68Sep 26, 2025Updated 5 months ago
- Official implementation for the paper "Model-based Diffusion for Trajectory Optimization". Model-based diffusion (MBD) is a novel diffusi…☆327Mar 27, 2025Updated 11 months ago
- RAPTOR: RAPid and Robust Trajectory Optimization for Robots☆167May 17, 2025Updated 9 months ago
- Use PyTorch Models with CasADi and Acados☆242Nov 1, 2023Updated 2 years ago
- Control Barrier Functions in Python and Jax☆151Feb 5, 2026Updated 3 weeks ago
- Community-built test set to benchmark QP solvers☆15May 7, 2025Updated 9 months ago
- ☆233Sep 3, 2023Updated 2 years ago
- Library of rigid-body dynamics algorithms for systems with kinematic loops.☆50Jan 30, 2026Updated last month
- Implementation of numerical solvers used in the Machines in Motion Laboratory☆85Jan 20, 2026Updated last month
- ☆43Oct 2, 2024Updated last year
- Conic Augmented Lagrangian Interior-Point SOlver☆77Dec 14, 2022Updated 3 years ago
- Differentiable collision detection for capsules in JAX.☆52Dec 2, 2023Updated 2 years ago
- Suite of parametrized controller with simulation environments for co-design of Humanoid Robots☆25Nov 3, 2025Updated 4 months ago
- an intuitive optimization tool tailored to robotics, based on CasADi☆81Feb 12, 2026Updated 2 weeks ago
- Model Predictive Control in JAX☆176Feb 13, 2026Updated 2 weeks ago
- A Python-based Trajectory Optimization Framework☆22Oct 29, 2024Updated last year
- ☆30Jan 23, 2023Updated 3 years ago
- Fast and embedded solvers for nonlinear optimal control and nonlinear model predictive control☆1,249Updated this week
- Pytorch Implementation of TinyMPC☆63Jan 17, 2024Updated 2 years ago
- Numerical experiments for the paper: "MPCGPU: Real-Time Nonlinear Model Predictive Control through Preconditioned Conjugate Gradient on t…☆72Dec 12, 2024Updated last year
- Quadratic programming solvers in Python with a unified API☆735Feb 17, 2026Updated 2 weeks ago
- Templated C++/CUDA implementation of Model Predictive Path Integral Control (MPPI)☆253Dec 3, 2025Updated 3 months ago
- Official implementation for the paper "Full-Order Sampling-Based MPC for Torque-Level Locomotion Control via Diffusion-Style Annealing". …☆940May 28, 2025Updated 9 months ago
- HILO-MPC is a Python toolbox for easy, flexible and fast development of machine-learning-supported optimal control and estimation problem…☆198Updated this week
- The Advanced Proximal Optimization Toolbox☆544Feb 20, 2026Updated last week