AI Training Infrastructure Engineer, Figure

Helix Team

$150-400k

AWS
Kubernetes
GCP
Python
Terraform
Ansible
Azure
Puppet
Chef
PyTorch
Mid and Senior level
San Francisco Bay Area

5 days a week in office (San Jose, CA)

Figure

AI robotics company

Open for applications

Figure

AI robotics company

201-500 employees

B2BArtificial IntelligenceManufacturingDeep TechRobotics

Open for applications

$150-400k

AWS
Kubernetes
GCP
Python
Terraform
Ansible
Azure
Puppet
Chef
PyTorch
Mid and Senior level
San Francisco Bay Area

5 days a week in office (San Jose, CA)

201-500 employees

B2BArtificial IntelligenceManufacturingDeep TechRobotics

Company mission

To expand human capabilities through advanced AI.

Role

Who you are

  • The ideal candidate has experience building tools and infrastructure for a large-scale deep learning system
  • Strong software engineering fundamentals
  • Bachelor's or Master's degree in Computer Science, Robotics, Engineering, or a related field
  • Experience with Python and PyTorch
  • Experience managing HPC clusters for deep neural network training
  • Minimum of 4 years of professional, full-time experience building reliable backend systems

Desirable

  • Experience managing cloud infrastructure (AWS, Azure, GCP)
  • Experience with job scheduling / orchestration tools (SLURM, Kubernetes, LSF, etc)
  • Experience with configuration management tools (Ansible, Terraform, Puppet, Chef, etc)

What the job involves

  • Our Helix team is looking for an experienced Training Infrastructure Engineer, to take our infrastructure to the next level
  • This role is focused on managing the training cluster, implementing distributed training algorithms, data loaders, and developer tools for AI researchers
  • Design, deploy, and maintain Figure's training clusters
  • Architect and maintain scalable deep learning frameworks for training on massive robot datasets
  • Work together with AI researchers to implement training of new model architectures at a large scale
  • Implement distributed training and parallelization strategies to reduce model development cycles
  • Implement tooling for data processing, model experimentation, and continuous integration

Share this job

View 15 more jobs at Figure

Company

Funding (last 2 of 4 rounds)

Feb 2024

$675m

SERIES B

Jul 2023

$9m

SERIES A

Total funding: $854m

Our take

Humanoid robots are quickly moving from sci-fi to reality, with big tech companies racing to bring them into workplaces, and soon, our homes. It's a booming market, and Figure is one of the big names in the game.

Figure builds human-like robots that are powered by advanced AI that can walk, see, and even understand natural language. Its latest model, the Figure 02, is already piloting in a BMW factory, and its in-house AI system, Helix lets these robots perform complex tasks from simple voice prompts. The company's moving fast, and it's not slowing down anytime soon.

But moving fast comes with baggage. It recently cut ties with OpenAI after claiming a major breakthrough in robot intelligence, and it's now working completely solo. Still, Figure just secured major commercial deals and may be headed for a jaw-dropping $40B valuation. It's a high-stakes field that's as exciting as it is controversial and Figure is certainly one to watch out for.

Freddie headshot

Freddie

Company Specialist at Welcome to the Jungle