Estados Unidos
CONCEPT SOLUTIONS LLC
NVIDIA
Ingeniero/a de Software Senior — cuEquivariance
Estados Unidos
Santa Clara, California, Estados Unidos
PythonC++
Traducción automática. Consulta el original.
Sobre la oportunidad
NVIDIA lleva más de 25 años transformando los gráficos por ordenador, los videojuegos y la computación acelerada. Es un legado único de innovación impulsado por la gran tecnología y personas increíbles. Hoy, estamos aprovechando el potencial ilimitado de la IA para definir la próxima era de la computación. Una era en la que nuestra GPU actúa como el cerebro de ordenadores, robots y coches autónomos que pueden comprender el mundo. Hacer lo que nunca se ha hecho antes requiere visión, innovación y el mejor talento del mundo. Como NVIDIAN, te sumergirás en un entorno diverso y de apoyo donde todos se inspiran para hacer su mejor trabajo. Únete a nuestro grupo y descubre cómo puedes tener un impacto duradero en el mundo.
NVIDIA BioNeMo está construyendo la base computacional para la próxima generación de descubrimientos biológicos. Buscamos un Ingeniero/a de Software Senior para unirse al equipo de cuEquivariance, una biblioteca de NVIDIA que acelera las redes neuronales geométricas en GPUs de NVIDIA, permitiendo a los investigadores en biología molecular, ciencia de materiales y física entrenar y desplegar modelos equivariantes a escala. Este equipo construye y lanza los kernels de GPU y las interfaces de software de producción que potencian el deep learning equivariante en todo el campo científico. El trabajo abarca la ingeniería de kernels de CUDA, el desarrollo de bibliotecas de Python que involucran tanto PyTorch como JAX, y la colaboración directa con equipos de investigación y desarrolladores de frameworks externos. Si quieres trabajar donde la computación con GPU se une al deep learning basado en grafos, este es el puesto para ti. Tu trabajo se ejecutará en pipelines de producción en toda la comunidad científica.
¿Qué harás?
- Construir, implementar y optimizar kernels de CUDA para primitivas de redes neuronales equivariantes — productos tensoriales, polinomios segmentados y operaciones basadas en triángulos — dirigidos a un rendimiento máximo en generaciones de GPUs de NVIDIA.
- Ser responsable de la entrega de extremo a extremo de primitivas de ML geométricas aceleradas por GPU: desde la implementación hasta software validado y de calidad de producción del que dependen los frameworks externos.
- Construir y mantener las interfaces para PyTorch y JAX que exponen las primitivas de cuEquivariance a los desarrolladores de aplicaciones e investigadores.
- Impulsar la infraestructura de CI/CD para compilaciones de kernels multi-GPU, pruebas automatizadas de corrección y seguimiento de regresiones de rendimiento.
- Colaborar con equipos de Ciencia Aplicada e investigación para evaluar nuevas arquitecturas equivariantes y traducir prototipos en kernels de producción.
- Interactuar directamente con desarrolladores de frameworks de terceros y socios para alinear interfaces y asegurar que el software entregado se integre limpiamente en los pipelines de producción.
¿Qué necesitamos ver?
- Más de 6 años de experiencia en ingeniería de software con una sólida formación en CUDA y programación de GPU.
- Profundo conocimiento de C++ y Python; experiencia en la construcción y lanzamiento de bibliotecas de producción utilizadas por desarrolladores externos.
- Buena base en computación con GPU: jerarquía de memoria, ejecución a nivel de warp, ocupación y metodología de perfilado de rendimiento.
- Experiencia en la construcción o contribución a bibliotecas de software científico de producción, frameworks de ML o APIs de GPU orientadas al desarrollador.
- Familiaridad con conceptos de machine learning geométrica — equivariancia, representaciones de grupo, representaciones irreducibles o productos tensoriales — suficiente para trabajar eficientemente en el dominio.
- Grado/Máster en Ciencias de la Computación, Física, Matemáticas Aplicadas o un campo relacionado, o experiencia equivalente.
Formas de destacar entre la multitud
- Has contribuido o utilizado en profundidad un framework importante de redes neuronales que respeta la equivariancia: e3nn, MACE, NequIP, SE(3)-Transformers o similar.
- Experiencia práctica con desarrollo de kernels Triton u otras herramientas de autoría de kernels de GPU junto con CUDA.
- Experiencia en diseño de algoritmos de precisión mixta o conscientes de tensor cores para cargas de trabajo científicas o de ML.
- Doctorado o experiencia equivalente en química computacional, biofísica, física o informática con un enfoque en deep learning geométrica o HPC.
- Contribuciones a proyectos de ML geométrico o computación con GPU de código abierto.
Considerada ampliamente como uno de los empleadores más deseados del mundo de la tecnología, NVIDIA ofrece salarios muy competitivos y un paquete integral de beneficios. Mientras planificas tu futuro, consulta lo que podemos ofrecerte a ti y a tu familia en nvidiabenefits.com/ nvidiabenefits.com/
Tu salario base se determinará en función de tu ubicación, experiencia y la remuneración de los empleados en puestos similares. El rango salarial base es 184,000 USD - 287,500 USD para el Nivel 4, y 224,000 USD - 356,500 USD para el Nivel 5.
También serás elegible para acciones y beneficios nvidia.com/en-us/benefits/.
Las solicitudes para este puesto se aceptarán al menos hasta el 21 de agosto de 2026.
Esta oferta es para 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 (incluyendo 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