TF-Agents

The reliable, modular TensorFlow library for building, training, and deploying Reinforcement Learning (RL) and Contextual Bandits agents.

TF-Agents is your streamlined solution for RL development, built on TensorFlow 2.x and Keras. The library provides well-tested, modular components: Agents, Policies, and Environments. You can accelerate iteration by leveraging pre-implemented algorithms like DQN, PPO, and SAC, or quickly build custom solutions. It supports standard environment suites (e.g., OpenAI Gym) and integrates with DeepMind's Reverb for efficient replay buffers. This structure ensures fast code development, robust testing, and scalable deployment for your RL projects.

https://github.com/tensorflow/agents