← ClaudeAtlas

pufferliblisted

This skill should be used when working with reinforcement learning tasks including high-performance RL training, custom environment development, vectorized parallel simulation, multi-agent systems, or integration with existing RL environments (Gymnasium, PettingZoo, Atari, Procgen, etc.). Use this skill for implementing PPO training, creating PufferEnv environments, optimizing RL performance, or developing policies with CNNs/LSTMs.
aiskillstore/marketplace · ★ 334 · AI & Automation · score 80
Install: claude install-skill aiskillstore/marketplace
# PufferLib - High-Performance Reinforcement Learning ## Overview PufferLib is a high-performance reinforcement learning library designed for fast parallel environment simulation and training. It achieves training at millions of steps per second through optimized vectorization, native multi-agent support, and efficient PPO implementation (PuffeRL). The library provides the Ocean suite of 20+ environments and seamless integration with Gymnasium, PettingZoo, and specialized RL frameworks. ## When to Use This Skill Use this skill when: - **Training RL agents** with PPO on any environment (single or multi-agent) - **Creating custom environments** using the PufferEnv API - **Optimizing performance** for parallel environment simulation (vectorization) - **Integrating existing environments** from Gymnasium, PettingZoo, Atari, Procgen, etc. - **Developing policies** with CNN, LSTM, or custom architectures - **Scaling RL** to millions of steps per second for faster experimentation - **Multi-agent RL** with native multi-agent environment support ## Core Capabilities ### 1. High-Performance Training (PuffeRL) PuffeRL is PufferLib's optimized PPO+LSTM training algorithm achieving 1M-4M steps/second. **Quick start training:** ```bash # CLI training puffer train procgen-coinrun --train.device cuda --train.learning-rate 3e-4 # Distributed training torchrun --nproc_per_node=4 train.py ``` **Python training loop:** ```python import pufferlib from pufferlib import PuffeRL # Create v