AMD ROCm

ROCm es la respuesta de AMD a CUDA de NVIDIA. Es un conjunto de software que permite a los programas usar los procesadores gráficos de AMD para computación general: incluye controladores, una cadena de compilación, bibliotecas de matemáticas y comunicación, y herramientas de depuración y perfilado. Gran parte es de código abierto.

Para trabajo con IA, la pieza clave es la compatibilidad con frameworks. PyTorch, TensorFlow, JAX y otros funcionan sobre ROCm, y herramientas de inferencia como vLLM y muchas bibliotecas de modelos tienen versiones para AMD. Se puede portar código CUDA con HIP, una capa de programación parecida a CUDA, y grandes despliegues en la nube y en supercomputadores usan aceleradores AMD Instinct con ROCm.

ROCm es gratuito. Históricamente ha ido por detrás de CUDA en madurez y soporte del ecosistema, así que la compatibilidad con GPU y bibliotecas concretas varía y hay que comprobarla, aunque la distancia se ha ido acortando. Entre las alternativas están NVIDIA CUDA, Intel oneAPI, Apple Metal para Mac y la computación basada en Vulkan. ROCm es adecuado para desarrolladores y organizaciones que usan hardware AMD para IA.