파이토치 XLA
PyTorch 프레임워크를 Google TPU와 같은 XLA(Accelerated Linear Algebra) 장치에서 실행할 수 있게 해주는 라이브러리이다. 모델 계산 그래프를 XLA 연산으로 변환하여 TPU의 하드웨어 가속 성능을 활용하게 한다.
TPU 학습의 고통은 끝, PyTorch/XLA를 위한 초간편 프레임워크 등장
TPU 학습이 고통스러웠나요? PyTorch/XLA를 위한 초간편 프레임워크 등장
TPU 학습의 고통은 끝났다? 10B 모델까지 지원하는 easy-torch-tpu
TPU 학습의 고통은 끝, 10B 모델까지 지원하는 초간편 프레임워크