The Wayback Machine - http://web.archive.org/web/20200917071414/https://github.com/henry-prior/jax-rl
Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

jax-rl

Core Deep Reinforcement Learning algorithms using JAX for improved performance relative to PyTorch and TensorFlow. Control tasks rely on the DeepMind Control Suite; see repo for further setup if you don't have MuJoCo configured.

Current implementations

  • TD3
  • SAC
  • MPO
  • PPO
  • A2C/A3C
  • ACKTR

Run

To test each algorithm on cartpole swingup:

python main_dm_control.py --max_timestep 100000
python main_dm_control.py --policy SAC --max_timesteps 100000
You can’t perform that action at this time.