What problem does it solve?
This skill helps you set up fast reinforcement learning pipelines so you can train agents efficiently instead of spending time on slow environment simulation and inefficient training scaffolding.
Core Features & Use Cases
- High-performance PPO training (PuffeRL): Use vectorized environments and an optimized PPO+LSTM training flow to reach millions of steps per second.
- Custom environment development (PufferEnv): Implement Gymnasium-/PettingZoo-style tasks using the PufferEnv API with observation/action space helpers and templates.
- Vectorization & performance optimization: Scale parallel simulation using PufferLib’s worker/buffer patterns and tune throughput (num_envs, num_workers, envs_per_worker), including multi-agent setups.
- Policy development patterns: Build PyTorch policies (MLP/CNN/LSTM) and integrate training-ready actor/critic heads, including recurrence-friendly design.
- Framework integration: Emulate and integrate environments from Gymnasium and PettingZoo so you can train on existing benchmarks with minimal friction.
Quick Start
Use the pufferlib skill to build and run a PPO training workflow for an environment like Procgen Coinrun, then optimize throughput by tuning vectorization parameters and using an LSTM-capable policy when needed.