google/paxml

Pax là framework học máy dựa trên Jax để huấn luyện các mô hình quy mô lớn. Pax cho phép thử nghiệm và song song hóa hoàn toàn, đồng thời đạt tỷ lệ sử dụng flop mô hình hàng đầu trong ngành.