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☆123Updated last year
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated 2 years ago
- Differentiable collision detection for capsules in JAX.☆51Updated 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
- ☆14Updated 9 months ago
- Rex is a JAX-powered framework for sim-to-real robotics.☆52Updated 6 months ago
- Direct Trajectory Optimization in JAX☆25Updated last year
- ☆18Updated 5 years ago
- GPU-friendly, auto-differentiable LQR solver with JAX.☆209Updated 4 months ago
- ☆23Updated 5 years ago
- Python interface to Dojo☆68Updated 3 years ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆165Updated this week
- Implementation of Model Tensor Planning in JAX, TMLR 2025 & ICLR 2026.☆25Updated 6 months ago
- ☆12Updated 2 years ago
- A minimal(ish) reinforcement learning library that aggregates reliable implementations.☆21Updated 6 months ago
- ☆12Updated last year
- ☆17Updated 4 months ago
- improved Cross Entropy Method for trajectory optimization☆81Updated 4 years ago
- ☆229Updated 2 years ago
- ☆92Updated last week
- Open source implementation of "CppFlow: Generative Inverse Kinematics for Efficient and Robust Cartesian Path Planning" (ICRA 2024)☆23Updated last month
- Implement Differentiable Kinematics Tree & Planning Objectives in PyTorch given URDF robot models.☆45Updated last year
- Implement various MP methods.☆18Updated last year
- A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).☆13Updated last year
- Generative and Graphical Inverse Kinematics☆29Updated last year
- Robot Operation code for the LIS lab☆24Updated last month
- Code for R:SS 2021 paper RMP2: A Structured Composable Policy Class for Robot Learning.☆42Updated 4 years ago
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆169Updated 9 months ago
- Zonotopes in Python☆18Updated 7 months ago
- State-of-the-art method for model-based dexterous manipulation☆125Updated last year