mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/benchmarks
History
Jean-Baptiste Lespiau 6cb8737c1a Add a benchmark with many arguments. 
PiperOrigin-RevId: 393216026
2021-08-26 15:06:09 -07:00
..
api_benchmark.py
Add a benchmark with many arguments.
2021-08-26 15:06:09 -07:00
benchmark.py
[JAX] Move jax.util to jax._src_util.
2021-01-11 14:21:07 -08:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00