Helix AI Engineer, Training Performance
San Joseonsitemid$200K – $400K
via Greenhouse
About this role
Figure is an AI robotics company developing autonomous general-purpose humanoid robots. The goal of the company is to ship humanoid robots with human level intelligence. Its robots are engineered to perform a variety of tasks in the home and commercial markets. Figure is headquartered in San Jose, CA. Figure's vision is to deploy autonomous humanoids at a global scale. Our Helix team is looking for an experienced AI Training Performance Engineer to take our model training to the next level. This role is focused on improving distributed training frameworks for large scale model training, optimizing GPU kernels, exploring the relative gains of different accelerator types and co-designing our models to maximize utilization of our hardware.…
What we'd score you on
reqspace match rubricFive dimensions, recruiter-grade. Upload your resume and we'll generate a written explanation of where you fit and where the gaps are.
1
Skills match
For this role: python, c++, pytorch, jax
2
Level fit
This role is mid-level. We check your trajectory against it.
3
Domain experience
Your work in the role's domain matters more than your years total. We weight recent and direct experience.
4
Recency
A skill you used last quarter weighs more than one from five years ago. We grade on recency, not lifetime.
5
Location fit
This role is based in San Jose. We weight your proximity and willingness to relocate.
Score yourself on this role.
Free · no card · written explanation included
Skills in this role
Pulled from the job description. These are the keywords we'll weight when scoring your fit.
pythonc++pytorchjax
More at Figureai
- View →Helix AI Engineer, Video PretrainingSan Jose, CA
- View →Manufacturing Equipment TechnicianSan Jose, CA
- View →Helix AI Engineer, BackendSan Jose, CA
- View →Systems Integration Engineer - Hand SubsystemSan Jose, CA
- View →AI Data Operations ManagerUnited Arab Emirates
- View →Recruiting CoordinatorSan Jose, CA