Únete a NVIDIA como Ingeniero de Compiladores de Aprendizaje Profundo

Compañía: NVIDIA

Posición: Ingeniero de Software, PyTorch - Aprendizaje Profundo

Ubicación: Abierto para Talentos Globales

¿Estás listo para innovar en la intersección de IA y la ingeniería de software? NVIDIA está buscando Ingenieros de Compiladores de Aprendizaje Profundo capacitados para crear herramientas de vanguardia que mejoren las capacidades de los ingenieros de IA en todo el mundo. Aprovecha la oportunidad de impactar en el diseño, desarrollo e implementación de IA a escala global, utilizando miles de GPU. Únete a nuestro equipo dedicado a ofrecer una excelencia perpetua en experiencias de software en el hardware líder de NVIDIA con PyTorch.

Lo que Harás:

Como miembro crucial de nuestro equipo, tu papel se extenderá más allá de los límites tradicionales, impulsando la evolución de las cargas de trabajo de entrenamiento de aprendizaje profundo distribuido a través de tecnologías de compilador avanzadas. Tus responsabilidades principales incluirán:

• Realizar análisis profundos de las cargas de trabajo de aprendizaje profundo para identificar y resolver cuellos de botella en el rendimiento, anomalías funcionales e ineficiencias del sistema.
• Vincular los problemas de rendimiento directamente a defectos del compilador o potenciales de optimización no aprovechados y formular resoluciones robustas.
• Trabajar en colaboración para refinar las transformaciones de programas o innovar nuevas tras revisiones de rendimiento exhaustivas.
• Mantenerse al día con los últimos desarrollos en compiladores de aprendizaje profundo para proponer soluciones innovadoras que mejoren la eficiencia de los marcos de aprendizaje profundo.
• Realizar pruebas y validaciones exhaustivas para asegurar un rendimiento óptimo y calidad en el entrenamiento de modelos.

Lo que Necesitamos Ver:

Ven equipado con:

• Una licenciatura, maestría o doctorado en Ciencias de la Computación o un campo técnico relacionado, o experiencia comparable.
• Proficiencia en Python y familiaridad con marcos de aprendizaje automático como PyTorch o JAX.
• Un sólido entendimiento de conceptos de compiladores, incluyendo la interpretación abstracta, representaciones de código como la forma SSA y AST, generación de código y transformaciones de programas.
• Historial comprobado en el desarrollo de proyectos de software de considerable tamaño.
• Fuertes capacidades en comunicación verbal y escrita.

Destaca con:

• Contribuciones previas a proyectos de compiladores de aprendizaje profundo de código abierto como TVM, o involucramiento en marcos de aprendizaje profundo.
• Conocimiento profundo de los interiores de PyTorch y/o JAX.
• Experiencia en sistemas distribuidos, computación paralela y programación CUDA.
• Participación activa en la comunidad de código abierto.
• Experiencia en colaboración con equipos multidisciplinarios diversos.

Listo para llevar tu carrera al frente de la innovación tecnológica? Aplica ahora para formar parte de la misión de NVIDIA de pionerizar un futuro donde la IA y los compiladores de aprendizaje profundo se unan para crear aplicaciones y experiencias sin precedentes.