mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-22 12:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 508 Branches 263 Tags
rocm-main
Commit Graph

2 Commits

Author SHA1 Message Date
Roy Frostig
371935cc10 update README and several docs to typed RNG keys 2024-08-11 08:09:47 -07:00
jax authors
f498442daa [jax][benchmark] Added clearing caches for benchmarking compilation time in sparse JAX benchmarks 
PiperOrigin-RevId: 553179605
 2023-08-02 10:07:54 -07:00