Finding the best job has never been easier
Apple AIML - Machine Learning Engineer On-Device United States, Washington, Seattle 767835581
10.08.2024
Share
Description
As a member of this team, the successful candidate will build features for our on-device inference stack to support new and existing customers with ever evolving machine learning models. This person will interact with first and third part clients to figure out their functional and performance requirements and work collaboratively with backend teams to map those requirements to functional execution. They will perform analysis when needed to squeeze every ounce of memory for flop from the underlying hardware, and they will drive structural improvements that can be made in the on device machine learning stack.
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.
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
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 $135,400 and $250,600, and your base pay will depend on your skills, qualifications, experience, and location.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.
These jobs might be a good fit
