Senior Machine Learning Research Engineer, AIML
Core
Design software systems and algorithms for performant, scalable training and inference of large-scale deep learning models (LLMs, foundation models) across on-device and server environments.
Role type
Senior IC machine learning research engineer (system efficiency & infrastructure)
Builds
Core components for Apple's centralized ML platform, high-performance training/inference stacks, and open-source contributions
Domain
Generative AI, Large Language Models, ML Infrastructure, System-level optimization
Deliverable
production ML models
Required skills
PyTorch internals, Python programming, software–hardware co-design, LLM architectures, high-performance ML infrastructure
Preferred skills
AI/ML-optimized runtime stacks, parallelization algorithms, foundation model architectures, distributed training libraries (torch.distributed, DeepSpeed, FairScale), on-device inference, ML conference publications