Machine Learning Engineer, Foundation Models - AI, Search & Knowledge Platforms
$171600/y-$302200/y
Applea day ago
Highlight
OnsiteFull-timeSenior
Summary
Apple is seeking a Machine Learning Engineer, Foundation Models to join their team in Seattle, WA. The role involves optimizing inference for cutting-edge model architectures and building production-grade solutions for real-time model serving.
About the Role
As a Machine Learning Engineer, you will work alongside the Foundation Model Research team to enhance the performance of large-scale models. Your responsibilities will include collaborating with product teams to develop solutions that serve millions of customers, identifying bottlenecks in inference across various hardware, and mentoring other engineers in the organization.
About You
Required:
- 5+ 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 one of the 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.
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.
Seattle, WA