gebob19 / rl_with_jax Star 15 Code Issues Pull requests clear single-file JAX implementations of common RL algorithms reinforcement-learning qlearning ant pong dqn gym reinforce ddpg pendulum bipedal trpo ppo a2c jax td3 cheeta naturalgrad secondorder Updated Sep 5, 2021 Python