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
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
- Experience with foundation models and large language models (preferred)
- Published research or significant contributions to open-source ML projects (preferred)
- Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services
- Relocation might be eligible for discretionary bonuses or commission payments
