Software Engineer, Machine Learning Platform

Leading developer of Embodied AI technology creating advanced AI software and foundation models for autonomous vehicles.
Machine Learning
Senior Software Engineer
In-Person
101 - 500 Employees
4+ years of experience
AI · Automotive

Description For Software Engineer, Machine Learning Platform

Wayve, founded in 2017, is at the forefront of autonomous vehicle technology, developing Embodied AI solutions that enable vehicles to navigate complex environments safely and efficiently. As a Software Engineer in the Machine Learning Platform team, you'll be instrumental in building and maintaining the training infrastructure that powers their AI systems. The role combines software engineering excellence with machine learning optimization, working directly with research engineers to enhance model training efficiency and performance. You'll be part of a dynamic team that manages thousands of GPUs for their driving simulator and implements cutting-edge model optimization techniques. The position offers opportunities to work with state-of-the-art technology while contributing to the future of autonomous driving. Wayve values diversity and inclusion, fostering an environment where different perspectives are embraced and every team member's contribution matters. The company's mission to create autonomy that propels the world forward offers an exciting opportunity to work on challenging problems with real-world impact. This role is perfect for someone who combines infrastructure expertise with a passion for machine learning and autonomous systems.

Last updated 5 days ago

Responsibilities For Software Engineer, Machine Learning Platform

  • Build and maintain machine learning training infrastructure
  • Work with ML research engineers to optimize models for efficient training
  • Manage training job scheduling and orchestration
  • Provide GPU infrastructure support for driving simulator
  • Optimize ML models using tools like NVIDIA NSight

Requirements For Software Engineer, Machine Learning Platform

Python
Kubernetes
  • Minimum of 4 years experience in a similar field
  • Proficiency in Python
  • Knowledge of software engineering practices
  • Experience with concurrent, parallel and distributed computing
  • Experience with cloud infrastructure, preferably Azure
  • Experience with Docker, Kubernetes and Terraform
  • Experience working with ML frameworks (desired)
  • Experience profiling and optimizing ML models (desired)

Benefits For Software Engineer, Machine Learning Platform

Visa Sponsorship
  • Visa Sponsorship available

Interested in this job?

Jobs Related To Wayve Software Engineer, Machine Learning Platform

Machine Learning Engineer, Spatial Intelligence

Senior Machine Learning Engineer position at Wayve focusing on 3D vision and spatial intelligence for autonomous driving systems, requiring 3+ years of experience.

Machine Learning Engineer, Spatial Intelligence

Senior Machine Learning Engineer position at Wayve, focusing on 3D spatial intelligence and computer vision for autonomous driving systems.

Senior Machine Learning Engineer, Foundation Models

Senior Machine Learning Engineer position at Wayve, focusing on developing foundation models and AI systems for autonomous vehicles, requiring 7+ years of experience in AI research and robotics.

Senior Software Engineer - Windows AI Agent

Senior Software Engineer position at Microsoft focusing on Windows AI Agent development, specializing in scalable model infrastructure and cloud-based AI workflows.

Machine Learning Engineer

Senior Machine Learning Engineer role at Adobe, developing innovative ML models and deploying AI solutions for the Digital Experience platform. Salary range: $120,700-$228,600.