mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 20:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/benchmarks
History
Jeppe Klitgaard 17de89b16a feat: refactor code using pyupgrade 
This PR upgrades legacy Python code to 3.7+ code using pyupgrade:
```sh
pyupgrade --py37-plus --keep-runtime-typing **.py
```

a
2022-05-17 22:14:05 +01:00
..
api_benchmark.py
Add small and big matmul to api_benchmarks.
2022-03-17 14:51:17 -07:00
benchmark.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
gda_benchmark.py
Move PartitionSpec from sharded_jit.py to pxla.py. The public endpoint is via jax.experimental so that should be used (no changes to the public endpoint).
2022-04-06 15:19:19 -07:00
pmap_benchmark.py
Revert import path changes to examples/ and benchmarks/
2021-01-20 17:35:55 -08:00