Apple is looking for an experienced software engineer to build and maintain compute infrastructure to run ML workloads at Apple that include training, inferencing, fine tuning etc. The goal is to make ML workloads more stable, more reliable, more efficient and more cost effective.
Requirements
- 4+ years of hands-on experience in building scalable backend systems for training and evaluation of machine learning/deep learning models
- Proficient in relevant programming languages, like Python or Go
- Strong expertise in distributed systems, reliability and scalability, containerization, and cloud platforms.
- Proficient in cloud computing and data processing infrastructure and tools: Kubernetes, Ray, Beam, Flink
- Ability to clearly and concisely communicate technical and architectural problems, while working with partners to iteratively find solutions
- Proficient in working with and debugging accelerators, like: GPU, TPU, AWS Trainium
- Proficient in ML training and deployment frameworks, like: JAX, Tensorflow, PyTorch, TensorRT, vLLM
Responsibilities
- Collaborating with teams across Apple on ML workloads such as training, inferencing, fine tuning etc.
- Driving the design and delivery of critical features to facilitate ML compute workloads
- Effectively communicating complex features and systems in detail
- Understanding industry & company-wide trends to help assess & develop new technologies
- Scope, architect, and deliver innovative high-quality solutions
- Code using Go, Python
- Conduct code reviews
Other
- Bachelors in Computer Science, engineering, or a related field
- Help onboard new team members, provide mentorship and enable successful ramp up on your team's code bases
- Advance degrees in Computer Science, engineering, or a related field
- Apple is an equal opportunity employer that is committed to inclusion and diversity.
- Submit Resume
