PufferAI / PufferLibLinks
Simplifying reinforcement learning for complex game environments
☆2,081Updated this week
Alternatives and similar repositories for PufferLib
Users that are interested in PufferLib are comparing it to the libraries listed below
Sorting:
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆584Updated 7 months ago
- Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.☆774Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,784Updated last month
- Mastering Diverse Domains through World Models☆1,885Updated last month
- Really Fast End-to-End Jax RL Implementations☆882Updated 8 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆332Updated last week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆731Updated last month
- NanoGPT (124M) in 3 minutes☆2,610Updated last week
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆378Updated this week
- Multi-Agent Reinforcement Learning with JAX☆581Updated this week
- A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.☆2,838Updated last week
- Monte Carlo tree search in JAX☆2,494Updated last month
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,390Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆314Updated last week
- Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"☆554Updated 11 months ago
- The Fastest Deep Reinforcement Learning Library☆820Updated 3 weeks ago
- RL Environments in JAX 🌍☆763Updated last week
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆901Updated last year
- (WIP) A small but powerful, homemade PyTorch from scratch.☆553Updated this week
- UNet diffusion model in pure CUDA☆606Updated 11 months ago
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,804Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆455Updated 2 weeks ago
- A suite of test scenarios for multi-agent reinforcement learning.☆705Updated 2 weeks ago
- Evolution Strategies in JAX 🦎☆625Updated last week
- High throughput synchronous and asynchronous reinforcement learning☆905Updated last week
- If tinygrad wasn't small enough for you...☆716Updated last year
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,708Updated last week
- The Tensor (or Array)☆433Updated 9 months ago
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, T…☆7,178Updated last month
- ☆898Updated 11 months ago