rytse

sample-factory

High throughput synchronous and asynchronous reinforcement learning

GPT-PINN

Generative Pre-Trained Physics-Informed Neural Networks Implementation

BarrierNet

Safe robot learning

sb3-burn

Implementation of stable-baselines3 in rust with burn

l4casadi

Use PyTorch Models with CasADi and Acados in Python, C(++) or Matlab

efficient-kan

An efficient pure-PyTorch implementation of Kolmogorov-Arnold Network (KAN).

trajax

rotation-steerers

A steerer for D-dimensional keypoint descriptions is a DxD matrix that transforms the descriptions as if they were computed from a rotated image.

NeuralOperators.jl

DeepONets, (Fourier) Neural Operators, Physics-Informed Neural Operators, and more in Julia

OBCA

Optimization-Based Collision Avoidance - a path planner for autonomous navigation

dora

DORA (Dataflow-Oriented Robotic Application) is middleware designed to streamline and simplify the creation of AI-based robotic applications. It offers low latency, composable, and distributed dataflow capabilities. Applications are modeled as directed graphs, also referred to as pipelines.

schedule_free

Schedule-Free Optimization in PyTorch

terra

A grid world environment for high-level earthworks planning in JAX for RL.

PINNacle

Codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.

stereo_matching

stereo rectification and stereo matching for perspective camera and fisheye camera, based on opencv

movement_primitives

Dynamical movement primitives (DMPs), probabilistic movement primitives (ProMPs), and spatially coupled bimanual DMPs for imitation learning.

frunk

Funktional generic type-level programming in Rust: HList, Coproduct, Generic, LabelledGeneric, Validated, Monoid and friends.

egui-snarl

Customizable egui node-graph library

dust3r

DUSt3R: Geometric 3D Vision Made Easy

ORCAgym

phase-based Observations, Rewards, Coupling Ablation

quax

Multiple dispatch over abstract array types in JAX.

torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

safe-control-gym

PyBullet CartPole and Quadrotor environments—with CasADi symbolic a priori dynamics—for learning-based control and RL

mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

mpc-reinforcement-learning

Reinforcement Learning with Model Predictive Control

PythonLinearNonlinearControl

PythonLinearNonLinearControl is a library implementing the linear and nonlinear control theories in python.

differentiable-mpc

MRL

Code repository for the paper - "Matryoshka Representation Learning"

mpc.pytorch

A fast and differentiable model predictive control (MPC) solver for PyTorch.