openxla/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
해결하는 문제
XLA는 다양한 하드웨어에서 머신러닝 모델을 효율적으로 실행해야 하는 과제를 해결합니다. 모든 모델에 대해 하드웨어별 코드를 작성할 필요가 없도록 하여 GPU, CPU 및 특화된 ML 가속기에서 고성능 실행을 가능하게 합니다.
작동 방식
XLA는 컴파일러 역할을 수행하여 PyTorch, TensorFlow, JAX와 같은 인기 프레임워크의 모델을 가져와 대상 하드웨어 플랫폼에 맞게 최적화함으로써 고성능 실행을 보장합니다.
대상 사용자
이 프로젝트는 주로 ML 프레임워크 개발자, 컴파일러를 개발하고자 하는 XLA 기여자, 그리고 새로운 ML 프론트엔드 또는 하드웨어 백엔드에 대한 디버깅이나 지원 추가가 필요한 통합 개발자를 대상으로 합니다.
주요 특징
- PyTorch, TensorFlow, JAX를 포함한 주요 프레임워크 지원
- CPU, GPU 및 ML 가속기를 위한 크로스 플랫폼 최적화
- 오픈 소스 머신러닝 컴파일러
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트
- Dispatch