appleMillbrae, CA
Staff ML Compute & TPU Infrastructure Engineer
Staff ML Compute & TPU Infrastructure Engineer
Staff ML Compute & TPU Infrastructure Engineer
appleMillbrae, CA
yesterday
Computer Systems Engineers/ArchitectsComputer and Information Research ScientistsSoftware Developers
Computer Systems Design ServicesComputing Infrastructure Providers, Data Processing, Web Hosting, and Related ServicesSoftware Publishers
Apply for this role (opens in a new tab)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.
Also on the board Same function, level within a rung
Level
Lead
Location
Millbrae, CA
Occupation
Computer Systems Engineers/Architects
Industry
Computer Systems Design Services
Posted
yesterday