Apple is seeking to revolutionize artificial intelligence by developing sophisticated foundation models that power intelligent features across their product ecosystem.
Requirements
- Expert-level programming skills in Python
- Proficiency in machine learning frameworks such as Jax, PyTorch, TensorFlow
- Strong background in: Distributed training, Model optimization, and Machine learning infrastructure
- Experience with large-scale model training and deployment
- Familiarity with: Kubernetes, Docker, Cloud platforms (AWS, GCP, Azure)
- Distributed computing frameworks
- Experience with foundation models and large language models
- Background in multi-modal AI systems
Responsibilities
- Design and implement scalable, reliable and high-performance machine learning infrastructure for foundation models across text, image, speech, and multi-modal domains
- Collaborate with other teams to productionize state-of-the-art AI algorithms
- Optimize models for performance, efficiency, and on-device intelligence
- Implement machine learning systems with stringent privacy and security requirements
- May also be required to manage a small team of engineers.
Other
- MS or PhD in Computer Science, Machine Learning, or related technical field
- Base pay range for this role is between $181,100 and $318,400
- Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services
- Reimbursement for certain educational expenses - including tuition
- Discretionary bonuses or commission payments as well as relocation
