Generatived (Beta) | Cung cấp tin tức và xu hướng mới nhất về AI sáng tạo

JAX đã nổi lên như một công cụ mạnh mẽ cho các nhà đổi mới AI, cho phép đào tạo các mô hình phức tạp trên nhiều ứng dụng khác nhau. Hiệu suất và khả năng mở rộng của nó đặc biệt có lợi cho các công ty như Lightricks, vốn đang đạt được những bước tiến trong nền kinh tế sáng tạo với các mô hình tạo video hiệu suất cao. Nhóm Lightricks LTX-Video, do Yoav HaCohen và Yaki Bitterman dẫn đầu, đã phải đối mặt với những thách thức đáng kể về khả năng mở rộng quy mô với nền tảng trước đây của họ, và họ đã giải quyết vấn đề này bằng cách chuyển sang JAX.

Lightricks đang tìm cách cải thiện việc tạo nội dung bằng Generative AI) nhưng đã gặp phải trở ngại khi phát triển mô hình 13 tỷ tham số. Khung công tác trước đây của nhóm không thể đáp ứng số bước huấn luyện và khả năng mở rộng cần thiết. Một động thái chiến lược chuyển sang JAX đã mang lại những cải tiến ngay lập tức, giúp mở rộng quy mô hiệu quả và cho phép tăng 40% số bước huấn luyện mỗi ngày. Sự thay đổi này không chỉ đẩy nhanh quá trình phát triển mô hình mà còn tăng gấp đôi năng suất của nhóm.

Thành công của JAX là nhờ một hệ sinh thái toàn diện bao gồm các công cụ như Flax, Optax và Orbax, tất cả đều hoạt động liền mạch với nhau. Lightricks đã tận dụng những công cụ này để xây dựng một nền tảng mới mạnh mẽ cho hoạt động R&D hiệu quả. Công ty cũng tận dụng lõi TPU của Google để đào tạo, tận dụng tối đa thiết kế hệ thống tích hợp để đạt được hiệu suất tối ưu.

Kinh nghiệm của Lightricks với JAX đã chứng minh rõ ràng khả năng của nền tảng này trong việc hỗ trợ các nhóm vượt qua những thách thức kỹ thuật và đạt được những cải tiến hiệu suất đáng kể. Đội ngũ JAX tại Google vẫn cam kết hỗ trợ cộng đồng điện toán khoa học và khuyến khích người dùng chia sẻ kinh nghiệm của họ cũng như đóng góp vào sự phát triển của nền tảng.