Shardy

Shardy es un sistema de partición de tensores basado en MLIR para todos los dialectos. Se creó a partir de la colaboración de los equipos de GSPMD y PartIR, y en él se incorpora lo mejor de ambos sistemas, así como la experiencia compartida de los equipos y los usuarios.

Cómo comenzar

Si eres desarrollador de modelos y quieres usar Shardy, consulta Cómo comenzar a usar Shardy en JAX.

Para comenzar a obtener más información sobre los componentes que conforman Shardy, consulta la página Descripción general de Shardy.

Canales de comunicación

  • Los problemas y las solicitudes de funciones se pueden enviar en el repositorio openxla/shardy.
  • Puedes hacer preguntas sobre Shardy en el canal de OpenXLA en Discord.

Estado

Shardy está en desarrollo. El dialecto principal y las vinculaciones de C son completamente funcionales. Las vinculaciones de Python están en desarrollo.