Senior ML Platform Engineer - JAX & Performance

NVIDIA Gruppe

Full-time Other-General
Apply Now
Location
Remote, Remote, Switzerland
Posted
June 24, 2026

Job Description

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.

#J-18808-Ljbffr