Emilio Cota 13e875f8b8 benchmarks: add math unary benchmarks ...

These will be used for benchmarking FP approximations in XLA.

PiperOrigin-RevId: 503991586

2023-01-23 08:17:16 -08:00

..

api_benchmark.py

Added a pattern-match optimisation for inplace-select.

2022-12-23 16:05:56 -08:00

benchmark.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

gda_benchmark.py

Make MeshPspecSharding an alias for NamedSharding (it was the other way around before this CL).

2022-11-14 14:44:00 -08:00

math_benchmark.py

benchmarks: add math unary benchmarks

2023-01-23 08:17:16 -08:00

pmap_benchmark.py

Replace jax.xla.DeviceArray private type with the new public type jax.Array.

2022-09-28 16:34:10 -07:00