Matthew Johnson 42dd7cac43 simplify slicing jaxprs a little ...

Co-authored-by: Sharad Vikram <sharad.vikram@gmail.com>

2022-08-12 19:52:02 -07:00

..

api_benchmark.py

simplify slicing jaxprs a little

2022-08-12 19:52:02 -07:00

benchmark.py

feat: refactor code using pyupgrade

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

gda_benchmark.py

Fix the gda_xla_sharding_match benchmark which was regressing. This was happening because that function was executed from top to bottom a couple of times and each time a new mesh object was created violating the already created cache which doesn't happen in real life.

2022-07-27 23:08:55 -07:00

pmap_benchmark.py

Revert import path changes to examples/ and benchmarks/

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