☆234Sep 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…☆178Feb 23, 2025Updated last year
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,030Dec 17, 2025Updated 3 months ago
- Sampling-based model predictive control on GPU with JAX/MJX☆256Updated this week
- A fast trajectory optimization library written in Julia☆393Mar 27, 2025Updated 11 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Sep 12, 2023Updated 2 years ago
- Differentiable QP solver in JAX☆135Mar 9, 2026Updated last week
- Implements https://arxiv.org/pdf/2403.00748☆21Jan 26, 2026Updated last month
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆180Mar 10, 2026Updated last week
- Sparse nonlinear least squares in JAX☆273Feb 26, 2026Updated 3 weeks ago
- Performant, differentiable reinforcement learning☆125Aug 2, 2025Updated 7 months 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 iterative LQR (iLQR)☆44Mar 17, 2023Updated 3 years ago
- A differentiable physics engine for robotics☆379Sep 23, 2024Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- ☆627Jan 23, 2026Updated last month
- Conic Augmented Lagrangian Interior-Point SOlver☆77Dec 14, 2022Updated 3 years ago
- Implicit Differentiable Optimal Control (IDOC) with JAX☆12May 11, 2022Updated 3 years ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆3,094Updated this week
- Stanford PhD Thesis☆18Jan 11, 2023Updated 3 years ago
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- A Python reference implementation of rigid body dynamics algorithms☆18Jul 17, 2023Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆57Mar 6, 2026Updated 2 weeks ago
- A Julia package for constrained trajectory optimization using direct methods.☆29Jun 12, 2022Updated 3 years ago
- Rigid transform using Lie groups and Dual Quaternions, written in CasADi!☆88Jul 15, 2025Updated 8 months ago
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆25Oct 1, 2024Updated last year
- PyBullet CartPole and Quadrotor environments—with CasADi symbolic a priori dynamics—for learning-based control and RL☆853Nov 6, 2025Updated 4 months ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆242Dec 22, 2023Updated 2 years ago
- Differentiable collision detection between capsules and padded polygons☆15Oct 20, 2022Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆59Feb 15, 2026Updated last month
- nonlinear control optimization tool☆115Mar 22, 2025Updated 11 months ago
- A fast and differentiable model predictive control (MPC) solver for PyTorch.☆1,045Aug 27, 2024Updated last year
- Use PyTorch Models with CasADi for data-driven optimization or learning-based optimal control. Supports Acados.☆561Jun 5, 2025Updated 9 months ago
- Differentiable trajectory optimization in Julia.☆65Mar 9, 2026Updated last week
- Differentiable convex optimization layers☆2,069Mar 12, 2026Updated last week
- Optax is a gradient processing and optimization library for JAX.☆2,212Updated this week
- Rigid transforms + Lie groups for JAX☆324Apr 24, 2025Updated 10 months ago
- adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.☆212Updated this week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆753Oct 26, 2022Updated 3 years ago
- A library containing a collection of distance and similarity measures for data analysis☆16Feb 19, 2026Updated last month