Shardy est un système de partitionnement de tenseur basé sur MLIR pour tous les dialectes. Conçue grâce à la collaboration des équipes GSPMD et PartIR, elle intègre le meilleur des deux systèmes, ainsi que l'expérience partagée des équipes et des utilisateurs.
Premiers pas
Pour les développeurs de modèles qui souhaitent utiliser Shardy, consultez la section Premiers pas avec Shardy dans JAX.
Pour en savoir plus sur les composants de Shardy, consultez la page Présentation de Shardy.
Canaux de communication
- Vous pouvez signaler des problèmes et envoyer des demandes de fonctionnalités dans le dépôt openxla/shardy.
- Vous pouvez poser des questions sur Shardy sur le Discord OpenXLA.
État
Shardy est en cours de développement. Le dialecte de base et les liaisons C sont entièrement fonctionnels. Les liaisons Python sont en cours de développement.