ML Infra Engineer (TPU/Jax/Optimization)
About the Role
In this role you will help scale and optimize our training systems and core model code. You’ll own critical infrastructure for large-scale training, from managing GPU/TPU compute and job orchestration to building reusable and efficient JAX training pipelines. You’ll work closely with researchers and model engineers to translate ideas into experiments—and those experiments into production training runs. This is a hands-on, high-leverage role at the intersection of ML, software engineering, and scalable…
Unlock the apply link on every AI job
Browsing is free. Membership is $9 a year or $4.90 every three months and unlocks the apply link and the full description on every job. Cancel any time from your billing page.
- The apply link on every job, straight to the official posting
- Full job descriptions instead of the preview
- Every AI job, every day: thousands of listings from 1,000+ company career pages
- Save jobs and get daily alerts for your categories
Secure checkout by Dodo Payments·Cancel anytime from your billing page