O Shardy é um sistema de particionamento de tensor baseado em MLIR para todos os dialetos. Criada com a colaboração das equipes do GSPMD e do PartIR, ela incorpora o melhor dos dois sistemas e a experiência compartilhada de equipes e usuários.
Primeiros passos
Para desenvolvedores de modelos que querem usar o Shardy, consulte Primeiros passos com o Shardy no JAX.
Para saber mais sobre os componentes que compõem o Shardy, consulte a página Visão geral do Shardy.
Canais de comunicação
- Problemas e solicitações de recursos podem ser registrados no repositório openxla/shardy.
- Se você tiver dúvidas sobre o Shardy, pergunte no Discord do OpenXLA.
Status
O Shardy ainda está em desenvolvimento. O dialeto principal e as vinculações C são totalmente funcionais. As vinculações do Python estão em desenvolvimento.