mirrors/rocm_jax
1
0
Fork 0
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
Peter Hawkins 452f3c55e3 Rename jax._src.sharding_utils to jax._src.op_shardings. 
Move some more op_sharding related helpers to that module.

PiperOrigin-RevId: 522343010
2023-04-06 08:32:46 -07:00
..
api_benchmark.py
Rename jax._src.sharding_utils to jax._src.op_shardings.
2023-04-06 08:32:46 -07:00
math_benchmark.py
math_benchmark: add --set_env flag
2023-03-09 13:04:12 -08:00