mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 21:56:04 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,367 Commits 508 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Jake VanderPlas
ca39457ea9 JEX: move jax.linear_util to jax.extend.linear_util 2023-08-30 18:32:12 -07:00
Roy Frostig
a69f134cde add jax.extend.random.wrap_key_data 2023-08-26 11:39:25 -07:00
Roy Frostig
a71c0e6ecc create jax.extend.random as a copy of jax.prng 
Co-authored-by: Jake Vanderplas <jakevdp@google.com>
PiperOrigin-RevId: 559874051
 2023-08-24 14:41:56 -07:00
Roy Frostig
ca008f37e3 initiate jax.extend via docs and top-level module set-up 2023-05-15 15:47:06 -07:00