Apple · 7 hours ago
AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform u0026 Technology
Santa Clara, CA
Full-time
Onsite
Mid, Senior Level
$147K/yr - $272K/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’ll be responsible for collaborating with teams across Apple on ML workloads and driving the design and delivery of critical features to facilitate ML compute workloads.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Scope, architect, and deliver innovative high-quality solutions
Effectively communicate complex features and systems in detail
Collaborate with teams across Apple on ML workloads such training, inferencing, fine tuning etc
Understand industry & company-wide trends to help assess & develop new technologies
Code using Go, Python
Conduct code reviews
Help onboard new team members, provide mentorship and enable successful ramp up on your team's code bases
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/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
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