AIML - Staff Software Engineer, ML Compute - ML Platform & Technologies

As a staff software engineer, you have the following 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 our ML model training compute infrastructure. Collaborate with multi-functional teams, integrate with Kubernetes in on-premises and cloud provider clusters, and enable seamless integration with NVIDIA GPUs and other ML accelerators.

• 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 our ML model training compute infrastructure.
• Collaborate with multi-functional teams, integrate with Kubernetes in on-premises and cloud provider clusters, and enable seamless integration with NVIDIA GPUs and other ML accelerators.
• Partner with data scientists and machine learning engineers across different Apple organizations to define high-impact product features and deliver them with quality.
• Lead a group of engineers to deliver high-quality products/services.
• Support junior engineers by providing advice, mentoring, and educational opportunities.

• Bachelors in Computer Science
• 10+ years of industry related experience, working in collaborate environments
• Experience with public cloud infrastructure like: Kubernetes, Amazon EC2 and EKS, Google Cloud Platform
• Strong software development skills, with proficiency in relevant languages (ex. 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; can take feedback and iterate on a solution in a collaborative setting

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

Apple is where individual imaginations gather together, committing to the values that lead to great work! Every new product we build, service we create, or Apple Store experience we deliver is the result of us making each other’s ideas stronger.