mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
13,114 Commits 500 Branches 263 Tags
Commit Graph

4 Commits

Author SHA1 Message Date
Peter Hawkins
ba557d5e1b Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.". 
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.

PiperOrigin-RevId: 476167538
 2022-09-22 12:27:19 -07:00
Peter Hawkins
80aec7b25f Documentation improvements. 2022-03-08 09:37:33 -05:00
Matthew Johnson
984e8d79f0 make ravel_pytree unraveler dtype-polymorphic 
fixes #7809
 2021-12-14 14:47:35 -08:00
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