rocm_jax/benchmarks at 504b3c1b2543a333a56277f682475f895c4477c9 - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00

History

jax authors 504b3c1b25 roll forward with the fix: Make params arg in Compiled.call() position-only so that it does not conflict with the keyword args.

PiperOrigin-RevId: 481666211

2022-10-17 09:50:55 -07:00

..

api_benchmark.py

roll forward with the fix: Make params arg in Compiled.call() position-only so that it does not conflict with the keyword args.

2022-10-17 09:50:55 -07: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

Move array.py and sharding.py from experimental/ to _src/.

2022-09-27 10:06:52 -07: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