3+ years of machine learning experience in robotics
Bachelor's, Master's, or PhD in Robotics, Computer Science, or related field
Strong problem-solving skills for large-scale engineering challenges
Experience with production ML software deployment
Familiarity with cloud computing platforms (AWS, Azure, GCP)
Proficient in deep learning frameworks (PyTorch, JAX, TensorFlow)
Skilled in programming languages such as Python, C++, or Rust
Responsibilities
Contribute to the ML software stack for data, training, and evaluation
Design and implement scalable ML software infrastructure
Collaborate with the ML team on developing machine learning models
Develop test suites for ML pipeline performance and correctness
Utilize cloud platforms for compute capacity and data management
Stay updated on advancements in ML research
Assist in growing the Machine Learning and Data teams
Benefits
Flexible work environment
Opportunities for professional growth and development
Collaborative team culture
Access to cutting-edge technology and tools
Engagement in impactful projects with real-world applications
Full Job Description
Job Title: Robotics Machine Learning Engineer
Department: Software Engineering
Employment Type: Full-Time
FLSA Status: Exempt
Location: Houston, TX
Travel: 10%
About the Role
We're looking for a Machine Learning Engineer to drive Persona's large-scale ML infrastructure and facilitate the robot learning effort. We're looking for exceptional talent who dream big, thrive on challenges, and love seeing their efforts make an impact on Persona's humanoid robots doing real work in real-world environments.
You will be working extensively on large-scale data pipelines, training infrastructure, and robot evaluation procedures in simulation and on hardware. You will help Persona AI scale the ML effort effectively by managing data and training artifacts, and building solid production-level software across various ML components.
If you want to build a robust infrastructure to scale robot learning on all fronts, this role is for you.
What You Will Be Doing
Contribute to Persona's large-scale ML software stack across data, training, and evaluation.
Design and implement the ML software infrastructure with thoughtful abstraction and scalability.
Work with the ML team to craft and execute a comprehensive plan for the development of machine learning models.
Develop comprehensive test suites to ensure correct and performant execution of the ML pipelines.
Leverage cloud platforms for flexible compute capacity, data management, and task automation.
Keep up to date with the state of the art in ML research and development.
Collaborate in attracting, nurturing, and growing the Machine Learning and Data teams.
What We Are Looking For:
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.
Courage and grit to tackle complex large-scale engineering problems for robotics.
Experience with production ML software deployment.
Experience with cloud computing to develop models, store data, etc. (AWS, Azure, GCP).
Strong expertise with deep learning frameworks (PyTorch, JAX, TensorFlow, etc.).
Enthusiasm for working collaboratively in a fast-paced team environment.
Capable of writing high-quality software in Python, C++, or Rust.
Bonus Skills:
Experience with other aspects of ML applied to robotics, including robot simulation, locomotion, computer vision, robot hardware, and sensors.
Experience with other aspects of software engineering, including CI/CD, build systems, and Docker.
Have trained models and deployed them to robots in production environments.
