Fix mesh construction by remapping device IDs. - Githubissues

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users

https://orbax.readthedocs.io/

Apache License 2.0

307 stars 36 forks source link

Fix mesh construction by remapping device IDs. #1307

Closed copybara-service[bot] closed 3 weeks ago

copybara-service[bot] commented 3 weeks ago

Fix mesh construction by remapping device IDs.