There is some research in accelerating Reinforcement Learning by implementing the simulator on the GPU using Jax. Really neat. I'm curious if this could be done with Mojo, too?

https://arxiv.org/html/2311.10090v5

https://youtu.be/Jr_nGkCG3og?si=WKS7Hbz13A0QH4nf&t=520