Challenging Memory-based Deep Reinforcement Learning Agents

Library for industrial alignment.

Awesome In-Context RL: A curated list of In-Context Reinforcement Learning

Implementation of "MADiff: Offline Multi-agent Learning with Diffusion Models"

Accelerated minigrid environments with JAX

A collection of MARL benchmarks based on TorchRL

Hackable pure-torch implementation of diffusion models

Hydra sweeper integration of our favorite optimization packages, utilizing ask-and-tell interfaces.

The official repo for the paper "In-Context Imitation Learning via Next-Token Prediction"

From Imitation to Refinement -- Residual RL for Precise Visual Assembly

LayoutFlow: Flow Matching for Layout Generation [Andrade Guerreiro et al., ECCV 2024]

Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold

📦 Python library for Stochastic Processes Simulation and Visualisation

TorchCFM: a Conditional Flow Matching library

ALIEN is a CUDA-powered artificial life simulation program.

Minimal implementation of scalable rectified flow transformers, based on SD3's approach

NumPy / Jax drawing library.

Simple and readable code for training and sampling from diffusion models

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

JAX-based environments and RL baselines for studying goal misgeneralisation.

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab

🔥🔥🔥A curated list of papers on recent diffusion-based high-resolution image and video synthesis works.

Datasets with baselines for offline multi-agent reinforcement learning.

Clean single-file implementation of offline RL algorithms in JAX