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
Jake VanderPlas b8296bb06c benchmarks: add compilation of sparse operations 
PiperOrigin-RevId: 405752775
2021-10-26 15:41:21 -07:00
..
api_benchmark.py
benchmarks: add compilation of sparse operations
2021-10-26 15:41:21 -07:00
benchmark.py
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00