AI-Hypercomputer/torchprime

torchprime is a reference model implementation for PyTorch on TPU.

语言：Python · GitHub 星标：49 · Fork：16

在 GitHub 上查看原仓库