DTensor policy v2 refactor #1589

Closed

Assignees

Labels

opened

POC: #1542

Merge plan: Break down POC into smaller PRs that will be stacked as follows:

1. feat: refactor init of dtensor policy v2 #1709 Refactors init
2. feat: refactor common data utilities of dtensor policy v2 #1710 Refactors data processing
3. feat: refactor train utilities for dtensor policy v2 #1757 Refactors model forward, forward_backward, etc

Metadata

Assignees

hemildesai

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests