Tensors and Dynamic neural networks in Python with strong GPU acceleration. To use PyTorch with latest ROCm releases, see: https://rocm.docs.amd.com/en/latest/rocm-for-ai/pytorch.html - View it on GitHub