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-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
benchmarks
History
Yash Katariya
e08bc27bf0
Speed up GDA initialization by making local_shards lazy and hiding checks behind
config.jax_enable_checks
flag.
...
PiperOrigin-RevId: 438115859
2022-03-29 13:51:11 -07:00
..
api_benchmark.py
Add small and big matmul to api_benchmarks.
2022-03-17 14:51:17 -07:00
benchmark.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
gda_benchmark.py
Speed up GDA initialization by making local_shards lazy and hiding checks behind
config.jax_enable_checks
flag.
2022-03-29 13:51:11 -07:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00
