Desarrollo de backend CUDA para MLX permitirá compatibilidad con GPUs Nvidia en plataformas no Apple Silicon
Un futuro cercano, MLX integrará un backend CUDA que ampliará su compatibilidad y rendimiento en plataformas con GPUs Nvidia, más allá de Apple Silicon.
Los usuarios que deseen trabajar con MLX en plataformas que no sean Macs con chip M podrán beneficiarse próximamente de una innovación importante: la implementación de un backend CUDA en el proyecto. Actualmente en desarrollo y sin versión oficial lanzada, este avance facilitará que modelos MLX se adapten fácilmente a GPUs Nvidia, acelerando su difusión y optimización en distintos entornos.
El proyecto, disponible en GitHub y desarrollado por un usuario conocido como zcbenz, busca crear un entorno donde desarrolladores puedan escribir y probar código en Mac y, posteriormente, ejecutarlo en supercomputadoras o grandes clústeres, mejorando la experiencia y eficiencia en el entrenamiento y prueba de modelos de aprendizaje automático.
Una de las principales ventajas del backend CUDA es la compatibilidad con la memoria unificada de Nvidia, que simplifica el desarrollo y la migración de código entre diferentes plataformas de hardware. Además, la sólida comunidad de Nvidia en el ámbito científico y de cálculos masivos hace que esta opción sea especialmente atractiva para investigadores y desarrolladores en grandes centros de datos.
Este avance representa un paso importante en la expansión de MLX más allá de los entornos Apple Silicon, actualmente optimizados para la interfaz gráfica Metal. La transición de Metal a CUDA permitirá que modelos optimizados para MLX aprovechen la potencia de las GPUs Nvidia, facilitando su entrenamiento, prueba y distribución en distintas plataformas.
Por ahora, el proyecto sigue en fase de desarrollo y aún no cuenta con versión oficial. Se espera que en los próximos meses se publiquen más detalles y mejoras, pero sus impulsores confían en que, una vez finalizado, fomentará una mayor accesibilidad y flexibilidad en el uso de MLX en entornos científicos y de desarrollo de inteligencia artificial.