Shardy

Shardy là một hệ thống phân vùng tensor dựa trên MLIR cho tất cả các phương ngữ. Được xây dựng từ sự cộng tác của cả nhóm GSPMD và nhóm PartIR, công cụ này kết hợp những tính năng tốt nhất của cả hai hệ thống và kinh nghiệm chung của cả nhóm và người dùng.

Bắt đầu

Đối với các nhà phát triển mô hình muốn sử dụng Shardy, hãy xem bài viết Bắt đầu sử dụng Shardy trong JAX.

Để bắt đầu tìm hiểu thêm về các thành phần tạo nên Shardy, hãy xem trang Tổng quan về Shardy.

Kênh liên lạc

  • Bạn có thể gửi vấn đề và yêu cầu tính năng trong kho lưu trữ openxla/shardy.
  • Bạn có thể đặt câu hỏi về Shardy trên OpenXLA Discord.

Trạng thái

Shardy đang trong quá trình phát triển. Phương ngữ cốt lõi và các liên kết C hoạt động đầy đủ. Các liên kết python đang trong quá trình phát triển.