AIML - Sr. Software Engineer - ML Compute, ML Platform & Technology

$166,600 - $296,300/Yr

Apple - Seattle, WA

Full-time - Senior

Seattle, WA

Computer and Electronic Product Manufacturing

About the position

As a Senior Software Engineer in the ML Compute team at Apple, you will play a crucial role in designing, building, and maintaining the compute infrastructure that supports machine learning, artificial intelligence, and computer vision applications. This position involves owning critical compute components and collaborating with various teams to enhance the infrastructure that powers intelligent experiences across Apple products and services, impacting billions of users worldwide.

Responsibilities

• Own the architecture, design, development, and operations of large-scale systems designed for machine learning.
• Develop custom scheduling, resource management solutions, and fleet management for ML model training compute infrastructure.
• Collaborate with multi-functional teams, integrating with Kubernetes in on-premises and cloud provider clusters, and enabling seamless integration with NVIDIA GPUs and other ML accelerators.
• Partner with data scientists and machine learning engineers to define high-impact product features and deliver them with quality.
• Lead a group of engineers to deliver high-quality products/services, while mentoring junior engineers.

Requirements

• Bachelor's degree in Computer Science.
• 5+ years of industry-related experience in collaborative environments.
• Experience with public cloud infrastructure like Kubernetes, Amazon EC2 and EKS, Google Cloud Platform.
• Strong software development skills, with proficiency in relevant languages (e.g., Golang, Python).
• Strong problem-solving skills and ability to write performant and high-quality code.
• Proficient understanding of the software development process, including unit testing and release management.
• Excellent interpersonal skills; able to work independently as well as in a team.

Nice-to-haves

• Strong understanding of batch scheduling systems and high-performance computing environments.
• Experience using system monitoring tools, automated testing frameworks, and CI/CD pipelines.
• Experience with GPUs and/or other ML accelerators in the context of Machine Learning.
• A passion for making simple, robust, and scalable platforms used by other engineering teams.
• Flexibility/adaptability for working in a dynamic environment with different frameworks and requirements.
• Masters or PhD in Computer Science.

Benefits

• Comprehensive medical and dental coverage.
• Retirement benefits.
• Discounted products and free services.
• Reimbursement for certain educational expenses, including tuition.
• Opportunity to participate in Apple's discretionary employee stock programs.