Skip to content

A JAX Implementation of the Soft Actor Critic Algorithm

License

Notifications You must be signed in to change notification settings

ethanabrooks/SAC-JAX

 
 

Repository files navigation

SAC-JAX

A JAX Implementation of the Soft Actor Critic Algorithm

Requirements

Best run in docker:

docker build --tag=<your-tag> .

For requirement details consult environment.yml and Dockerfile.

Example run command

python main.py pendulum

This will select the config specified in configs/pendulum.json. Command line arguments can be used to override arguments in config.

Results

To do...

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 98.5%
  • Python 1.5%