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 20:06:05 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
benchmarks
History
Yash Katariya
d8cbb29d14
OpSharding doesn't have
__eq__
defined on it. Don't check sharding equality using opsharding until it does support that.
...
PiperOrigin-RevId: 462238497
2022-07-20 15:03:39 -07:00
..
api_benchmark.py
Avoid top-level aliases of jax.tree_util.*
2022-07-07 11:41:02 -07:00
benchmark.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
gda_benchmark.py
OpSharding doesn't have
__eq__
defined on it. Don't check sharding equality using opsharding until it does support that.
2022-07-20 15:03:39 -07:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00
