Closed copybara-service[bot] closed 6 months ago
Removed unused xmap wrappers
xmap is soft-deprecated with shard_map being the recommended alternative.
See https://jax.readthedocs.io/en/latest/notebooks/shard_map.html.
Removed unused xmap wrappers
xmap is soft-deprecated with shard_map being the recommended alternative.
See https://jax.readthedocs.io/en/latest/notebooks/shard_map.html.