Staff/Sr. Machine Learning Engineer, Foundation Models - AI, Search & Knowledge Platforms
$171600/y-$302200/y
Apple17 hours ago
Highlight
OnsiteFull-timeSenior
Summary
Apple is seeking a Staff/Sr. Machine Learning Engineer for Foundation Models within their AI, Search & Knowledge Platforms team in San Francisco, CA. The role focuses on optimizing and deploying cutting-edge machine learning models to enhance various Apple services.
About the Role
As a Staff/Sr. Machine Learning Engineer, you will collaborate with the Foundation Model Research team to optimize inference for advanced model architectures. You will work closely with product teams to develop production-grade solutions that serve millions of customers in real-time. Your responsibilities will include building tools to identify bottlenecks in inference across different hardware and use cases, as well as mentoring and guiding other engineers in the organization.
About You
Required:
- 7+ years of experience leading and driving complex, ambiguous projects.
- Experience with high throughput services, particularly at supercomputing scale.
- Proficient with running applications on Cloud (AWS / Azure or equivalent) using Kubernetes, Docker.
- Familiar with GPU programming concepts using CUDA.
- Familiar with popular ML frameworks like Pytorch or Tensorflow.
Preferred:
- Proficient in building and maintaining systems written in modern languages (e.g., Golang, Python).
- Familiar with fundamental Deep Learning architectures such as Transformers and Encoder/Decoder models.
- Familiarity with Nvidia TensorRT-LLM, vLLM, DeepSpeed, Nvidia Triton Server.
- Experience writing custom CUDA kernels using CUDA or OpenAI Triton.
Benefits
- Base pay range between $171.6k and $302.2k, depending on skills and experience.
- Opportunity to participate in Apple’s discretionary employee stock programs.
- Comprehensive medical and dental coverage.
- Retirement benefits.
- Discounted products and free services.
- Reimbursement for certain educational expenses related to career advancement.
- Potential eligibility for discretionary bonuses or commission payments.
Apple
We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This is where your work can make a difference in people’s lives. Including your own. Apple is an equal opportunity employer that is committed to inclusion and diversity. Visit apple.com/careers to learn more.
San Francisco, CA