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

1 Commits

Author SHA1 Message Date
Peter Hawkins
e7e5140dc9 Move implementation of jax.flatten_util to jax._src.flatten_util. Add a jax.flatten_util shim. 
Change as part of cleaning up the jax.* namespace.

PiperOrigin-RevId: 395551093
 2021-09-08 13:54:25 -07:00