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

$166,600 - $296,300/Yr

Apple - Seattle, WA

Full-time - Mid Level

Seattle, WA

Computer and Electronic Product Manufacturing

About the position

As a Senior Software Engineer in the ML Compute team at Apple, you will be responsible for designing, building, and maintaining the compute infrastructure that supports machine learning, artificial intelligence, and computer vision applications. This role involves shaping critical compute components and contributing to the infrastructure that enables intelligent experiences across all 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 and integrate with Kubernetes in on-premises and cloud provider clusters.
• Enable 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 and support junior engineers through mentoring and educational opportunities.

Requirements

• Proven experience in software engineering with a focus on machine learning systems.
• Strong knowledge of large-scale system architecture and design principles.
• Experience with Kubernetes and cloud provider clusters.
• Familiarity with NVIDIA GPUs and ML accelerators.
• Ability to collaborate effectively with cross-functional teams.

Nice-to-haves

• Experience in mentoring junior engineers.
• Knowledge of data pipeline development and machine learning applications.
• Familiarity with resource management and scheduling solutions.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses including tuition
• Discretionary bonuses or commission payments
• Relocation assistance