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

..

checkify_guide.md

Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

2023-02-03 14:28:45 -08:00

flags.md

Change jaxdb->jdb and add option to force a backend

2022-07-29 12:51:27 -07:00

index.md

Make JAX Debugging and Profiling guides more visible, move Profiling to User Guides from Notes

2023-01-19 23:05:24 +00:00

print_breakpoint.md

Update docs that jax.debug is unsupported on Cloud TPUs

2023-02-01 00:12:51 +00:00