Senior RL Training Infrastructure Engineer (TPU/JAX)

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