Lists (3)
Starred repositories
VeOmni: Scaling any Modality Model Training to any Accelerators with PyTorch native Training Framework
ByteCheckpoint: An Unified Checkpointing Library for LFMs
A Python interface for reinforcement learning environments
Convert Machine Learning Code Between Frameworks
A high-performance distributed file system designed to address the challenges of AI training and inference workloads.
1 million FPS multi-agent driving simulator
A curated list of awesome exploration RL resources (continually updated)
Collection of reinforcement learning algorithms
Processed / Cleaned Data for Paper Copilot
Interactively inspect module inputs, outputs, parameters, and gradients.
Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation
Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official implementation of the Soft Actor-Critic algorithm.
Best practices & guides on how to write distributed pytorch training code
Simple single-file baselines for Q-Learning in pure-GPU setting
For optimization algorithm research and development.
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
Massively Parallel Deep Reinforcement Learning. 🔥
Simple and easily configurable grid world environments for reinforcement learning
Isaac Gym Reinforcement Learning Environments
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
A high-performance reinforcement learning library in jax specialized for robotic learning
a modular reinforcement learning library with JAX agents
Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)