Direct Trajectory Optimization in JAX
☆27Nov 12, 2024Updated last year
Alternatives and similar repositories for drax
Users that are interested in drax are comparing it to the libraries listed below
Sorting:
- Sampling-based model predictive control on GPU with JAX/MJX☆255Updated this week
- Differential Dynamic Programming and Iterative LQR in Drake☆78Jun 10, 2024Updated last year
- Suite of parametrized controller with simulation environments for co-design of Humanoid Robots☆25Nov 3, 2025Updated 4 months ago
- The Simple Simulator: Simulation Made Simple☆216Jan 7, 2026Updated 2 months ago
- New version: https://github.com/chungmin99/pyroki☆19Apr 13, 2025Updated 10 months ago
- Differentiable QP solver in JAX☆132Oct 3, 2024Updated last year
- A PyTorch implementation of MPC as a Function Approximator☆19Sep 27, 2021Updated 4 years ago
- Official implementation for the paper "Model-based Diffusion for Trajectory Optimization". Model-based diffusion (MBD) is a novel diffusi…☆328Mar 27, 2025Updated 11 months ago
- GPU-Accelerated Trajectory Optimization☆34Feb 26, 2026Updated last week
- A Python implementation of "Expansion-GRR: Efficient Generation of Smooth Global Redundancy Resolution Roadmaps"☆12Feb 10, 2026Updated 3 weeks ago
- System identification with MuJoCo☆83Jan 15, 2025Updated last year
- Generative Predictive Control☆57Sep 30, 2025Updated 5 months ago
- Rex is a JAX-powered framework for sim-to-real robotics.☆52Jun 11, 2025Updated 8 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
- ☆101Dec 11, 2025Updated 2 months ago
- Implementation of GTMP in JAX, RA-L 2025 & ICRA 2026.☆31Apr 16, 2025Updated 10 months ago
- A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).☆13Jul 18, 2024Updated last year
- Performant, differentiable reinforcement learning☆23Jun 16, 2023Updated 2 years ago
- an intuitive optimization tool tailored to robotics, based on CasADi☆81Feb 12, 2026Updated 3 weeks ago
- RBDL - Rigid Body Dynamics Library☆12Jul 27, 2022Updated 3 years ago
- ☆17Jan 22, 2016Updated 10 years ago
- CppADCodeGen with an easy Eigen interface and Python bindings.☆15May 23, 2022Updated 3 years ago
- 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
- Proximal Policy Optimization (PPO) written in C++ with PyTorch (LibTorch)☆17Jul 21, 2024Updated last year
- Non-linear model predictive control (NMPC) library☆63Jun 12, 2025Updated 8 months ago
- Python module with simulation backends☆17Oct 22, 2020Updated 5 years ago
- ☆41Updated this week
- A versatile and efficient C++ library for real-time constrained trajectory optimization☆281Updated this week
- Examples for PlaCo library☆22Dec 1, 2025Updated 3 months ago
- Stanford PhD Thesis☆18Jan 11, 2023Updated 3 years ago
- Implement various MP methods.☆18Jul 12, 2024Updated last year
- ☆15Oct 6, 2021Updated 4 years ago
- NonOpt: Nonconvex, Nonsmooth Optimizer☆19Feb 10, 2026Updated 3 weeks ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆179Updated this week
- A Python reference implementation of rigid body dynamics algorithms☆18Jul 17, 2023Updated 2 years ago
- gym environment for drone 2D active perception in dynamic environment☆15Feb 9, 2026Updated 3 weeks ago
- Numerical Inverse Kinematics solver based on JAX + MJX☆145Jul 31, 2025Updated 7 months ago
- Implement Differentiable Kinematics Tree & Planning Objectives in PyTorch given URDF robot models.☆46Aug 14, 2024Updated last year
- Python bindings for open source MuJoCo☆38Feb 21, 2022Updated 4 years ago