xla
35789406 - [torchax][distributed] Use local device for default jax device (#8667)

Commit

273 days ago

[torchax][distributed] Use local device for default jax device (#8667) Co-authored-by: Siyuan Liu <lsiyuan@google.coim>

References

Author

lsy323

Parents