Ingeniero de infraestructura de aprendizaje automático

Acerca de Nebio:

Nebius lidera una nueva era en infraestructura de nube para la economía global de IA. Estamos construyendo una plataforma de nube de IA de pila completa que apoya a desarrolladores y empresas desde la capacitación de datos y modelos hasta la implementación de producción, sin el costo y la complejidad de construir una gran infraestructura interna de IA/ML.

Construido por ingenieros, para ingenieros. Desde la orquestación de GPU a gran escala hasta la optimización de la inferencia, somos dueños de los problemas difíciles en la computación, el almacenamiento, las redes y la IA aplicada.

Cotizamos en Nasdaq (NBIS) y con sede en Ámsterdam, tenemos una presencia global con centros de I+D en Europa, el Reino Unido, América del Norte e Israel. Nuestro equipo de más de 1500 personas incluye cientos de ingenieros con amplia experiencia en hardware, software e I+D de IA.

el papel

Buscamos un ingeniero de ML/AI altamente capacitado para unirse a nuestro equipo para liderar y respaldar la evaluación comparativa de plataformas GPU para cargas de trabajo de aprendizaje automático y IA. Desempeñará un papel fundamental en la evaluación del rendimiento del hardware basado en GPU para diversos marcos de aprendizaje profundo e inteligencia artificial, lo que permitirá tomar decisiones basadas en datos para la optimización de la plataforma y el desarrollo de hardware de próxima generación.

Sus responsabilidades incluirán:

• Trabajar en estrecha colaboración con equipos de desarrollo y hardware para perfilar y analizar el rendimiento de la GPU a nivel del sistema y del kernel.

• Evaluar y comparar el rendimiento de GPU en diferentes plataformas, arquitecturas y pilas de software (por ejemplo, CUDA, ROCm).

• Depurar y optimizar cargas de trabajo de ML para ejecutarlas de manera eficiente en hardware GPU, identificando y resolviendo cuellos de botella en el rendimiento.

• Realizar pruebas de aceptación para nuevos clústeres de GPU, garantizando que el hardware y el software cumplan con los requisitos de rendimiento, estabilidad y compatibilidad para las cargas de trabajo de IA.

• Realizar experimentos en diversas configuraciones de sistemas GPU para evaluar el impacto de distintas estrategias de interconexión y optimizaciones a nivel de sistema en el rendimiento y la escalabilidad.

• Desarrollar herramientas y paneles para visualizar métricas de desempeño, visualizar métricas de desempeño, cuellos de botella y tendencias.

• Contribuir a herramientas internas, marcos y mejores prácticas.

Esperamos que tengas:

• Una comprensión profunda de los fundamentos teóricos del aprendizaje automático.

• Comprensión profunda de los aspectos de rendimiento del entrenamiento e inferencia de grandes redes neuronales (datos/tensor/contexto/paralelismo experto, descarga, núcleos personalizados, características de hardware, optimizaciones de atención, procesamiento por lotes dinámico, etc.)

• Amplia experiencia con marcos modernos de aprendizaje profundo (PyTorch, JAX, Megatron-LM, Tensort-LLM)

• Buen conocimiento de la pila de GPU: CUDA, NCCL, controladores y bibliotecas relevantes

• Familiaridad con entornos en contenedores (por ejemplo, Docker, Kubernetes).

• Fuerte comunicación y capacidad para trabajar de forma independiente.

Formas de destacar entre la multitud:

• Familiaridad con los marcos de inferencia LLM modernos (vLLM, SGLang, TensorRT)

• Experiencia en Python y herramientas de creación de perfiles de rendimiento (por ejemplo, Nsight, nvprof, perf).

• Familiaridad con plataformas de aprendizaje automático en la nube como AWS, GCP y Azure ML.

• Contribuciones a herramientas de evaluación comparativa de aprendizaje automático de código abierto

Beneficios y ventajas:

• Compensación competitiva

• Crecimiento profesional y oportunidades de aprendizaje.

• Flexibilidad y propiedad

• Cultura colaborativa e innovadora

• Oportunidad de trabajar en proyectos de IA impactantes

• Entorno internacional y equipos talentosos.

¿Cómo es trabajar en Nebius?

Rápido movimiento - Pensamiento audaz - Crecimiento constante - Impacto significativo - Confianza y propiedad real - Oportunidad para dar forma al futuro de la IA

Declaración de igualdad de oportunidades:

Nebius es un empleador que ofrece igualdad de oportunidades. Estamos comprometidos a fomentar un lugar de trabajo inclusivo y diverso y a brindar igualdad de oportunidades laborales en todos los aspectos del empleo. No discriminamos por motivos de raza, color, religión, sexo (incluido el embarazo), origen nacional, ascendencia, edad, discapacidad, información genética, estado civil, condición de veterano, orientación sexual, identidad o expresión de género, o cualquier otra característica protegida por la ley aplicable.

Los solicitantes deben estar autorizados para trabajar en el país en el que presentan la solicitud y deberán presentar prueba de elegibilidad laboral como condición de contratación.

Si necesita adaptaciones durante el proceso de solicitud, háganoslo saber.

Fuente: Jobicy (https://jobicy.com/jobs/155964-ml-infrastructure-engineer)