moss-rl 0.0.4

A Python library for Reinforcement Learning.

Moss: A Python library for Reinforcement Learning

Moss is a Python library for Reinforcement Learning based on jax.

Installation

To get up and running quickly just follow the steps below:

Installing from PyPI: Moss is currently hosted on PyPI, you can simply install Moss from PyPI with the following command:

pip install moss-rl

Installing from github: If you are interested in running Moss as a developer, you can do so by cloning the Moss GitHub repository and then executing following command from the main directory (where setup.py is located):

pip install -e ".[dev]"

After installation, open your python console and type

import moss
print(moss.__version__)

If no error occurs, you have successfully installed Moss.

Quick Start

This is an example of Impala to train Atari game(use envpool).

python examples/atari/impala.py --task_id Pong-v5 --learning_rate 1e-3