Machine Learning Engineer, Training an Acceleration

Apple • Full-time • US, WA, Seattle • 1w ago

Summary

The Video Computer Vision org is a centralized applied research and engineering team responsible for developing real-time on-device Computer Vision and Machine Perception technologies across Apple products. We focus on a balance of research and development to deliver Apple quality, state-of-the-art experiences. Our team prides itself on innovating through the full stack, and partnering with HW, SW and ML teams to influence the sensor and silicon roadmap that brings our vision to life! To succeed within this role, you should have shown experience in several of the following areas:

Key Qualifications

Strong experience in the area of developing machine learning training framework, or hardware acceleration of machine learning tasks

Strong software engineering skillsStrong Python programming

Strong communication skills; great work ethic and teamwork

Familiar with hardware architecture, cache utilization, data streaming model

Familiar with network optimization algorithm including quantization, sparsification, experience with knowledge distillation or NAS is a plus

Experience on computer vision and machine learning model development is a big plus

Experience on management of a team of 2-10 people is a plus

Description

We’re looking for strong software engineer/leads to build a next generation Deep Learning technology stack to accelerate on-device machine learning capabilities and emerging innovations. You’ll be part of close nit software developers and deep learning experts working in the area of hardware aware neural network optimization, algorithms, and neural architecture search. We’re looking for candidates with strong software engineering skills, passionate about machine learning, computational science and hardware.

RESPONSIBILITIES:

* Design and develop APIs for common and emerging deep learning primitives: layers, tensor operations, optimizers and more specific hardware features.

* Implement efficient tensor operations and DNN training algorithms.

* Train and evaluate DNNs for the purpose of benchmarking neural network optimization algorithms. Our framework reduces latency and power consumption of neural networks found in many Apple products. * Perform research in emerging areas of efficient neural network development including quantization, pruning, compression and neural architecture search, as well as novel differentiable compute primitives.

* We encourage publishing novel research at top ML conferences.

Education & Experience

You've got ONE of the following three: 1. PhD in Machine Learning, Statistics, Optimization or related field with experience building production systems or have equivalent experience in industry 2. MS in CS with 4+ years of experience in working with large ML projects 3. BS in CS with 6+ years of experience in the industry with at least 3 years in machine learning.

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 $115,000 and $217,500, 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.

Role Number: 200441961