azzeddineCH/Imapal-Cartpole-Agent-in-Jax

training script for CartPole Impala agent in Jax. the code is inspired from the official Haiku repository

Jupyter Notebook