Skip to content

[Feature Request]: Support for DTensor #1147

Open
@fuadAbdallah

Description

@fuadAbdallah

Background and Feature Description

Enable distributed and sharded training or document how this can be implemented by other means.

With the ever increasing size of models and datasets, DTensor seems to be the most flexible way to implement sharded training.

It would be great to know if you plan to implement/wrapt it.

API Definition and Usage

No response

Alternatives

Document how distributed, sharded training is possible

Risks

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requesthelp wantedExtra attention is needed

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions