This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-15 19:36:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
benchmarks
History
Peter Hawkins
88cc254f2c
[JAX] Replace uses of jax.interpreters.pxla.ShardedDeviceArray with jax.Array.
...
PiperOrigin-RevId: 508463147
2023-02-09 13:39:41 -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
[JAX] Replace uses of jax.interpreters.pxla.ShardedDeviceArray with jax.Array.
2023-02-09 13:39:41 -08:00
