XLA (가속 선형 대수학)는 머신러닝을 위한 오픈소스 컴파일러입니다. XLA 컴파일러는 PyTorch, TensorFlow, JAX와 같은 인기 프레임워크의 모델을 가져와 GPU, CPU, ML 가속기 등 다양한 하드웨어 플랫폼에서 고성능 실행을 위해 모델을 최적화합니다.
OpenXLA 프로젝트의 일환으로 XLA는 Alibaba, Amazon Web Services, AMD, Apple, Arm, Google, Intel, Meta, NVIDIA 등 업계를 선도하는 ML 하드웨어 및 소프트웨어 회사가 공동으로 빌드합니다.
주요 이점
어디서나 빌드: XLA는 이미 TensorFlow, PyTorch, JAX와 같은 주요 ML 프레임워크에 통합되어 있습니다.
어디서나 실행: GPU, CPU, ML 가속기 등 다양한 백엔드를 지원하며 더 많은 지원을 추가할 수 있는 플러그인 가능한 인프라를 포함합니다.
성능 극대화 및 확장: 프로덕션 테스트를 거친 최적화 패스와 모델 병렬화를 위한 자동 파티셔닝으로 모델의 성능을 최적화합니다.
복잡성 제거: MLIR의 성능을 활용하여 단일 컴파일러 도구 모음에 최고의 기능을 제공하므로 다양한 도메인별 컴파일러를 관리할 필요가 없습니다.
미래지향적: XLA는 업계를 선도하는 ML 하드웨어 및 소프트웨어 공급업체의 공동작업을 통해 빌드된 오픈소스 프로젝트로, ML 업계의 최신 기술을 사용하도록 설계되었습니다.
문서
XLA에 대해 자세히 알아보려면 왼쪽의 링크를 확인하세요. XLA 개발자가 처음이라면 XLA 아키텍처로 시작하여 참여를 읽어보세요.