jax authors eb875cd5dd Added a pattern-match optimisation for inplace-select. ...

PiperOrigin-RevId: 497425937

2022-12-23 16:05:56 -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

pmap_benchmark.py

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

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