On-device ML Infrastructure Engineer (Compiler & Runtime)
On-device ML Infrastructure Engineer (Compiler & Runtime)
Posted on :
02-08-2025
Employer Active
1 Vacancy
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Share
Send me jobs like this
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Yearly Salary
$ 181100 - 318400
Vacancy
1 Vacancy
Posted on :
02-08-2025
Job Description
We are building the first end-to-end developer experience for ML development that by taking advantage of Apples vertical integration allows developers to iterate on model authoring optimization transformation execution debugging profiling and an engineer in this role you will be tasked with building critical compiler and runtime infrastructure that powers almost all on device machine learning features across Apple devices. You will have the opportunity to leverage Apples unique vertical machine learning stack that goes all the way down to custom silicon and drive impact on a wide range of Apple features and responsibilities:Design build and maintain critical machine learning infrastructure that powers Apples on device machine learning with downstream hardware compilers to best leverage Apples on device machine learning with first and third party users to adopt our infrastructure and apply best practices when they implement machine learning on Apple our infrastructure can run optimally for a wide range of first and third party machine learning models.
- Bachelors in Computer Science Engineering or related discipline.
- Highly proficient in C. Familiarity with Python and/or Swift
- Familiarity with Operating Systems and Embedded Programming.
- Sound understanding of ML fundamentals including common architectures such as Transformers.
- Experience with any on-device ML stack such as TFLite ONNX ExecuTorch etc.
- Experience with open source machine learning models (Mistral Phi Gemma Huggingface etc)
- Experience with any compiler stack (MLIR/LLVM/TVM/...).
- Experience with any ML authoring framework (PyTorch TensorFlow JAX etc.).
- Experience with machine learning accelerators and GPU programming.
- Good communication skills including ability to communicate with cross-functional audiences.
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 $181100 and $318400 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 Apples 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 Apples Employee Stock Purchase Plan. Youll 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 employees also have the opportunity to become an Apple shareholder through participation in Apples 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 Apples Employee Stock Purchase Plan. Youll 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.
Employment Type
Full-Time
Key Skills
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.
