Shardy — это тензорная система секционирования на основе MLIR для всех диалектов. Созданная в результате сотрудничества команд GSPMD и PartIR , она объединяет лучшее из обеих систем, а также общий опыт команд и пользователей.
Начиная
Для разработчиков моделей, желающих использовать Shardy, ознакомьтесь с разделом «Начало работы с Shardy в JAX» .
Чтобы узнать больше о компонентах, составляющих Shardy, посетите страницу обзора Shardy .
Каналы связи
- Проблемы и пожелания по функциям можно отправлять в репозиторий openxla/shardy .
- Вопросы, касающиеся Шарди, можно задать на OpenXLA Discord .
Статус
Шарди находится в стадии разработки. Основной диалект и привязки C полностью функциональны. Привязки Python находятся в стадии разработки.