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-19 13:26:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
nn
History
Sergei Lebedev
c576d328bd
Added
lax.axis_size
and switched all existing usage of
psum(1, ...)
to it
...
PiperOrigin-RevId: 748604842
2025-04-17 02:22:25 -07:00
..
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
functions.py
Added
lax.axis_size
and switched all existing usage of
psum(1, ...)
to it
2025-04-17 02:22:25 -07:00
initializers.py
Merge pull request
#26765
from Qwlouse:patch-1
2025-03-04 08:30:45 -08:00
