pytorch-xla

PyTorch 프레임워크를 Google TPU와 같은 XLA(Accelerated Linear Algebra) 장치에서 실행할 수 있게 해주는 라이브러리이다. 모델 계산 그래프를 XLA 연산으로 변환하여 TPU의 하드웨어 가속 성능을 활용하게 한다.