Principal Product Manager, AI Frameworks

En NVIDIA, nos encontramos con clientes donde están en su viaje de IA en nuestras GPUs - esto significa que construimos lo mejor en los marcos de clase en OSS y apoyamos un ecosistema robusto de otros marcos OSS. El equipo de compiladores de PyTorch de NVIDIA construye y se eleva a la pila que se encuentra entre el hardware PyTorch y NVIDIA, que abarca la antorcha. compilar, surgiendo sustratos de compiladores, y la infraestructura de optimización de agente nativo que se está construyendo para la próxima era de computación acelerada. Este papel creará y dirigirá la estrategia de producto en todo el arco. Implica el envío de las últimas características de hardware en la antorcha. compilar hoy. También incluye desarrollar la representación compartida canónica en el compilador y ecosistema de tiempo de ejecución de NVIDIA. Además, se centra en la elaboración de cómo los agentes participarán en el trabajo de aprendizaje profundo en el futuro. Estamos buscando a alguien que entienda a los compiladores y pueda operar en la intersección de la arquitectura de sistemas, la ingeniería marco y la estrategia de producto orientada al cliente trabajando directamente con el liderazgo de ingeniería y los clientes externos más sofisticados de NVIDIA, incluyendo laboratorios de modelos fronterizos, inferencia/entrenamiento y equipos de marco RL como vLLM, SGLang, torchtitan, megatron-core y programas de hardware co-design. Como gerentes de productos de NVIDIA, nos asociamos con líderes de NVIDIA para definir una estrategia clara de productos y equipos de marketing para construir planes de ir al mercado. La organización Product Management en NVIDIA es un grupo flexible, fuerte e impactante que se centra en permitir el aprendizaje profundo en todos los casos de uso de GPU y proporcionar grandes productos para nuestros usuarios. Buscamos a un individuo con una rara mezcla de habilidades de producto, profundidad técnica y pasión para unirse a nuestro equipo. ¿Eso suena familiar? ¡Si es así, nos encantaría saber de ti! Lo que harás: Propietario de la estrategia para la cartera de compiladores PyTorch de NVIDIA, incluyendo: • torch.compile — mantener y evolucionar la trayectoria de PyTorch de NVIDIA, impulsar el soporte HW, resolver los problemas de los clientes a través de formas dinámicas, el rendimiento del kernel y compilar sobrecarga • definir la hoja de ruta e ir al mercado para la capa de representación compartida de NVIDIA a través de marcos, compiladores, bibliotecas del núcleo y tiempos de ejecución • Flujos de trabajo de compiladores nativos de agentes - configurar la visión del producto para cómo los sistemas de agente participarán en tareas de optimización • Funciones de optimización entre canales: construyen requisitos de producto para capacidades como megakernels y aseguran que envían con valor real demostrado Estrategia de producto líder en la hoja de ruta: • Definir ahora/Siguiente/Máster prioridades en estrecha colaboración con los líderes de ingeniería • Traducir señales de ecosistema (VLLM RFCs, SGLang CUDA Graph patterns, TorchTitan GraphTrainer plans, Meta's upstream priorities) en decisiones de productos priorizadas Participar directamente con clientes y socios: • Representar NVIDIA en los foros de contribuyentes y ecosistemas de PyTorch; Definir métricas de éxito y criterios de liberación: • Establecer puertas de rendimiento y hitos de adopción • Establecer barra para lo que "probado" significa para nuevas optimizaciones antes de comprometerse a roadmap Lo que necesitamos ver: • 15 años más en gestión de productos técnicos, con 5 años de poseer un compilador, tiempo de ejecución o producto de sistemas de bajo nivel a escala • Licenciado en Ciencias de la Computación, Ingeniería Eléctrica, un campo técnico relacionado, o experiencia equivalente. • Experiencia con los productos OSS primero y estrategia de contribución de corriente • Registro de envío y adopción de conducción para productos de infraestructura orientados al desarrollo • Comprende cómo los marcos de inferencia utilizan la tecnología de compilador, donde adoptan antorcha. compilar, donde lo rodean, y por qué • Comprende cómo las nuevas características de hardware crean nuevos requisitos de compilador • Puede escribir documentos de estrategia claros y defensibles y saber cómo obtener una liberación de acceso anticipado: • Fuerte instinto para dónde concentrar la inversión vs. difundirla Formas de destacar de la multitud: • Comprensión profunda de la pila de compiladores PyTorch y cómo influir en la estrategia en este ecosistema • Ha trabajado en cómo los agentes participan en los flujos de trabajo de optimización a nivel de sistemas • Familiaridad con la infraestructura compiladora basada en MLIR y cómo mapea los primitivos del hardware NVIDIA • Puede razonar sobre las operaciones de optimización entre canales para definir la barra adecuada para "probado" • Perfiles de rendimiento del kernel de lectura cómoda y depurar cómo antorcha. compilar puede ayudar a cualquier modelo #LI-Hybrid Su salario base se determinará en función de su ubicación, experiencia y el pago de los empleados en posiciones similares. El salario base es de 240.000 USD - 379.500 USD. También será elegible para la equidad y los beneficios. Las solicitudes para este trabajo serán aceptadas al menos hasta el 27 de julio de 2026. Esta publicación es para una vacante existente. NVIDIA utiliza herramientas de IA en sus procesos de reclutamiento. NVIDIA se compromete a fomentar un entorno de trabajo inclusivo y orgulloso de ser un empleador de igualdad de oportunidades. Al valorar la diversidad en nuestros empleados actuales y futuros, no discriminamos (incluidas 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.