Roy Frostig
|
672a013b3a
|
docs: prefer "summary" to "tl;dr"
This was common and we typically just mean "summary."
|
2024-08-28 11:04:14 -07:00 |
|
jax authors
|
fc1e1d4a65
|
Add freshness metablock to JAX OSS docs.
PiperOrigin-RevId: 645508135
|
2024-06-21 14:50:49 -07:00 |
|
Yash Katariya
|
52a7701dda
|
Replace usage of {in|out}_axis_resources with {in|out}_shardings
PiperOrigin-RevId: 513040164
|
2023-02-28 14:29:09 -08:00 |
|
Peter Hawkins
|
428189f8fb
|
Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.
This change updates:
* {jax.experimental.maps.Mesh, jax.interpreters.pxla.Mesh} to jax.sharding.Mesh
* {jax.experimental.PartitionSpec, jax.experimental.pjit.PartitionSpec, jax.interpreters.pxla.PartitionSpec, jax.pxla.PartitionSpec} to jax.sharding.PartitionSpec
* jax.experimental.maps.NamedSharding to jax.sharding.NamedSharding.
PiperOrigin-RevId: 506994892
|
2023-02-03 14:28:45 -08:00 |
|
lenamartens
|
8a4b9d6aad
|
Fix typo in checkify guide.
|
2023-01-05 17:04:53 +00:00 |
|
lenamartens
|
3db909ee8d
|
Checkify: Add documentation for adding run-time values to error message.
|
2022-12-12 15:56:40 +00:00 |
|
lenamartens
|
0256301617
|
tweak checkify docs.
|
2022-07-29 18:15:36 +01: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 |
|