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-18 21:06:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
nn
History
jax authors
872e6c0ec4
Merge pull request
#25766
from carlosgmartin:nn_initializers_variance_scaling_mode_fan_geo_avg
...
PiperOrigin-RevId: 721928532
2025-01-31 15:41:50 -08: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
Don't allow users to query
tracer.sharding
even under sharding in types mode.
2025-01-20 15:12:47 -08:00
initializers.py
Add mode='fan_geo_avg' to nn.initializers.variance_scaling.
2025-01-31 17:52:22 -05:00
