SheepRL is out!
Introducing SheepRL: unleashing the Ultimate RL Framework!
SheepRL is a new framework for Reinforcement Learning. We believe that RL is the future of AI, and we want to make it accessible to everyone.
Fully written in PyTorch, SheepRL is designed based on four principles:
- Ease of use: SheepRL is designed to be easy to use, so that you can focus
- Scalability: SheepRL is built on top of Lightning Fabric, which allows to easily scale your experiments to multiple GPUs and machines.
- Extensibility: SheepRL is designed to be easily extensible, so that you can easily add new algorithms and environments.
- Independence: many RL repositories tightly couple the RL algorithm with the environment, making it harder to extend them beyond the gym interface. Like other RL frameworks, SheepRL is built on top of OpenAI Gym, which provides a standard interface to interact with the environment. However, in SheepRL we developed decoupled algorithms, that are indipendent from the environment, mimicking a real production scenario.
Exisiting RL frameworks are either too complex to use, or too simple to be useful in complex settings. SheepRL is the perfect balance between the two. Inspired by CleanRL but with a more modular design, SheepRL let you focus on the core aspect of the RL algorithm of choice, abstracting away secondary aspects like buffer or models implementation. At the same time, it mantains its readability, so that you can easily understand what’s going on under the hood.
We believe in Open Source code. That’s why we release SheepRL under the Apache 2.0 license. Feel free to use it for your projects, and if you want to contribute, we are more than happy to accept your pull requests!
You can easily access the code on our GitHub repository. Simply follow the comprehensive readme guide to swiftly install and dive into the magic.
Picture this: Within a mere five minutes, you’ll have your first agent trained and ready to conquer RL challenges with grace. Try it out:
git clone https://github.com/Eclectic-Sheep/sheeprl.git
cd sheeprl
python3.10 -m venv .venv
. .venv/bin/activate
pip install .
sheeprl ppo
Get started with SheepRL today!