More jobs:
Senior RL Training Infrastructure Engineer; TPU/JAX
Job in
Zürich, 8058, Zurich, Kanton Zürich, Switzerland
Listed on 2026-06-11
Listing for:
Apple
Apprenticeship/Internship
position Listed on 2026-06-11
Job specializations:
-
Business
Computer Science
Job Description & How to Apply Below
Location: Zürich
Apple is looking for a skilled engineer to join its Core Foundation Models team in Zürich. The role involves designing and optimizing large-scale ML training infrastructure, particularly focusing on reinforcement learning with TPU-based systems using JAX. Candidates should have a PhD or MSc in Computer Science, solid experience with GPUs/TPUs, and a strong proficiency in PyTorch or JAX. This position is ideal for those passionate about high-performance computing and distributed systems.
#J-18808-Ljbffr
Position Requirements
10+ Years
work experience
Note that applications are not being accepted from your jurisdiction for this job currently via this jobsite. Candidate preferences are the decision of the Employer or Recruiting Agent, and are controlled by them alone.
To Search, View & Apply for jobs on this site that accept applications from your location or country, tap here to make a Search:
To Search, View & Apply for jobs on this site that accept applications from your location or country, tap here to make a Search:
Search for further Jobs Here:
×