Messing around with RL agents in jax.
Currently supports Proximal Policy Optimisation (PPO) with cartpole environment.
Requires Jax, the best way to install is from here
All other packages are specified in requirements.txt
pip install -r requirements.txt