Enable job alerts via email!

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

Apple Inc.

Seattle (WA)

On-site

USD 166,000 - 297,000

Full time

27 days ago

Boost your interview chances

Create a job specific, tailored resume for higher success rate.

Job summary

Join a forward-thinking company as a Staff Software Engineer, where you'll lead the development of cutting-edge machine learning platforms. This role offers the chance to work with innovative technologies, collaborate with talented teams, and shape the future of machine learning solutions. You'll be responsible for designing and implementing scalable systems, managing resources, and supporting junior engineers. With a competitive salary and comprehensive benefits, this opportunity is perfect for those passionate about technology and eager to make a significant impact in a dynamic environment.

Benefits

Comprehensive medical and dental coverage
Retirement benefits
Discounted products and services
Educational reimbursement
Discretionary bonuses
Employee stock purchase plan

Qualifications

  • 10+ years of experience in collaborative environments with strong software development skills.
  • Proficient in cloud infrastructure and programming languages like Golang and Python.

Responsibilities

  • Own architecture and development of large-scale machine learning systems.
  • Lead engineers to deliver high-quality products and mentor junior team members.

Skills

Software Development
Problem Solving
Golang
Python
Interpersonal Skills

Education

Bachelor in Computer Science
Master in Computer Science
PhD in Computer Science

Tools

Kubernetes
Amazon EC2
Google Cloud Platform
CI/CD Pipelines
System Monitoring Tools

Job description

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

Seattle, Washington, United States Machine Learning and AI

Description

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.
  • Partner with data scientists and machine learning engineers across different Apple organizations to define high impact product features and deliver them with quality. In this role you are building the platform upon which other teams will develop data pipelines and machine learning applications.
  • Lead a group of engineers to deliver high quality products/services. Be able to stay on top of innovative technologies and apply them in the job. During the process, support junior engineers by providing advice, mentoring and educational opportunities.
Minimum Qualifications
  • Bachelors in Computer Science
  • 10+ years of industry related experience, working 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 (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
Preferred Qualifications
  • 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

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 $166,600 and $296,300, 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.

Get your free, confidential resume review.
or drag and drop a PDF, DOC, DOCX, ODT, or PAGES file up to 5MB.