PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Reinforcement Learning Python Packages

Python packages with the GitHub topic reinforcement-learning. Sorted by relevance, with stars and monthly downloads.
sgl-project
sglang

SGLang is a high-performance serving framework for large language models and multimodal models.

306.7M 27K 6K
ray-project
ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

53M 42K 8K
wandb
wandb

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

25.2M 11K 864
Farama-Foundation
gymnasium

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

6.3M 12K 1K
unslothai
unsloth

Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt-oss locally.

1.9M 64K 6K
unslothai
unsloth-zoo

Web UI for training and running open models like Gemma 4, Qwen3.6, DeepSeek, gpt-oss locally.

1.4M 64K 6K
pytorch
torchrl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

1.4M 3K 450
DLR-RM
stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

1.3M 13K 2K
pytorch
torchtnt

A lightweight library for PyTorch training tools and utilities

680K 2K 300
google-deepmind
dm-control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

426K 5K 746
JudgmentLabs
judgeval

The open source post-building layer for agents. Our environment data and evals power agent post-training (RL, SFT) and monitoring.

368K 1K 91
Farama-Foundation
pettingzoo

An API standard for multi-agent reinforcement learning environments, with popular reference environments and related utilities

357K 3K 484
Farama-Foundation
shimmy

An API conversion tool for popular external reinforcement learning environments

346K 208 24
Stable-Baselines-Team
sb3-contrib

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

300K 712 240
sgl-project
sglang-kernel

SGLang is a high-performance serving framework for large language models and multimodal models.

273K 27K 6K
sgl-project
sgl-kernel

SGLang is a high-performance serving framework for large language models and multimodal models.

254K 27K 6K
bulletphysics
pybullet

Bullet Physics SDK: real-time collision detection and multi-physics simulation for VR, games, visual effects, robotics, machine learning etc.

219K 14K 3K
deepmind
open-spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

193K 5K 1K
open-thought
reasoning-gym

[NeurIPS 2025 Spotlight] Reasoning Environments for Reinforcement Learning with Verifiable Rewards

168K 1K 118
ARISE-Initiative
robosuite

robosuite: A Modular Simulation Framework and Benchmark for Robot Learning

147K 2K 707
Future-House
fhlmi

Framework enabling modular interchange of language agents, environments, and optimizers

140K 133 17
Future-House
fhaviary

A language agent gym with challenging scientific tasks

120K 260 32
hud-evals
hud-python

OSS RL environment + evals toolkit

89K 248 57
Toni-SM
skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

74K 1K 141
    • Data from PyPI, GitHub, ClickHouse, and BigQuery