AIML – Machine Learning Engineer, Machine Learning Platform & Infrastructure Job ID- 996 Job ID- 3234

MaziCTools; United States; Remote; USD 143,000 - 264,000; Full time

Benefits offered by this job

Stock programs

Discretionary bonuses

Relocation assistance

Job summary

Apple, located in Cupertino, seeks an experienced ML Engineer to build features for on-device inference and optimize compiler/tooling for ML workloads. This role focuses on delivering high-performance, low-latency ML solutions for mobile and embedded devices.

The candidate will convert models to target devices, write tests, and collaborate with HW/SW teams. Strong C/C++, Python skills and ML framework familiarity are essential for success.

Qualifications

• Bachelor's/Master's/PhD in CS or related field.
• At least 4 years in ML Engineering, System Software Engineering, or related.
• Strong proficiency in C/C++ and Python.
• Familiarity with ML fundamentals and ML frameworks.

Responsibilities

• Build features for on-device inference stack to support accuracy-preserving techniques.
• Develop ML compilers, runtimes, execution kernels, and debugging/tooling for ML models.

Skills

C/C++

Python

ML fundamentals

ML frameworks

Communication skills

Swift

On-device ML

Education

BS/MS/PhD in Computer Science or related field

Tools

PyTorch

JAX

MLIR/LLVM

Job description

Cupertino, California, United States

Summary

Posted: Sep 5, 2024

Weekly Hours: 40

Description

• Build features for our on-device inference stack to support the most relevant accuracy preserving, general purpose techniques that empower model developers to compress and accelerate SoTA models (e.g., LLMs) in apps
• This includes building Machine Learning compilers, runtimes, execution kernels, optimizations on ML models, tooling for debugging/visualization of ML models, etc.
• Convert models from a high-level ML framework to a target device (CPU, GPU, Neural Engine) for optimal functional accuracy and performance
• Write unit and system integration tests to ensure functional correctness and avoid performance regressions
• Diagnose performance bottlenecks and work with HW & SW Arch teams to co-design solutions that further improve latency, power, and memory footprint of neural network workloads
• Analyze impact of model optimization (compression/quantization etc) on model quality by partnering with modeling and adaptation teams across diverse product use cases

Minimum Qualifications

• Bachelors/Masters/PhD in Computer Science or related fields.
• At least 4 years of experience in Machine Learning (ML) Engineering, System Software Engineering, or related fields.
• Strong proficiency in C/C++ and Python.
• Familiarity with ML fundamentals.
• Familiarity with developing or using ML Frameworks.

Preferred Qualifications

• Experience with PyTorch/JAX Machine Learning frameworks.
• Experience with Machine Learning model inference (serving) is a big plus.
• Experience with MLIR / LLVM compiler technologies is a big plus.
• Experience with on-device machine learning or system software design is a big plus.
• Experience with Swift is a plus.
• Excellent communication skills.
• Passion for designing Software systems, APIs, and extensible products.

Pay & Benefits

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 $143,100 and $264,200, 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.

AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure