Machine Learning Engineer
Vayu Robotics
Software Engineering
San Jose, 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 codeExperience with PyTorchUnderstanding of the following ML concepts:Imitation learningReinforcement learningAdversarial learningSelf-supervised learning