☆233Sep 3, 2023Updated 2 years ago
Alternatives and similar repositories for trajax
Users that are interested in trajax are comparing it to the libraries listed below
Sorting:
- 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
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- A Julia package for constrained iterative LQR (iLQR)☆44Mar 17, 2023Updated 2 years ago
- A fast trajectory optimization library written in Julia☆389Mar 27, 2025Updated 11 months ago
- Differentiable QP solver in JAX☆132Oct 3, 2024Updated last year
- Conic Augmented Lagrangian Interior-Point SOlver☆77Dec 14, 2022Updated 3 years ago
- Sparse nonlinear least squares in JAX☆271Jan 21, 2026Updated last month
- Sampling-based model predictive control on GPU with JAX/MJX☆251Feb 21, 2026Updated last week
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆57Jan 7, 2026Updated last month
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Sep 12, 2023Updated 2 years ago
- Source code for the examples accompanying the paper "Learning convex optimization control policies."☆85Nov 14, 2022Updated 3 years ago
- A Julia package for constrained trajectory optimization using direct methods.☆29Jun 12, 2022Updated 3 years ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆176Feb 19, 2026Updated last week
- ☆620Jan 23, 2026Updated last month
- A differentiable physics engine for robotics☆369Sep 23, 2024Updated last year
- RL Environments in JAX 🌍☆864May 30, 2025Updated 9 months ago
- Differentiable trajectory optimization in Julia.☆65Nov 25, 2025Updated 3 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆3,075Updated this week
- Performant, differentiable reinforcement learning☆125Aug 2, 2025Updated 6 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆261Oct 31, 2025Updated 4 months ago
- A Python package for probabilistic state space modeling with JAX☆929Jan 6, 2026Updated last month
- adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.☆212Updated this week
- nonlinear control optimization tool☆115Mar 22, 2025Updated 11 months ago
- We are implementing differentiable models of robot manipulators, which allows us to learn typically assumed to be known models of robots …☆261Mar 22, 2023Updated 2 years ago
- Differentiable convex optimization layers☆2,060Updated this week
- PyBullet CartPole and Quadrotor environments—with CasADi symbolic a priori dynamics—for learning-based control and RL☆842Nov 6, 2025Updated 3 months ago
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- A fast and differentiable model predictive control (MPC) solver for PyTorch.☆1,038Aug 27, 2024Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆751Oct 26, 2022Updated 3 years ago
- Rigid transform using Lie groups and Dual Quaternions, written in CasADi!☆88Jul 15, 2025Updated 7 months ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Use PyTorch Models with CasADi for data-driven optimization or learning-based optimal control. Supports Acados.☆557Jun 5, 2025Updated 8 months ago
- ☆188Mar 6, 2025Updated 11 months ago
- Differentiable collision detection between capsules and padded polygons☆15Oct 20, 2022Updated 3 years ago
- Stanford PhD Thesis☆18Jan 11, 2023Updated 3 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Model Predictive Path Integral (MPPI) with approximate dynamics implemented in pytorch☆675Nov 14, 2024Updated last year
- ☆31Jan 25, 2023Updated 3 years ago