Distributed Training Support #3

Open

Labels

opened

on Sep 16, 2026

Currently we only test with training on a single node (either PJRT or IFRT). In theory it shouldn't be a big deal to extend beyond a single node, but I would have to look into handling world size, rank, metrics etc.

to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests