jax authors
|
fc1e1d4a65
|
Add freshness metablock to JAX OSS docs.
PiperOrigin-RevId: 645508135
|
2024-06-21 14:50:49 -07:00 |
|
Mark Sandler
|
bab1098866
|
Fixes broken examples, and (invalid) comment for PartitionSpec
PiperOrigin-RevId: 517531823
|
2023-03-17 16:09:45 -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 |
|
Jake VanderPlas
|
ab1f3a824c
|
DOC: clarify purpose of the jax-array doc
|
2023-01-27 09:25:20 -08:00 |
|
jax authors
|
74d6c219b5
|
Merge pull request #13312 from nouiz:jax_array_doc
PiperOrigin-RevId: 502534154
|
2023-01-17 03:00:19 -08:00 |
|
Frederic Bastien
|
4d76b7c938
|
Small doc fixes.
|
2022-12-12 07:05:05 -08:00 |
|
Matthew Johnson
|
1185c895ca
|
in jax.Array notebook, polish beginning and tweak title and some wording
|
2022-12-10 22:16:54 -08:00 |
|
yashkatariya
|
70d50814b1
|
Add cross-linking for the migration guide and the parallelism with JAX
tutorial
Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>
|
2022-12-01 14:42:59 -08:00 |
|
Yash Katariya
|
1824be772e
|
Update jax_array_migration.md
|
2022-11-18 08:11:22 -08:00 |
|
Yash Katariya
|
29d75324a3
|
Add a date till which jax.Array can be disabled
|
2022-11-18 08:09:31 -08:00 |
|
Peter Hawkins
|
9f2a6acb61
|
Revert: Add deprecation warnings to DA, SDA and GDA.
This change is currently overly noisy for users.
PiperOrigin-RevId: 489455729
|
2022-11-18 06:06:13 -08:00 |
|
Yash Katariya
|
52a2428073
|
Add deprecation warnings to DA, SDA and GDA.
PiperOrigin-RevId: 489314189
|
2022-11-17 14:51:29 -08:00 |
|
Yash Katariya
|
c42bad85ef
|
Make MeshPspecSharding an alias for NamedSharding (it was the other way around before this CL).
PiperOrigin-RevId: 488473538
|
2022-11-14 14:44:00 -08:00 |
|
Yash Katariya
|
6897d37562
|
Add docstrings for jax.Array APIs make_array_from_callback and make_array_from_single_device_arrays.
PiperOrigin-RevId: 487929688
|
2022-11-11 15:21:10 -08:00 |
|
Yash Katariya
|
f0c0689a8a
|
Remove internal information
|
2022-11-10 19:09:19 -08:00 |
|
Yash Katariya
|
73935a5bd1
|
Update jax_array_migration.md
|
2022-11-10 17:23:16 -08:00 |
|
Yash Katariya
|
aa66b939f9
|
Fix the docs build
|
2022-11-10 17:08:57 -08:00 |
|
Yash Katariya
|
b49a1bda15
|
Add jax.Array migration doc to OSS
PiperOrigin-RevId: 487673643
|
2022-11-10 16:46:30 -08:00 |
|