Senior Software Engineer, JAX
NVIDIA · Zurich
Description du poste
About the role
NVIDIA is seeking senior engineers to develop and optimize its AI platform using JAX, focusing on performance improvements in deep learning frameworks.
Key responsibilities
- Design and implement core JAX components and drive peak performance on NVIDIA hardware.
- Collaborate with AI researchers and product leaders to build future‑proof models and tools.
- Build tooling that increases efficiency for teams developing AI‑based systems.
- Bridge cutting‑edge numerical computing research with real‑world product applications.
Required profile
- B.S. in Computer Science, Computer Engineering, or equivalent experience.
- 6+ years of relevant software engineering experience.
- Strong communication and planning skills, able to work across multi‑functional teams and geographies.
Required skills
- Proficiency in C/C++ and Python programming.
- Experience with machine learning frameworks (e.g., PyTorch, TensorFlow, scikit‑learn).
- Understanding of CPU and GPU architectures, numeric libraries, and modular software design.
- Familiarity with JAX, Autograd, tracing, code generation, and DSL compilers.
Questions fréquentes
Pourquoi signalez-vous cette offre ?
Aller plus loin
Salaires, guides et recherches en Suisse.
Salaire : Software Engineer Sur la base de 11 offres en SuisseSalaires par métier
Postulez en 30 secondes
Entrez votre email pour postuler. Un compte sera cree automatiquement.
En continuant, vous acceptez nos conditions d'utilisation.
Deja un compte ? Connexion
Une question sur cette offre ?
Posez-la ici : vous recevrez le récapitulatif de l'offre par e-mail, tout de suite.
Publie il y a 1 jour
Expire dans 1 mois
4 vues · 0 interesses
Boostez vos chances
Importez votre CV : nous vous proposons les offres qui matchent votre profil.
Analyse de votre CV en cours...
NVIDIA
Zurich