Jake VanderPlas
|
2a7e336888
|
Explicitly re-export all names from jax.debug
|
2023-04-07 07:24:59 -07:00 |
|
Sharad Vikram
|
805073f36a
|
Add inspect_array_sharding, enabling looking at shardings in pjit-ted functions
PiperOrigin-RevId: 476237731
|
2022-09-22 17:36:56 -07:00 |
|
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 |
|
Sharad Vikram
|
2d8b228706
|
Add function to visualize Shardings
|
2022-09-19 13:27:08 -07:00 |
|
Sharad Vikram
|
4386a0f909
|
Add debugging tools under jax.debug and documentation
Co-authored-by: Matthew Johnson <mattjj@google.com>
Co-authored-by: Lena Martens <lenamartens@google.com>
|
2022-07-28 20:07:26 -07:00 |
|