AIML - Machine Learning Engineer (On-Device Machine Learning), Machine Learning Platform and Infrastructure
Apple
Summary
Description
Minimum Qualifications
- 5+ years proven programming skills using standard ML tools such as C/C++, CUDA/Metal, PyTorch, Tensorflow
- Experience building APIs and/or core components of ML frameworks and strong attention to detail
- Capacity to iterate on ideas, work with a variety of partners from all parts of the stack — from apps to Compilation, HW Arch, and Power/Performance analysis
- Excellent problem-solving (e.g. via building forward-looking prototype systems), critical thinking, strong communication, and collaboration skills.
Key Qualifications
Preferred Qualifications
- Knowledge of existing ML Framework implementation (Jax, PyTorch, TensorFlow)
- Knowledge of code execution down to the hardware/accelerator level
- Hands-on experience working on LLVM, compiler technologies, optimization techniques like quantization and sparsity-induction is a huge plus
- M.S. or B.S. or equivalent experience in Computer Science
Education & Experience
Additional Requirements
Pay & Benefits
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. Learn more about your EEO rights as an applicant.