💼 Full-Time Position

Senior RL Training Infrastructure Engineer (TPU/JAX)

🏢
Apple
📍 zürich, zürich, Switzerland
📍
Location
zürich, Switzerland
📅
Posted
June 05, 2026
⏰
Type
Full-Time
🎯

Full-Time Opportunity: This is a permanent, full-time position with a competitive package and real career growth potential.

Job Description

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