Machine Learning Engineer

Vayu Robotics
Vayu Robotics

Software Engineering

San Francisco, CA, USA

Posted on Mar 13, 2026
What You Will Do: Contribute to the development of ML models for autonomy. Build training pipelines involving simulated and real data on distributed GPU infrastructure. Develop a testing framework to measure model performance. Automate the generation of challenging scenarios using adversarial training. Required Skills: Habit of writing clean and efficient code. Experience with PyTorch. Understanding of the following ML concepts: Imitation learning, Reinforcement learning, Adversarial learning, Self-supervised learning.