Hacker News Front Page · ilreb ·
google-pytorch/torch_tpu
TorchTPU ofrece un backend de PyTorch para ejecutar operaciones y modelos en las TPU de Google, con kernels ATen personalizados, una caché de compilación para cálculos XLA e integración con `torch.compile()`. También incluye soporte para entrenamiento distribuido, ejemplos de modelos y utilidades de depuración y evaluación de rendimiento. La documentación describe la instalación mediante pip o la compilación de wheels con Bazel, además de la configuración de herramientas para desarrollar en C++.
Contribute to google-pytorch/torch_tpu development by creating an account on GitHub.