anindex / ssax
Implementation of Sinkhorn Step in JAX, NeurIPS 2023.
☆42Updated 3 months ago
Alternatives and similar repositories for ssax:
Users that are interested in ssax are comparing it to the libraries listed below
- Pytorch Implementation of TinyMPC☆58Updated last year
- Implementation of GTMP in JAX.☆19Updated last month
- Playing diabolo with two robot arms in ROS + Gazebo☆30Updated 3 years ago
- Produce a simulate-able SDF of an arbitrary mesh with convex decomposition.☆25Updated 2 years ago
- Direct Trajectory Optimization in JAX☆17Updated 3 months ago
- Programmatically compile xacro files to URDF (no ROS required).☆32Updated last month
- Example URDF file external data loader plugin for Rerun☆23Updated this week
- Sampling-based model predictive control on GPU with JAX/MJX☆84Updated last month
- CNN+LSTM type "classical" visuomotor behavior cloning framework☆15Updated 11 months ago
- Set of utilities for visualizing robots in web-based visualizer MeshCat.☆29Updated 3 weeks ago
- For imitation learning using real robot (PR2 only now)☆8Updated 10 months ago
- ☆33Updated 2 years ago
- ☆28Updated 2 months ago
- Code for R:SS 2021 paper RMP2: A Structured Composable Policy Class for Robot Learning.☆36Updated 3 years ago
- Implementation of Motion Planning via Optimal Transport (MPOT) in PyTorch, NeurIPS 2023.☆53Updated last year
- An efficient task-space capacity calculation package for robotics and biomechanics☆28Updated this week
- Implementation of Riemannian Motion Policies in ROS☆26Updated 6 years ago
- State-of-the-art method for model-based dexterous manipulation☆87Updated 6 months ago
- Model Predictive Path Integral Control (MPPI) with PyTorch☆13Updated last year
- Differentiable collision detection for capsules in JAX.☆40Updated last year
- ☆30Updated 2 weeks ago
- Implement various MP methods.☆14Updated 7 months ago
- Open source implementation to the paper "IKFlow: Generating Diverse Inverse Kinematics Solutions"☆60Updated 2 months ago
- System identification with MuJoCo☆46Updated last month
- assemble robot using GUI☆19Updated 7 months ago
- Sandbox code for "Tutorial on Ergodic Control @ ICRA 2024"☆20Updated 7 months ago
- ☆35Updated this week
- Software that integrates various imitation learning methods and benchmark task environments to provide baselines for robot manipulation☆79Updated this week
- ☆13Updated 6 months ago