jax_sandbox JAX messaround (to be honest, I don't know why I am doing this because DeepMind already has ACME out and Google Research has Dopamine, which is like 100x better than this kinda garbage repo).