3 ms·
> For simple stuff, we can compare JAX to PyTorch on a 4090, and JAX seems faster by 10-50%. It's way way way faster on TPU. What benchmarks are you looking at
by chillee 4y ago
> For simple stuff, we can compare JAX to PyTorch on a 4090, and JAX seems faster by 10-50%. It's way way way faster on TPU.
What benchmarks are you looking at here?
- VHRanger 4y agoIndividual user benchmarks on github and 2-3 blogs. Not super scientific, Im sorry to say, but methodologically benchmarking this is really hard.