[Feature][AutoDeploy] TP sharding of Mamba layers #8201
Description
🚀 The feature, motivation and pitch
Tensor parallel sharding of Mamba layers. Required for Nemotron models
Alternatives
No response
Additional context
No response
Before submitting a new issue...
- Make sure you already searched for relevant issues, and checked the documentation and examples for answers to frequently asked questions.
Metadata
Metadata
Assignees
Labels
Type
Projects
Status