Sr. ML Infrastructure Engineer, Apple Data Platform

Cupertino, California, United States Software and Services

The Apple Data Platform (ADP) group builds the data platform that enables the next generation of intelligent experiences on all Apple products and services. ADP empowers Apple engineers to deliver ML-driven products and innovations rapidly and at scale. We are looking for an experienced engineer who can bring their passion for machine learning, infrastructure, big data, and distributed systems to build world class data+ML platform/products at scale. You will work with many cross functional teams and lead the planning, execution and success of technical projects with the ultimate purpose of improving ML experience for Apple customers. Are you passionate about building scalable, reliable, maintainable infrastructure and solving data problems at scale? Come join us and be part of the Data Infrastructure journey.

Description

Apple Ray leverages open-source Ray to offer a unified framework for processing of complex data+ML pipelines. It enables the next generation of intelligent experiences for Apple products and services by combining data and processing layers into one unified end-to-end workflow that eliminates the complexity of running multiple independent jobs while significantly improving the hardware resource efficiency and development speed. Tight integration of Apple Ray with Apple Data services makes it the go-to solution when dealing with complex and large-scale data and ML pipelines. The team enables future Apple intelligent products by making cutting edge ecosystem of data+ML technologies for large-scale and efficient systems for all data and ML engineers within Apple. As a member of the Apple Ray team, your responsibilities will include:

• Designing, implementing, and maintaining distributed systems to build world-class ML platforms/products at scale
• Diagnose, fix, improve, and automate complex issues across the entire stack to ensure maximum uptime and performance
• Design and extend services to improve functionality and reliability of the platform
• Monitor system performance, optimize for cost and efficiency, and resolve any issues that arise
• Build relationships with stakeholders across the organization to better understand internal customer needs and enhance our product better for end users

Minimum Qualifications

• 5+ years of experience in distributed systems with deep knowledge in computer science fundamentals
• Experience with containerization and orchestration technologies, such as Docker and Kubernetes.
• Experience in delivering data and machine learning infrastructure in production environments
• Experience configuring, deploying and troubleshooting large scale production environments
• Experience in designing, building, and maintaining scalable, highly available systems that prioritize ease of use
• Experience with alerting, monitoring and remediation automation in a large scale distributed environment
• Extensive programming experience in Java, Python or Go
• Strong collaboration and communication (verbal and written) skills
• B.S., M.S., or Ph.D. in Computer Science, Computer Engineering, or equivalent practical experience

Preferred Qualifications

• Understanding of the ML lifecycle and state of the art ML Infrastructure technologies
• Experience with GPU and other type of HPC infrastructure
• Experience with training framework like PyTorch, Tensorflow, JAX
• Deep understanding of Ray and KubeRay
• Experience with ML Training/Inference profiling and optimization

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 $181,100 and $318,400, 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.