Estados Unidos
CONCEPT SOLUTIONS LLC
NVIDIA
Senior Manager, Software Engineering - JAX
Estados Unidos
Santa Clara, California, Estados Unidos
Traducción automática. Consulta el original.
Sobre la oportunidad
JAX es el framework elegido para la investigación de IA más ambiciosa, y los centros de datos en los que se ejecuta evolucionan rápidamente. Los superordenadores de NVIDIA se están volviendo más heterogéneos, combinando GPUs, CPUs y LPUs, y JAX necesita evolucionar para funcionar bien en todos ellos. NVIDIA busca un Senior Engineering Manager para definir e impulsar la estrategia de JAX de NVIDIA, coordinando múltiples equipos y flujos de trabajo para garantizar que JAX ofrezca el máximo rendimiento en toda la plataforma, desde cargas de trabajo de un solo acelerador hasta cientos de miles de dispositivos en los superordenadores más grandes jamás construidos.
Una parte fundamental de este puesto es garantizar que JAX esté listo en arquitecturas de próxima generación, incluidas Rubin Ultra, LPUs y las CPUs que las impulsan, desde el primer día. Hacerlo bien requiere mantenerse al día con la dirección de la IA. Esto significa dar soporte a las necesidades emergentes en entrenamiento, post-entrenamiento, inferencia y robótica, conectando las nuevas capacidades de hardware con las tendencias de IA. Únete a nosotros para construir el equipo y la tecnología detrás de los próximos avances en IA.
¿Qué harás?
- Impulsar la estrategia de contribución de ingeniería en todo el ecosistema de JAX, incluidos el núcleo de JAX, XLA, Shardy, MPMD y bases de código a nivel de aplicación como MaxText, priorizando dónde las contribuciones de NVIDIA crean el máximo impacto en el rendimiento, la adopción y la influencia del ecosistema.
- Fomentar el trabajo en equipo a través de los límites organizacionales para dar forma a la plataforma de IA de NVIDIA en equipos de compiladores (XLA, TileIR, CUDA), tiempo de ejecución, redes y hardware.
- Construir asociaciones profundas con proyectos clave de código abierto, especialmente el equipo de JAX de Google, apoyando a nuestros socios mientras representamos los intereses de NVIDIA en la configuración de la dirección del proyecto.
- Diseñar procesos y definir criterios de éxito claros para mantener a los equipos alineados con los resultados.
- Construir planes de personal, revisar la capacidad del equipo y tomar decisiones estratégicas de contratación para mantenerse a la vanguardia de las necesidades actuales y futuras.
- Liderar, hacer crecer y mentorizar una organización de ingeniería de alto rendimiento, desarrollando líderes técnicos y creando espacio para que su equipo experimente, prototipe y construya sistemas innovadores que mejoren productos y procesos.
¿Qué necesitamos ver?
- Máster o Doctorado en Ciencias de la Computación, Ingeniería Informática o un campo relacionado (o experiencia equivalente).
- 10+ años de experiencia total en ingeniería de software, con 4+ años en gestión de ingeniería o liderazgo técnico.
- Sólido conocimiento técnico en software de sistemas, compiladores, computación de alto rendimiento o frameworks de IA.
- Experiencia demostrada liderando estrategias y proyectos a nivel de dominio que abarcan límites organizacionales, construyendo consenso, influyendo sin autoridad directa y adaptando la comunicación a la audiencia, desde ingenieros hasta ejecutivos.
- Historial de propiedad de resultados en múltiples proyectos y equipos, equilibrando compensaciones e impulsando el progreso con información incompleta.
- Experiencia en la identificación de brechas en el ciclo de desarrollo y la definición de flujos de trabajo entre equipos y organizaciones para mejorar la eficiencia.
- Experiencia liderando equipos que contribuyen a proyectos a gran escala con colaboradores externos.
¿Formas de destacar entre la multitud?
- Experiencia práctica con JAX y sus sistemas como XLA, HLO, PJRT, Shardy o el sistema de trazado y transformación de JAX.
- Experiencia liderando equipos que trabajan en infraestructura de compilador o tiempo de ejecución para frameworks de deep learning (JAX, PyTorch, TensorFlow o equivalente).
- Familiaridad con los componentes de la plataforma de IA de NVIDIA: programación de GPU, CUDA, cuDNN, NCCL, TensorRT, Transformer Engine y objetivos de cómputo heterogéneo (GPU, CPU, LPU).
- Experiencia impulsando esfuerzos de optimización de rendimiento para cargas de trabajo de entrenamiento de IA distribuida a gran escala, post-entrenamiento (RLHF, alineación), inferencia o robótica (multi-nodo, multi-GPU).
- Experiencia impulsando la puesta en marcha de frameworks o software en nuevas arquitecturas de hardware, coordinando entre equipos de silicio, controladores, compiladores y frameworks.
Tu salario base se determinará en función de tu ubicación, experiencia y el salario de los empleados en puestos similares. El rango salarial base es de 272,000 USD - 431,250 USD.
También serás elegible para acciones y beneficios en nvidia.com/en-us/benefits/.
Las solicitudes para este puesto se aceptarán al menos hasta el 7 de agosto de 2026.
Esta oferta corresponde a una vacante existente.
NVIDIA utiliza herramientas de IA en sus procesos de contratación.
NVIDIA se compromete a fomentar un entorno de trabajo inclusivo y se enorgullece de ser un empleador que ofrece igualdad de oportunidades. Dado que valoramos enormemente la diversidad en nuestros empleados actuales y futuros, no discriminamos (incluso en nuestras prácticas de contratación y promoción) por motivos de raza, religión, color, origen nacional, género, expresión de género, orientación sexual, edad, estado civil, condición de veterano, condición de discapacidad o cualquier otra característica protegida por la ley.
Más oportunidades con un solo perfil
¿Quieres optar a más puestos sin repetir todo el proceso? Crea tu perfil para que te encuentren las empresas que usan Nort.
Crear mi perfil