Skip to main content
costa-huang
Projects
cleanRL
Reports
Pytorch JIT vs JAX JIT

Pytorch JIT vs JAX JIT

Costa
Created on January 5|Last edited on January 5

without GPU:


20406080100Time (seconds)100200300400500
100k200k300k400kglobal_step100200300400500
100k200k300k400kglobal_step10002000300040005000
Run set
4


With GPU


Run set
3


Created with ❤️ on Weights & Biases.

https://wandb.ai/costa-huang/cleanRL/reports/Pytorch-JIT-vs-JAX-JIT--VmlldzozMjY4MjQ2
Made with Weights & Biases. Sign up or log in to create reports like this one.