Robotics Machine Learning Engineer

2 days ago

Houston, TX, USAMid Level

Responsibilities

Contribute to Persona AI’s large-scale ML software stack across data, training, and evaluation.
Design and implement scalable ML software infrastructure and abstractions.
Collaborate with the ML team to plan and develop machine learning models.
Develop comprehensive test suites for correct and performant ML pipeline execution.
Use cloud platforms for compute capacity, data management, and task automation.
Manage data and training artifacts and build production-level software across ML components.
Collaborate in attracting, nurturing, and growing the Machine Learning and Data teams.

At least 3 years of experience applying machine learning directly to robotics.
Bachelor’s, Master’s, or PhD in Robotics, Computer Science, or a related field.
Experience with production ML software deployment.
Experience using cloud computing for model development and data storage, including AWS, Azure, or GCP.
Strong expertise with deep learning frameworks such as PyTorch, JAX, or TensorFlow.
Ability to write high-quality software in Python, C++, or Rust.
Collaborative approach and willingness to work in a fast-paced team environment.
Bonus: experience with robot simulation, locomotion, computer vision, robot hardware, and sensors.
Bonus: experience with CI/CD, build systems, and Docker.
Bonus: experience training models and deploying them to robots in production environments.

Competitive compensation, performance-based bonus, 100% employer-covered medical benefits for most healthcare plans, early-stage equity, and competitive PTO.
Company-wide paid winter break from December 24th through January 2nd.
Full-time exempt position based in Houston, Texas, with 10% travel.
Opportunity to work on humanoid robotics for heavy-industry environments alongside a collaborative team.