Robotics Machine Learning Engineer

Persona AI Inc is hiring a Robotics Machine Learning Engineer in Houston, TX to support the robot learning effort through large-scale machine learning infrastructure. The role emphasizes data pipelines, training infrastructure, and robot evaluation across simulation and on-hardware execution.

Responsibilities

• Contribute to Persona’s large-scale ML software stack spanning data, training, and evaluation.
• Design and implement ML software infrastructure using clear abstraction and an emphasis on scalability.
• Partner with the ML team to define and carry out a comprehensive plan for developing machine learning models.
• Create thorough test suites to support correct and performant execution of ML pipelines.
• Use cloud platforms to enable flexible compute capacity, data management, and task automation.
• Stay current with advances in ML research and development.
• Collaborate in attracting, nurturing, and growing the Machine Learning and Data teams.

Requirements

• 3+ years of experience in machine learning directly applied to robotics.
• Bachelor’s, Master’s, or PhD in Robotics, Computer Science, or a related field.
• Experience with production ML software deployment.
• Cloud computing experience for model development, data storage, and related workflows (AWS, Azure, GCP).
• Strong deep learning framework expertise, including PyTorch, JAX, and/or TensorFlow.
• Enthusiasm for collaborative work in a fast-paced team environment.
• Ability to write high-quality software in Python, C++, or Rust.

Technology

• AWS, Azure, GCP
• PyTorch, JAX, TensorFlow
• Python, C++, Rust

Bonus Skills

• Experience with additional ML areas applied to robotics, including robot simulation, locomotion, computer vision, robot hardware, and sensors.
• Experience across other software engineering areas such as CI/CD, build systems, and Docker.
• Experience training models and deploying them to robots in production environments.

Benefits

• Competitive compensation
• Performance-based bonus
• 100% employer covered medical benefits for most healthcare plans
• Early-stage equity
• Competitive PTO
• Company-wide paid winter break between December 24th and January 2nd

Travel, Employment, and Location

• Location: Houston, TX (onsite)
• Employment type: Full-Time
• Department: Software Engineering
• FLSA status: Exempt
• Travel: 10%