garage is a toolkit for developing and evaluating reinforcement learning algorithms, and an accompanying library of state-of-the-art implementations built using that toolkit.
The toolkit provides wide range of modular tools for implementing RL algorithms, including:
- Composable neural network models
- Replay buffers
- High-performance samplers
- An expressive experiment definition interface
- Tools for reproducibility (e.g. set a global random seed which all components respect)
- Logging to many outputs, including TensorBoard
- Reliable experiment checkpointing and resuming
- Environment interfaces for many popular benchmark suites
- Supporting for running garage in diverse environments, including always up-to-date Docker containers
See the latest documentation for getting started instructions and detailed APIs.
pip install garage
The table below summarizes the algorithms available in garage.
Algorithm | Framework(s) |
---|---|
CEM | numpy |
CMA-ES | numpy |
REINFORCE (a.k.a. VPG) | PyTorch, TensorFlow |
DDPG | PyTorch, TensorFlow |
DQN | TensorFlow |
DDQN | TensorFlow |
ERWR | TensorFlow |
NPO | TensorFlow |
PPO | TensorFlow |
REPS | TensorFlow |
TD3 | TensorFlow |
TNPG | TensorFlow |
TRPO | TensorFlow |
garage supports Python 3.5+.
We currently support PyTorch and TensorFlow for implementing the neural network portions of RL algorithms, and additions of new framework support are always welcome. PyTorch modules can be found in the package garage.torch
and TensorFlow modules can be found in the package garage.tf
. Algorithms which do not require neural networks are found in the package garage.np
.
The package is available for download on PyPI, and we ensure that it installs successfully into environments defined using conda, Pipenv, and virtualenv.
All components use the popular gym.Env
interface for RL environments.
The most important feature of garage is its comprehensive automated unit test and benchmarking suite, which helps ensure that the algorithms and modules in garage maintain state-of-the-art performance as the software changes.
Our testing strategy has three pillars:
- Automation: We use continuous integration to test all modules and algorithms in garage before adding any change. The full installation and test suite is also run nightly, to detect regressions.
- Acceptance Testing: Any commit which might change the performance of an algorithm is subjected to comprehensive benchmarks on the relevant algorithms before it is merged
- Benchmarks and Monitoring: We benchmark the full suite of algorithms against their relevant benchmarks and widely-used implementations regularly, to detect regressions and improvements we may have missed.
Release | Build Status | Last date of support |
---|---|---|
v2019.02 | October 31st, 2019 |
Garage releases a new stable version approximately every 4 months, in February, June, and October. Maintenance releases have a stable API and dependency tree, and receive bug fixes and critical improvements but not new features. We currently support each release for a window of 8 months.
If you use garage for academic research, you are highly encouraged to cite the following paper on the original rllab implementation:
- Yan Duan, Xi Chen, Rein Houthooft, John Schulman, Pieter Abbeel. "Benchmarking Deep Reinforcement Learning for Continuous Control". Proceedings of the 33rd International Conference on Machine Learning (ICML), 2016.
garage is based on a predecessor project called rllab. The garage project is grateful for the contributions of the original rllab authors, and hopes to continue advancing the state of reproducibility in RL research in the same spirit.
rllab was originally developed by Rocky Duan (UC Berkeley/OpenAI), Peter Chen (UC Berkeley), Rein Houthooft (UC Berkeley/OpenAI), John Schulman (UC Berkeley/OpenAI), and Pieter Abbeel (UC Berkeley/OpenAI).