AIML Staff ML Infrastructure Engineer, ML Platform & Technology ML Compute

Apple

Posted on : 04-10-2025

Employer Active

1 Vacancy

Job Alert

You will be updated with latest job alerts via email

Valid email field required

Send jobs

Send me jobs like this

Job Alert

You will be updated with latest job alerts via email

Valid email field required

Send jobs

Job Location

San Francisco, CA - USA

Monthly Salary

Not Disclosed

Salary Not Disclosed

Vacancy

1 Vacancy

Posted on : 04-10-2025

Job Description

As a staff engineer on ML Compute team your work will include:- Lead the development of the infrastructure to run large-scale workloads on the Cloud such as Apache Spark Ray and distributed training.- Optimize platform efficiency and throughput by improving resource management capabilities with schedulers like Apache YuniKorn and Kueue.- Integrate new features from core distributed computing and ML frameworks into the platform offering them to production users and providing support.- Enhance the platforms scalability performance and observability through improved monitoring and logging.- Drive the architectural evolution of the platform by adopting modern cloud-native technologies to improve system performance efficiency and scalability.- Reduce dev-ops efforts by automating and streamlining operational processes.- Mentor engineers in areas of your expertise fostering skill growth and knowledge sharing.

Bachelors in Computer Science engineering or a related field
6 years of hands-on experience in building scalable backend systems for training and evaluation of machine learning models
Proficient in relevant programming languages like Python or Go
Strong expertise in distributed systems reliability and scalability containerization and cloud platforms
Proficient in cloud computing infrastructure and tools: Kubernetes Ray PySpark
Ability to clearly and concisely communicate technical and architectural problems while working with partners to iteratively find solutions

Advance degrees in Computer Science engineering or a related field.
Hands-on experience with cloud-native resource management and scheduling tools like Apache YuniKorn.
Experience with advanced architecture for distributed data processing and ML workloads.
Proficient in working with and debugging accelerators like: GPU TPU AWS Trainium.

Required Experience:

Staff IC

Employment Type

Full-Time

Company Industry

Key Skills

Apply Now

About Company

Apple

Report This Job

Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.

Start Now

Dr.Job AutoApply

3X your job search with AutoApply's AI for faster dream job results.

AIML Staff ML Infrastructure Engineer, ML Platform & Technology ML Compute

Apple

Job Description

Employment Type

Company Industry

Key Skills

About Company

Similar Jobs

Founding ML Engineer

ML (EL) Teacher Posting

Lead Data Scientist (AI & ML)

Senior Staff Engineer

Technology Graduate Programme (C# .Net, Python, Java, C++, Platform)

Senior Platform Engineer B2B Integration

Staff Pharmacist

Staff 25H