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
#J-18808-Ljbffr
Apply for This Position
Ready to take the next step? Click the button below to submit your application.
Submit Application