Large Machine Learning System Optimization Engineer

Large Machine Learning System Optimization Engineer The Video Computer Vision org is a centralized applied research and engineering team responsible for developing real-time on-device Language, 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. Key Qualifications
•   Strong experience in the area of developing machine learning training framework, or hardware acceleration of machine learning tasks
•   Strong software engineering skills in Python
•   Strong communication skills; phenomenal work ethic and collaboration
•   Familiar with network optimization algorithm including quantization, sparsification, experience with knowledge distillation or NAS is a plus
•   Experience with hardware architecture, cache utilization, data streaming model is a plus
•   Experience on large language models (LLMs), large computer vision and machine learning model, and generative AI model development is a big plus
•   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. Description We're looking for strong ML software engineer/leads to build a next generation Deep Learning technology stack to accelerate on-device large 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. Education & Experience BS and a minimum of 10 years relevant industry experience
•   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 $161,700.00 and $284,900.00, 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 take affirmative action to ensure 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.