AIML - ML Researcher, Foundation Models

Requirements

• Constantly seeking challenging problems to work on and new area to learn from.
• Proven track record in training or deployment of large models or building large-scale distributed systems.
• Proficient programming skills in Python and one of the deep learning toolkits such as JAX, PyTorch, or Tensorflow.
• Ability to work in a collaborative environment.

Nice to Haves

• PhD, or equivalent practical experience, in Computer Science, or related technical fields.
• High performance parallel and distributed computing.

What You'll Be Doing

• Work with a close-knit and fast-growing team of world-class engineers and scientists.
• Tackle challenging problems in foundation models and deep learning.
• Identify and develop novel applications of deep learning in Apple products.