Peter Hawkins 98b75cf27b Prune accidental exports from jax.interpreters.pxla. ...

These imports do not appear to have users outside JAX itself.

PiperOrigin-RevId: 507835295

2023-02-07 11:16:42 -08:00

..

api_benchmark.py

Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

2023-02-03 14:28:45 -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

Prune accidental exports from jax.interpreters.pxla.

2023-02-07 11:16:42 -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