ResearchBenchmarksPPO training on PokeJAXPPO training on PokeJAXFollow681Steps Per Second (SPS)Showdown-11.432168.334348.1527.866Mar 12, 2026Evaluation resultsMethodMethodLinksSteps Per Second (SPS)Speedup FactorShowdown2026.03681—PokeJAXImplementation=JAXImplementation=JAX2026.0315.222,320