Persona AI is developing and commercializing rugged, multi-purpose humanoid robots that perform real work. The company is looking for a Machine Learning Intern to help build and improve their machine learning models and infrastructure for these robots.
Requirements
- Experience with deep learning frameworks (Pytorch, JAX, TensorFlow, etc.)
- Experience with cloud computing to develop models, manage data, etc. (AWS, Azure, GCP)
- Strong understanding of the state of the art research in robot learning (behavior cloning and reinforcement learning for manipulation, world models, etc.).
- Understanding of the challenges of deploying neural network models in the real world.
- Capable of writing high-quality software.
Responsibilities
- Work with the ML team to develop and improve machine learning models and the infrastructure.
- Monitor and evaluate the performance of models in the real world.
- Collaborate on the design and development of the Persona ML software stack and support its application in manipulation, navigation, locomotion, and perception.
Other
- Courage and grit to tackle some of the hardest problems in embodied AI.
- Enthusiasm for working collaboratively in a high paced team environment.
- Thrive in fast-paced and ambiguous environments.
- Strong first principles thinker.
- An advanced degree (Master's or PhD) in computer science, robotics, machine learning, or another related field.
- Published papers at top ML/Robotics conferences (ICML, ICRA, CoRL, RSS, NeurIPS).