Matthew Johnson 148173630f add an optional fastpath for api_util.shaped_abstractify ...

also add a benchmark for it, 8.7ms -> 0.2ms on my machine

Co-authored-by: Yash Katariya <yashkatariya@google.com>

2022-07-27 15:14:37 -07:00

..

api_benchmark.py

add an optional fastpath for api_util.shaped_abstractify

2022-07-27 15:14:37 -07:00

benchmark.py

feat: refactor code using pyupgrade

2022-05-17 22:14:05 +01:00

gda_benchmark.py

OpSharding doesn't have __eq__ defined on it. Don't check sharding equality using opsharding until it does support that.

2022-07-20 15:03:39 -07:00

pmap_benchmark.py

Revert import path changes to examples/ and benchmarks/

2021-01-20 17:35:55 -08:00