PufferAI / PufferLib
Simplifying reinforcement learning for complex game environments
☆2,008Updated this week
Alternatives and similar repositories for PufferLib
Users that are interested in PufferLib are comparing it to the libraries listed below
Sorting:
- Really Fast End-to-End Jax RL Implementations☆870Updated 8 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆322Updated last week
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆564Updated 6 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆312Updated 2 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,779Updated 2 weeks ago
- RL Environments in JAX 🌍☆744Updated 10 months ago
- The Fastest Deep Reinforcement Learning Library☆795Updated this week
- Multi-Agent Reinforcement Learning with JAX☆575Updated this week
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆751Updated last week
- 1 million FPS multi-agent driving simulator☆407Updated this week
- Mastering Diverse Domains through World Models☆1,819Updated last month
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆725Updated this week
- If tinygrad wasn't small enough for you...☆718Updated last year
- Tutorials on tinygrad☆374Updated last month
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆371Updated this week
- (WIP) A small but powerful, homemade PyTorch from scratch.☆550Updated this week
- Can you design a controller to steer a simulated car?☆236Updated 4 months ago
- The Tensor (or Array)☆432Updated 9 months ago
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, T…☆6,973Updated last month
- ☆218Updated last week
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆414Updated last week
- UNet diffusion model in pure CUDA☆602Updated 10 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆836Updated 6 months ago
- NanoGPT (124M) in 3 minutes☆2,533Updated 2 weeks ago
- Schedule-Free Optimization in PyTorch☆2,154Updated last month
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆440Updated 2 weeks ago
- The n-gram Language Model☆1,419Updated 9 months ago
- DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 S…☆1,812Updated 5 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆235Updated last month
- ♟️ Vectorized RL game environments in JAX☆473Updated 2 months ago