Rl Python Packages

k8s-agent-sandbox

agent-sandbox enables easy management of isolated, stateful, singleton workloads, ideal for use cases like AI agent runtimes and reinforcement learning (RL).

tensorlake

Tensorlake is a serverless runtime for sandboxes and deploying background agentic applications

gem-llm

A Gym for Agentic LLMs

sb3-contrib

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

judgeval

The Continuous-Improvement Stack for Agents. Our environment data and evals power agent improvement and monitoring.

torchrl

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

neptune

📘 The experiment tracker for foundation model training

torchrl-nightly

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

hud-python

RL environments + evals for AI agents. Define once, train anything.

tianshou

An elegant PyTorch deep reinforcement learning library.

dopamine-rl

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

neptune-client

📘 The experiment tracker for foundation model training

hud

RL environments + evals for AI agents. Define once, train anything.

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

meta-agents-research-environments

Meta Agents Research Environments is a comprehensive platform designed to evaluate AI agents in dynamic, realistic scenarios. Unlike static benchmarks, this platform introduces evolving environments where agents must adapt their strategies as new information becomes available, mirroring real-world challenges.

torchstudio

Deep Learning Experiment

navix

Accelerated minigrid environments with JAX

flashbax

⚡ Flashbax: Accelerated Replay Buffers in JAX

twistypuzzle

A Simulator of Twisty Puzzles

rl-zoo3

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

awex

A high-performance RL training-inference weight synchronization framework, designed to enable second-level parameter updates from training to inference in RL workflows

multi-agent-rlenv

Strongly typed reinforcement learning environment framework

id-echo

A very fast distributed replay buffer for reinforcement learning

dsse

The Drone Swarm Search project provides an environment for SAR missions built on PettingZoo, where agents, represented by drones, are tasked with locating targets identified as shipwrecked individuals.