Ingeniero de Software Senior en JAX

Sobre esta oportunidad

NVIDIA está buscando ingenieros senior apasionados por la inteligencia artificial para contribuir al desarrollo y optimización de su plataforma de IA. Específicamente, trabajarás con JAX, una herramienta revolucionaria que permite diferenciar entre código de control Python y código NumPy. El objetivo es construir una plataforma pulida, rápida, modular y bien coordinada que maneje datos, entrenamiento y análisis para una amplia variedad de soluciones de deep learning. Si tienes sólidas habilidades de programación, excelente capacidad para diseñar sistemas y te apasiona llevar la investigación en computing numérico al siguiente nivel, este es tu lugar.

Perfil buscado

Buscamos a un profesional senior con una combinación única de habilidades técnicas profundas y capacidad para trabajar en equipos multidisciplinarios. Necesitamos alguien que entienda no solo cómo escribir código excelente, sino también cómo diseñar sistemas que resuelvan problemas complejos en machine learning a escala. Debes ser comunicador fluido, tanto en términos técnicos como estratégicos, y tener una mentalidad orientada a la excelencia del usuario final.

Requisitos clave

• Licenciatura en Ciencias de la Computación, Ingeniería Informática o campo relacionado (o experiencia equivalente comprobada)
• Mínimo 6 años de experiencia relevante en desarrollo de software
• Dominio avanzado de C/C++ y Python
• Experiencia comprobada con frameworks de machine learning y su funcionamiento interno (PyTorch, TensorFlow, scikit-learn, etc.)
• Historial demostrando desarrollo de soluciones orientadas al cliente, balanceando features y resolución de bugs
• Fundamentos sólidos en arquitecturas CPU y GPU, librerías numéricas y diseño modular de software
• Excelentes habilidades de comunicación verbal y escrita
• Capacidad probada de colaborar efectivamente con equipos multifuncionales, principios y arquitectos, coordinando trabajo a través de límites organizacionales y geográficos

Lo que harás día a día

• Jugarás un rol significativo en los esfuerzos de NVIDIA por contribuir al ecosistema JAX
• Diseñarás e implementarás componentes core de JAX, logrando máximo rendimiento en productos NVIDIA
• Colaborarás con investigadores aplicados en IA y líderes técnicos para construir modelos a prueba de futuro
• Crearás herramientas que aumenten la eficiencia de equipos que desarrollan sistemas basados en IA
• Cerrarás la brecha entre la investigación más reciente en computing numérico, simulación y deep learning con sus aplicaciones en productos reales

¿Qué te haría destacar?

• Comprensión profunda de JAX, Autograd, tracing, code generation y compiladores DSL
• Experiencia en distribución de entrenamiento de deep learning: multi-GPU, multi-node, sincrónico vs asincrónico
• Antecedentes en ciclos de software shipping completos (desarrollo, deploy, release, CI/CD)
• Experiencia construyendo sistemas distribuidos y servicios a gran escala

Beneficios de la posición

• Trabajar en tecnología que está transformando el mundo de la inteligencia artificial
• Colaboración con algunos de los mejores investigadores y ingenieros en el campo
• Impacto global: tu trabajo influirá en sistemas de IA utilizados en toda la industria
• Oportunidad de crecer en una de las empresas tecnológicas más innovadoras del mundo
• Flexibilidad total de trabajo remoto
• Ambiente que valora la excelencia técnica y la innovación constante

¿Por qué considerar esta empresa?

NVIDIA es líder indiscutible en computación acelerada y es fundamental en el ecosistema de inteligencia artificial global. La empresa invierte significativamente en investigación y desarrollo, especialmente en herramientas como JAX que definen el futuro del computing numérico. Trabajar aquí significa formar parte de una organización que establece estándares en la industria y tiene impacto directo en cómo se construye la IA en todo el mundo.

Rango salarial

Los salarios base varían según tu ubicación, experiencia y nivel. Para Polonia, los rangos son: 292,500 PLN - 507,000 PLN para Nivel 4, y 375,000 PLN - 650,000 PLN para Nivel 5. NVIDIA ofrece compensación competitiva y beneficios comprehensivos que van más allá del salario base.

¿Te interesa? Aplica directamente desde el enlace de la vacante. ¡Esperamos conocer tu perfil!