NVIDIA conçoit des processeurs graphiques, des systèmes de calcul et des logiciels pour l'intelligence artificielle et les applications graphiques.
Senior Software Engineer, JAX
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of d…
Repères sur NVIDIA
- Domaine officiel
- nvidia.com
- Offres ouvertes
- 100
Détails de l’offre
La description complète publiée par NVIDIA.
Description de l’offre
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code. Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX. To achieve this and provide an excellent user experience, you have strong programming skills and excellent system design skills. You should have excellent communication and planning skills for this role.
Come join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world!
What you will be doing
- Play meaningful role in NVIDIA's effort in contributing to JAX.
- Design and implement JAX core components and drive peak performance on NVIDIA products.
- Work with AI applied researchers and leaders to build future-proof models
- Build tools that will increase the efficiency of teams developing AI-based systems.
- Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products. What we need to see:
- BS in Computer Science or Computer Engineering or related field (or equivalent experience).
- 6+ years relevant experience.
- C/C++ and Python programming.
- Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.).
- Proven ability developing customer-facing solutions, balancing feature requests and bugs.
- Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design.
- Highly motivated with excellent verbal and written communication skills.
- Ability to work successfully with multi-functional teams, principles and architects. Coordinates effectively across organizational boundaries and geographies. Ways to stand out from the crowd:
- Understanding of JAX, Autograd, tracing, code generation and DSL compilers and their design.
- Understanding of deep learning training in distributed contexts: multi-GPU, multi-node, synchronous vs asynchronous.
- Background with software shipping cycles (dev, deploy, release, CI).
- Experience building distributed systems and services at large scale.
Candidater à cette offre
Comment candidater ?
L’offre est publiée sur le site de NVIDIA. Ouvrez l’offre originale ↗ pour y déposer votre candidature. Screeningpass ne remplit pas de formulaire à votre place.
À qui écrire chez NVIDIA pour ce poste ?
Lancez une recherche de contact depuis cette page. Screeningpass vous propose jusqu’à trois personnes liées au poste. Vous choisissez celle à qui écrire, puis Screeningpass cherche son e-mail professionnel et prépare un message. Vous le relisez et l’envoyez vous-même avec Gmail. Une adresse e-mail n’est pas toujours disponible, et une réponse n’est jamais garantie.
Combien coûte la recherche d’un contact ?
Une recherche coûte 1 crédit. Le crédit vous est rendu si la recherche échoue. L’inscription offre 3 crédits, soit 3 recherches.
À quoi sert Screeningpass ?
Screeningpass rassemble des offres de stage, d’alternance et d’emploi publiées par les entreprises. Pour une offre, Screeningpass trouve une personne à contacter dans l’entreprise, cherche son e-mail professionnel et prépare un message à partir de votre CV. Voir le fonctionnement en détail.
D’autres offres chez NVIDIA
99 autres offres actuellement disponibles.
Vous développerez l’infrastructure de compilation pour la programmation GPU en Python, notamment avec CUTLASS CuTeDSL. Le poste couvre la conception de DSL et de dialectes MLIR, l’optimisation de kernels GPU ainsi que le profilage et la validation de leurs performances. Il s’adresse à une personne qui poursuit ou détient un diplôme en informatique ou dans une discipline technique connexe, avec des compétences en Python, C++ et programmation GPU.
Le poste développe les partenariats commerciaux de NVIDIA avec des éditeurs de logiciels indépendants et des partenaires cloud en EMEA. Il vise à intégrer les logiciels et frameworks NVIDIA aux offres partenaires et à accélérer leur adoption auprès des entreprises. La fonction coordonne les équipes commerciales, techniques et produit et suit le pipeline ainsi que les résultats d’adoption.
NVIDIA recherche un Technical Account Manager spécialisé en HPC et cloud pour accompagner ses clients entreprise. Le poste consiste à résoudre des problèmes techniques liés à l’installation, à l’exploitation et à la maintenance de technologies de centres de données, notamment en IA et HPC. La personne travaillera avec des équipes d’ingénierie, de marketing et de support, et contribuera à documenter les méthodes de résolution.
NVIDIA recherche un Senior Account Manager pour développer ses solutions d’IA et d’informatique accélérée auprès d’acteurs des marchés de capitaux en EMEA. Le poste consiste à gérer des comptes stratégiques, accroître l’adoption des plateformes NVIDIA et piloter la croissance des revenus. Une expérience de 12 ans ou plus dans la vente de solutions aux décideurs ainsi qu’un diplôme de niveau Bachelor sont demandés.
NVIDIA recherche un Solution Architect spécialisé en IA souveraine pour accompagner des clients publics au Royaume-Uni et en Europe. Le poste consiste notamment à concevoir et déployer des solutions d’IA, à soutenir des preuves de concept et à conseiller les équipes techniques des clients et partenaires. Une expérience d’au moins cinq ans, ainsi que la maîtrise de Python, PyTorch, Kubernetes et des conteneurs, sont demandées.
NVIDIA recherche un stagiaire en ingénierie logicielle au sein de son équipe de simulation physique à Zurich, pour contribuer au développement de fonctionnalités Newton et de workflows destinés à l’apprentissage robotique. Le stage de six mois, prévu en 2027, porte notamment sur la simulation numérique, la robotique, les expérimentations et le développement logiciel avec des outils comme PhysX, MuJoCo, Isaac Sim, Isaac Lab, Warp et CUDA.