Senior Machine Learning Engineer
Apple
Full-time
On-site
New York City, New York, United States
Machine Learning
The ML Platform team is responsible for bringing numerous features to advertisers and consumers while simultaneously supporting scalable modeling and continuous experimentation by all Ad Platforms teams. As a key contributor to this team, you will design and develop secure and scalable back-end systems. You will enjoy building high-performing, elegant systems from the ground up, in close partnerships with various teams. You will also possess keen judgment in selecting technologies and building the right solution for the interesting challenges we get to tackle here. You will have the opportunity to define and refine architectures to meet the unique ad network challenges we must solve. You will play a meaningful role building machine learning products which deliver on Apple's privacy commitments and change the way advertising works with data.
Join us and contribute to a culture that emphasizes reliability, simplicity, and scalability. You will join a team of world-class machine learning engineers hungry to apply leading-edge technologies to deliver extraordinary experiences to our customers. We are one team, nurturing each other’s growth and supporting each other in delivering for our customers!
- Experience designing and developing ML infrastructure/frameworks for training and inference.
- Experience of model quantization, tensor parallelism, and inference optimizations (e.g ONNX Runtime, TensorRT, vLLM).
- Experience building machine learning models using frameworks like PyTorch, TensorFlow.
- Experience building AI/ML tooling and/or infrastructure (e.g FeatureStore, VectorDB).
- Experience working on distributed systems (e.g Ray, Spark, Kubernetes).
- Experience performance tuning & trouble-shooting.
- Pride in building tools to automate routine tasks, organized & detailed.
- Familiarity with CI/CD tooling.
- Strong problem solving and debugging skills.
- PhD in Computer Science with 2+ years building production machine learning systems, or (b) MS in CS with 4+ years of engineering experience and 2+ years building production machine learning systems, or (c) BS in CS with 5+ years of engineering experience and 2+ years building production machine learning system.
- Ability to communicate effectively, both written and verbal, with technical and non-technical multi-functional teams.
- Results oriented with a desire to work in a fast-paced and collaborative work environment.
- Prior experience in advertising industry, federated learning and privacy-preserving ML techniques is a huge plus.