Senior ML Platform Engineer - JAX & Performance

NVIDIA Gruppe is looking for senior engineers in Switzerland to enhance its AI platform, focusing on deep learning performance optimizations using JAX. You will play a pivotal role in developing core components and working with AI researchers to create efficient tools for AI-based systems.

The ideal candidate has over 6 years of relevant experience, proficiency in C/C++ and Python, and expertise in machine learning frameworks. Join NVIDIA to contribute to groundbreaking advancements in numerical computing and machine learning.