#
Starred repositories
Official implementation for the paper "Full-Order Sampling-Based MPC for Torque-Level Locomotion Control via Diffusion-Style Annealing". DIAL-MPC is a novel sampling-based MPC framework for legged …
Official implementation for the paper "Model-based Diffusion for Trajectory Optimization". Model-based diffusion (MBD) is a novel diffusion-based trajectory optimization framework that employs a dy…
Fatrop is a nonlinear optimal control problem solver that aims to be fast, support a broad class of optimal control problems and achieve a high numerical robustness.
Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, JAXADi enables the creation of efficient code that runs smoot…
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
Get up and running with Llama 3.2, Mistral, Gemma 2, and other large language models.
State-of-the-art method for model-based dexterous manipulation
Einsum-like high-level array sharding API for JAX
Implementation of drake contact surfaces in MuJoCo.
Example repository to show how to build and deploy jupyterlite with pixi
A lightweight library for portable low-level GPU computation using WebGPU.
NVIDIA Math Libraries for the Python Ecosystem
Experimental tool to embed a pixi project in a PDF (e.g. research paper)
🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX
Jupyter Notebooks as Markdown Documents, Julia, Python or R scripts
pyOpenSci's guidebook for package authors, reviewers, and editors
PyTorch Implementation of Physics-informed Neural Networks
Rigid transform using Lie groups and Dual Quaternions, written in CasADi!
Experimental repository for Moveit2 - Drake integration
A grid world environment for high-level earthworks planning in JAX for RL.
🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning
2D and 3D physics engines focused on performance.
