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
Yash Katariya 3572bb2db0 [Rollback] 
Allow uncommitted single device PyArray in C++ pjit path.

PiperOrigin-RevId: 482084898
2022-10-18 19:42:10 -07:00
..
api_benchmark.py
[Rollback]
2022-10-18 19:42:10 -07:00
benchmark.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
gda_benchmark.py
Move array.py and sharding.py from experimental/ to _src/.
2022-09-27 10:06:52 -07:00
pmap_benchmark.py
Replace jax.xla.DeviceArray private type with the new public type jax.Array.
2022-09-28 16:34:10 -07:00