Dopamax

Dopamax  GitHub Stars for Dopamax

Reinforcement learning in pure JAX.