mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 06:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/benchmarks
History
Peter Hawkins cdd36b1113 Improve API benchmarks. 
Add benchmarks for different dispatch arg arities.

Add more blocking before and after benchmark loops that don't otherwise block.
2021-03-03 20:50:45 -05:00
..
api_benchmark.py
Improve API benchmarks.
2021-03-03 20:50:45 -05: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