Senior Math Libraries Engineers - Python APIs

¿Eres un ingeniero de software experto y autodidacta apasionado por el diseño y desarrollo de APIs de Python para bibliotecas matemáticas? NVIDIA está buscando individuos talentosos como tú para unirse a nuestro equipo dinámico, ayudando a avanzar las capacidades de nuestras bibliotecas numéricas aceleradas por GPU de alto rendimiento para frameworks basados en Python.

En la última década, Python se ha convertido en el lenguaje de programación preferido por los profesionales en IA, ciencia de datos y computación de alto rendimiento (HPC), gracias a potentes frameworks como TensorFlow y PyTorch. NVIDIA ha desempeñado un papel fundamental en esta transformación al proporcionar frameworks de aprendizaje profundo acelerados por GPU que ofrecen interfaces de programación de alto nivel y eficientes. Nuestras implementaciones optimizadas permiten a los desarrolladores centrarse en las aplicaciones mientras aprovechan la aceleración de GPU de vanguardia.

A medida que nuestras bibliotecas matemáticas continúan creciendo y extendiéndose a nuevos dominios, nuestro objetivo es hacer que estos avances sean accesibles para todos los desarrolladores en el ecosistema de Python.

Lo que harás

• Diseñar y desarrollar APIs de Python para bibliotecas matemáticas (por ejemplo, álgebra lineal)
• Prototipar integraciones de las APIs desarrolladas en frameworks específicos
• Ajuste de rendimiento, optimización y evaluación comparativa de software numérico en diversas arquitecturas de CPU y GPU
• Proporcionar liderazgo técnico y tutoría a ingenieros de biblioteca
• Colaborar estrechamente con la gestión de productos y otros socios internos y externos para comprender los requisitos de características y rendimiento
• Identificar oportunidades para mejorar el rendimiento del software y la experiencia del usuario

Lo que necesitamos ver

• Doctorado o Máster en Ciencias de la Computación, Matemáticas Aplicadas o un campo relacionado de ciencia o ingeniería (o experiencia equivalente)
• Excelentes habilidades en programación Python y C++ y diseño de software, incluyendo diseño de pruebas funcionales y de rendimiento
• Experiencia desarrollando APIs de Python y código Pythonic
• Familiaridad con uno o más de Numpy, CuPy, SciPy, JAX, PyTorch, TensorFlow
• Más de 6 años de experiencia desarrollando, depurando y optimizando aplicaciones de alto rendimiento en plataformas de computación paralela
• Sólida comprensión de métodos y cálculos numéricos fundamentales en ciencia, ingeniería o aprendizaje profundo
• Experiencia comprobada liderando y dirigiendo proyectos de desarrollo de software en un entorno ágil
• Solidas habilidades de colaboración, comunicación y documentación

Maneras de destacarse

• Experiencia en programación paralela con GPUs (CUDA o OpenCL), multi-threading, o MPI
• Buen conocimiento de la arquitectura de hardware de CPU y/o GPU

¿Por qué NVIDIA?

Considerado uno de los empleadores más deseables del mundo tecnológico, NVIDIA emplea a algunos de los talentos más innovadores y trabajadores a nivel global. ¡Si eres creativo y tomas la iniciativa, queremos saber de ti!

El rango salarial base para este rol es de $180,000 a $339,250, determinado en función de tu ubicación, experiencia y el salario de los empleados en puestos similares. También serás elegible para acciones y beneficios.

NVIDIA acepta solicitudes de empleo de manera continua. Estamos comprometidos a fomentar un entorno de trabajo diverso y estamos orgullosos de ser un empleador que ofrece igualdad de oportunidades. La diversidad es muy valorada en nuestros empleados actuales y futuros, y no discriminamos (incluyendo en nuestras prácticas de contratación y promoción) por raza, religión, color, nacionalidad, 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.