Description
Provides the ROCm-enabled PyTorch build for Python tensor and neural-network workloads. It helps supported AMD GPUs accelerate model training, inference, and scientific computation.
ROCm workloads can be resource-heavy and hardware-specific. Check GPU support, drivers, memory, and runtime compatibility before using it for production jobs.