stable-baselines3

Train single-agent RL algorithms like PPO, SAC, and DQN in Gymnasium environments.

Updated Jun 30, 2026
One-click install
npx skills add https://github.com/SunS1eep1ng/codex-dotfiles --skill stable-baselines3-suns1eep1ng
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: stable-baselines3
Source: https://github.com/SunS1eep1ng/codex-dotfiles/tree/main/skills/stable-baselines3
Command: npx skills add https://github.com/SunS1eep1ng/codex-dotfiles --skill stable-baselines3-suns1eep1ng

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires stable_baselines3, gymnasium, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill streamlines reinforcement learning experiments with pre-trained PyTorch-based models for PPO, SAC, DQN, TD3, A2C, and other state-of-the-art RL algorithms. It facilitates single-agent training with Gymnasium environments, providing an integrated experience from environment design to agent evaluation.

Core Features & Use Cases

  • RL Algorithms: Provides stable implementations of top-tier reinforcement learning algorithms.
  • PyTorch Integration: Uses PyTorch for ease of implementation and deployment.
  • Single-Agent Training: Best for single-agent RL with Gymnasium environments.
  • Use Case: Use this Skill to train an RL agent for a complex task like navigating a robot in a 3D environment or playing a game like Atari.

Quick Start

Run the 'train_rl_agent.py' script in the 'scripts' directory to start training a PPO agent on a Gymnasium environment.

Frequently Asked Questions about stable-baselines3

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I train a reinforcement learning agent with PyTorch and Gymnasium?

To train a reinforcement learning agent with PyTorch and Gymnasium, you can use the provided script to run pre-trained algorithms like PPO, SAC, or DQN using a scikit-learn-like API for fast single-agent training and evaluation.

What state-of-the-art RL algorithms are available for single-agent training?

Available state-of-the-art RL algorithms for single-agent training include PPO, SAC, DQN, TD3, and A2C. These robust PyTorch-based implementations integrate directly with Gymnasium environments for seamless agent development.

Can I use stable-baselines3 to train an agent for Atari games or 3D robot navigation?

Yes, you can use stable-baselines3 to train an RL agent for complex single-agent tasks like navigating a robot in a 3D environment or playing Atari games, utilizing its integrated PyTorch and Gymnasium environment support.

Does this Skill require a specific API style to implement reinforcement learning models?

This Skill uses a scikit-learn-like API to implement reinforcement learning models. This design provides an integrated experience from environment design to agent evaluation, streamlining the PyTorch-based training process.

What's the best way to evaluate single-agent RL models in Gymnasium environments?

The best way to evaluate single-agent RL models in Gymnasium environments is to use this Skill's integrated evaluation features alongside robust algorithms like PPO and SAC, ensuring accurate performance assessment after training.