Senior Machine Learning Engineer

Apple Inc.; New York (NY); On-site; USD 143,000 - 265,000; Full time

Job summary

Apple is seeking an ML Engineer to contribute to the ML Platform team in New York City. You'll design and develop backend systems that support scalable modeling and enhance advertising features. Ideal candidates are experienced in machine learning frameworks and distributed systems, offering them the chance to innovate within a collaborative environment and contribute to Apple’s privacy commitments. This role promises competitive compensation, comprehensive benefits, and opportunities for professional growth.

Benefits

Comprehensive medical and dental coverage

Retirement benefits

Employee stock programs

Discounted products and free services

Reimbursement for formal education

Qualifications

• Experience designing and developing ML infrastructure/frameworks for training and inference.
• Familiarity with model optimization techniques.
• Strong problem-solving and debugging skills.

Responsibilities

• Design and develop secure and scalable back-end systems for ML.
• Build high-performing, elegant systems from the ground up.
• Automate routine tasks and improve performance through tuning.

Skills

ML infrastructure/frameworks

model quantization

distributed systems

AI/ML tooling

problem solving

Education

PhD in Computer Science

MS in Computer Science

BS in Computer Science

Tools

PyTorch

TensorFlow

Kubernetes

Ray

Spark

Job description

New York City, New York, United States Software and Services

Description

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!

Minimum Qualifications

• 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.

Preferred Qualifications

• 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.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $143,100 and $264,200, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant .