High-Performance Llama 2 Training and Inference with PyTorch/XLA on Cloud TPUs

Published in PyTorch Blog, 2023