Staff ML Compute & TPU Infrastructure Engineer
appleMillbrae, CA
Staff ML Compute & TPU Infrastructure Engineer
appleMillbrae, CA
yesterday
Occupations
Computer Systems Engineers/ArchitectsComputer and Information Research ScientistsSoftware DevelopersIndustries
Computer Systems Design ServicesComputing Infrastructure Providers, Data Processing, Web Hosting, and Related ServicesSoftware PublishersAbout the role
Apple is seeking an AIML Staff ML Infrastructure Engineer to advance pre-training infrastructure in the SF Bay Area. You will work on TPUs, JAX/XLA, and distributed training to push performance and scalability across large foundation models. You will lead kernel development for attention/MoE, optimize workloads, and mentor engineers while collaborating with cross-functional teams to achieve engineering excellence.
Matching similar jobs
JOB OVERVIEW
Experience level
Lead
Location
Millbrae, CA
Occupation
Computer Systems Engineers/Architects
Industry
Computer Systems Design Services
Posted
yesterday
Tired of running searches?
Rank the roles you'd take once, and matches like these arrive on their own.
CREATE PROFILE (opens in a new tab)