ResearchBenchmarksRandom action rollout on PokeJAXFollow500Simulation Speed (SPS)PokeJAX1.84131.17260.5389.83Mar 12, 2026Evaluation ResultsMethodMethodLinksSimulation Speed (SPS)Speedup FactorPokeJAXImplementation=JAX, JA...Implementation=JAX, JAX batch=65Kb2026.0350023,810ShowdownProcesses=1pProcesses=1p2026.0321—