danielkelshaw / riemaxLinks
Riemannian geometry in JAX
☆16Updated 7 months ago
Alternatives and similar repositories for riemax
Users that are interested in riemax are comparing it to the libraries listed below
Sorting:
- Differentiable QP solver in JAX☆124Updated last year
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated 2 years ago
- A PyTorch implementation of TTGO algorithm and the applications presented in the paper "Tensor Train for Global Optimization Problems in …☆19Updated last year
- Differentiable collision detection for capsules in JAX.☆51Updated 2 years ago
- improved Cross Entropy Method for trajectory optimization☆81Updated 4 years ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆164Updated this week
- ☆230Updated 2 years ago
- Rex is a JAX-powered framework for sim-to-real robotics.☆52Updated 6 months ago
- ☆15Updated 10 months ago
- GPU-friendly, auto-differentiable LQR solver with JAX.☆211Updated 5 months ago
- ☆23Updated 5 years ago
- ☆18Updated 5 years ago
- Python interface to Dojo☆70Updated 3 years ago
- Repository containing the code for Implicit Manifold Gaussian Process.☆16Updated last year
- A minimal(ish) reinforcement learning library that aggregates reliable implementations.☆22Updated 7 months ago
- Implementation of Model Tensor Planning in JAX, TMLR 2025 & ICLR 2026.☆25Updated 7 months ago
- ☆13Updated 2 years ago
- Direct Trajectory Optimization in JAX☆25Updated last year
- ☆17Updated 5 months ago
- Implementation of Motion Planning via Optimal Transport (MPOT) in PyTorch, NeurIPS 2023.☆63Updated 2 months ago
- A pytorch implementation of the paper "Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Images", NIPS, 2015☆15Updated 6 years ago
- A brief JAX tutorial with examples from control theory☆12Updated 3 years ago
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆172Updated 10 months ago
- Learning Discontinuous Contact Dynamics with Smooth, Implicit Representations☆50Updated 3 years ago
- Implementation of different Relative Entropy Policy Search flavors☆13Updated 4 years ago
- This repository contains the code for the paper "Geometry-aware Bayesian Optimization in Roboticsusing Riemannian Matérn Kernels" (CoRL'2…☆21Updated 2 years ago
- A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).☆13Updated last year
- Unified contact simulaton and collision detection☆44Updated 2 years ago
- ☆17Updated 6 years ago
- Jax/Flax Implementation of TD-MPC2☆70Updated 2 weeks ago