TorchTPU

Hugging Face支持TorchTPU原生PyTorch后端

Hugging Face宣布TorchTPU为PyTorch带来原生TPU后端，Transformers、Diffusers和Kernels可直接在TPU上运行，只需将模型发送到“tpu”。该后端包含适配TPU的Pallas内核和分布式训练支持，此前所需的Optimum TPU不再必需。