mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 504 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Roy Frostig
98f790f5d5 update package/API reference docs to new-style typed PRNG keys 2024-03-07 12:40:09 -08:00
jax authors
311dc9cfde Add truncated normal initializer to jax.nn 
PiperOrigin-RevId: 563576354
 2023-09-07 16:23:42 -07:00
Jake VanderPlas
81e627d5bd DOC: make API doc titles more uniform 2023-01-18 10:59:42 -08:00
Peter Hawkins
d3d666d081 Document jax.nn.initializers. 2022-03-07 17:26:04 -05:00
James Bradbury
cc49d8b325 add docs for jax.nn 2019-08-29 18:15:36 -07:00