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
Qiao Zhang 850bd66242 [JAX] Prune unused inputs in jit. 
- Python part based on: https://github.com/google/jax/pull/6567
- Added cpp_jit path to handle pruned args

PiperOrigin-RevId: 371743277
2021-05-03 11:41:29 -07:00
..
api_benchmark.py
[JAX] Prune unused inputs in jit.
2021-05-03 11:41:29 -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