Open-Source Models for VLA Arena

Official implementation of "NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flows"

turn your phone into a robot teleop setup

cadrille: Multi-modal CAD Reconstruction with Online Reinforcement Learning

Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025

[ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentials"

RoboVerse: Towards a Unified Platform, Dataset and Benchmark for Scalable and Generalizable Robot Learning

Vintix: Action Model via In-Context Reinforcement Learning - - — ICML 2025

MLGym A New Framework and Benchmark for Advancing AI Research Agents

The official implementation of flow Q-learning (FQL)

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

An Open-Ended Agentic Simulator

Machine Learning library for the emerging Mojo/Python ecosystem

XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025

TORAX: Tokamak transport simulation in JAX

A JAX research toolkit for building, editing, and visualizing neural networks.

Material for gpu-mode lectures

A CLI for following arxiv authors.

[NeurIPS '23 Spotlight] Thought Cloning: Learning to Think while Acting by Imitating Human Thinking

Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"

Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training

Foundation Policies with Hilbert Representations (ICML 2024)

Official implementation of the paper "Linear Transformers with Learnable Kernel Functions are Better In-Context Models"

Because we don't have enough time to read everything

A reinforcement learning framework based on MLX.

Official Implementation for "In-Context Reinforcement Learning for Variable Action Spaces"

Official Implementation for "In-Context Reinforcement Learning from Noise Distillation"