AI Infrastructure Engineer

Figure is an AI Robotics company developing a general purpose humanoid. Our humanoid robot, Figure 02, is designed for commercial tasks and the home. We are based in Sunnyvale, CA and require 5 days/week in-office collaboration. It’s time to build.Figure’s vision is to deploy autonomous humanoids at a global scale. Our AI team is looking for an experienced Distributed Systems Engineer to take our AI infrastructure to the next level. This role is focused on building AI training and deployment, behavior evaluation, and data management software. The ideal candidate has experience building tools and infrastructure for a large-scale deep learning system.Responsibilities:Design and implement software tools used to collect and manage data, train deep neural networks and deploy them on humanoid robotsCollaborate with our AI and robotics engineers to identify software requirements and take the lead on implementing themRun and maintain reliable backend distributed systems at scaleCollaborate with our customers on defining and executing the software strategy for deploying humanoid robots in productionRequirements:Bachelor's or Master's degree in Computer Science, Robotics, Engineering, or a related fieldExperience with Python and an ML framework (PyTorch, JAX, TensorFlow, etc.Minimum of 4 years of professional, full-time experience building reliable backend systemsExperience with Linux and command line toolsExperience using and managing data stores (Postgres, MySQL, ElasticSearch, Redis, etc.)Bonus Qualifications:Experience managing HPC clusters for deep neural network trainingExperience 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.)Experience building data annotation and dataset management toolsThe US base salary range for this full-time position is between $140,000 - $220,000 annually.The pay offered for this position may vary based on several individual factors, including job-related knowledge, skills, and experience. The total compensation package may also include additional components/benefits depending on the specific role. This information will be shared if an employment offer is extended.