> Markdown version of [/jobs/ext/151547-aiml-staff-ml-infrastructure-engineer](https://www.wearedevelopers.com/jobs/ext/151547-aiml-staff-ml-infrastructure-engineer). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# AIML - Staff ML Infrastructure Engineer

- **Company:** Apple Inc.
- **Location:** San Francisco, CA, United States
- **Contract:** Permanent contract
- **Skills:** Amazon Web Services, Cloud Computing, Software Debugging, Distributed Systems, Python (Programming Language), Machine Learning, Tensorflow, Cloud Platform System, Pytorch, Backend, Containerization, Pyspark, Kubernetes, Information Technology, TensorRT, Programming Languages
- **Published:** May 17, 2026
- **Apply:** https://www.dice.com/job-detail/4bf66080-7bc6-4de8-b64d-71753e1f48ab

## About the Role

Bachelors in Computer Science, engineering, or a related field\n6+ years of hands-on experience in building scalable backend systems for training and evaluation of machine learning models\nProficient in relevant programming languages, like Python or Go\nStrong expertise in distributed systems, reliability and scalability, containerization, and cloud platforms\nProficient in cloud computing infrastructure and tools: Kubernetes, Ray, PySpark\nAbility to clearly and concisely communicate technical and architectural problems, while working with partners to iteratively find

Advance degrees in Computer Science, engineering, or a related field\nProficient in working with and debugging accelerators, like: GPU, TPU, AWS Trainium\nProficient in ML training and deployment frameworks, like: JAX, Tensorflow, PyTorch, TensorRT, vLLM



## Related Videos

- [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning)
- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [Developing the Backend with Stefan Lingler, CTO at Shpock](https://www.wearedevelopers.com/videos/100360-developing-the-backend-with-stefan-lingler-cto-at-shpock)
- [PySpark - Combining Machine Learning & Big Data](https://www.wearedevelopers.com/videos/44-pyspark-combining-machine-learning-big-data)
- [Empowering Retail Through Applied Machine Learning](https://www.wearedevelopers.com/videos/976-empowering-retail-through-applied-machine-learning)
- [Serverless deployment of (large) NLP models ](https://www.wearedevelopers.com/videos/158-serverless-deployment-of-large-nlp-models)

## Related Articles

- [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)
- [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development)
- [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models)
- [MLOps – What’s the deal behind it?](https://www.wearedevelopers.com/magazine/125-mlops-what-s-the-deal-behind-it)
- [Dev Digest 120 - Apple and peers](https://www.wearedevelopers.com/magazine/455-dev-digest-120-apple-and-peers)
- [ Dev Digest 213: Petrol Prices, Agentic Workflows, AI Skills and CODE100!](https://www.wearedevelopers.com/magazine/718-dev-digest-213-petrol-prices-agentic-workflows-ai-skills-and-code100)