This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-17 04:16:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
benchmarks
History
Yash Katariya
bfbdfa87e7
Add a warmup loop to pmap_simple_8_devices_100_args benchmark so as to not measure the compile time.
...
PiperOrigin-RevId: 401402336
2021-10-06 19:51:35 -07:00
..
api_benchmark.py
Add a warmup loop to pmap_simple_8_devices_100_args benchmark so as to not measure the compile time.
2021-10-06 19:51:35 -07:00
benchmark.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00
