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
George Necula 428377afb3
Added type annotations and removed unused imports (#2472) 
* Added type annotations and removed unused imports

* Adjusted type hints for pytype
2020-03-21 13:54:30 +01:00
..
benchmark.py
Added type annotations and removed unused imports (#2472)
2020-03-21 13:54:30 +01:00
pmap_benchmark.py
Changed to pmap_benchmark to make it runnable in Google (#2448)
2020-03-19 06:56:59 +01:00