Official implementation of iterated Shared Soft Actor-Critic (iS-SAC) in JAX

User installation

We recommend using Python 3.11.5. In the folder where the code is, create a Python virtual environment, activate it, update pip and install the package and its dependencies in editable mode:

python3 -m venv env
source env/bin/activate
pip install --upgrade pip setuptools wheel
pip install -e .[dev,gpu]

Running experiments

The script launch_job/dmc/launch.sh trains an iS-SAC (K=9) agent with the SimbaV2 architecture and BatchNorm on a local machine, on the DMC task dog-walk.