More jobs:
Senior ML Infrastructure Engineer — TPU & JAX
Job in
California, Moniteau County, Missouri, 65018, USA
Listed on 2026-09-04
Listing for:
Socket.dev
Full Time
position Listed on 2026-09-04
Job specializations:
-
Software Development
AI Reliability/ Performance Engineer
Job Description & How to Apply Below
Location: California
Apple is seeking a senior ML systems engineer to drive optimization for large-scale foundation model training on TPUs. You will profile and optimize JAX/XLA workloads across compute, memory, and communication, and develop high-performance TPU kernels for core ML operations.
The role requires 6+ years in high-performance ML or distributed systems, strong Python skills, and expertise in distributed systems and performance optimization.
#J-18808-LjbffrPosition Requirements
10+ Years
work experience
To View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×