danielkelshaw / riemax
Riemannian geometry in JAX
☆13Updated last week
Alternatives and similar repositories for riemax:
Users that are interested in riemax are comparing it to the libraries listed below
- A PyTorch implementation of TTGO algorithm and the applications presented in the paper "Tensor Train for Global Optimization Problems in …☆18Updated 3 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated last year
- ☆18Updated 4 years ago
- Differentiable QP solver in JAX☆85Updated 4 months ago
- Repository containing the code for Implicit Manifold Gaussian Process.☆13Updated 6 months ago
- Score-Guided Planning☆13Updated last year
- ☆34Updated 3 years ago
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to …☆32Updated 2 years ago
- A brief JAX tutorial with examples from control theory☆12Updated 2 years ago
- This repository contains the code for the paper "Geometry-aware Bayesian Optimization in Roboticsusing Riemannian Matérn Kernels" (CoRL'2…☆19Updated last year
- Python interface to Dojo☆65Updated 2 years ago
- ☆21Updated 4 years ago
- This repository contains the source code to perform Geometry-aware Bayesian Optimization (GaBO) on Riemannian manifolds.☆50Updated 3 years ago
- improved Cross Entropy Method for trajectory optimization☆73Updated 3 years ago
- Differentiable collision detection for capsules in JAX.☆40Updated last year
- Python bindings to some optimization benchmarks (robotics problems), in order to constrained optimization solvers. Includes also an inter…☆21Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆24Updated last week
- Implementation of Motion Planning via Optimal Transport (MPOT) in PyTorch, NeurIPS 2023.☆53Updated last year
- ☆11Updated last year
- ☆20Updated 2 months ago
- A PyTorch library for all things nonlinear control and reinforcement learning.☆44Updated 3 years ago
- ☆44Updated 3 weeks ago
- A pytorch implementation of the paper "Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Images", NIPS, 2015☆14Updated 5 years ago
- Direct Trajectory Optimization in JAX☆17Updated 3 months ago
- Optimal Control Tutorials☆14Updated 4 years ago
- Performant, differentiable reinforcement learning☆25Updated last year
- Robot Reinforcement Learning on the Constraint Manifold☆50Updated 3 years ago
- Tools for exploiting Morphological Symmetries in robotics☆64Updated 5 months ago
- A differentiable simulator for robotic systems.☆30Updated 2 years ago