Apple · 1 day ago
AIML - Senior ML Infrastructure Engineer, ML Platform & Technologies - ML Compute
San Francisco, CA
Full-time
Onsite
Mid, Senior Level
$181K/yr - $318K/yr
4+ years expApple is where individual imaginations gather together, committing to the values that lead to great work. As a senior engineer on the ML Compute team, you will drive large-scale training initiatives and operationalize ML workloads while enhancing distributed cloud training techniques.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Drive large-scale training initiatives to support our most complex models
Operationalize large-scale ML workloads on Kubernetes
Enhance distributed cloud training techniques for foundation models
Design and integrate end-to-end lifecycles for distributed ML systems
Develop tools and services to optimize ML systems beyond model selection
Architect a robust MLOps platform to support seamless ML operations
Collaborate with cross-functional engineers to solve large-scale ML training challenges
Research and implement new patterns and technologies to improve system performance, maintainability, and design
Lead complex technical projects, defining requirements and tracking progress with team members
Mentor engineers in areas of your expertise, fostering skill growth and knowledge sharing
Cultivate a team centered on collaboration, technical excellence, and innovation
Qualification
Required
Bachelors in Computer Science, engineering, or a related field
4+ years of hands-on experience in building scalable backend systems for training and evaluation of machine 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 infrastructure and tools: Kubernetes, Ray, PySpark
Ability to clearly and concisely communicate technical and architectural problems, while working with partners to iteratively find solutions
Preferred
Advance degrees in Computer Science, engineering, or a related field
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
Benefits
Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation
Company
Apple
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
10001+ employees
H1B Sponsorship
Apple has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)
Funding
Current Stage
Public CompanyTotal Funding
$5.67BKey Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity
Leadership Team
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase