Reinforcement learning with Jax Simple reinforcement learning examples with JAX and Haiku. The aim of this repository is to provide simple examples on how to develop Deep Reinforcement learning using JAX. References JAX - Composable Transformations of Python+NumPy programs 2018