Software Engineer, Machine Learning Platform

Leading developer of Embodied AI technology creating autonomous driving systems using advanced AI software and foundation models.
$120,000 - $180,000
Machine Learning
Senior Software Engineer
In-Person
101 - 500 Employees
5+ 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 are revolutionizing the automotive industry. As a Software Engineer on the ML Platform team, you'll play a crucial role in building and maintaining the machine learning training infrastructure that powers our autonomous driving systems.

The position offers unique opportunities to work with cutting-edge technology, focusing on distributed training of large models and optimization of ML infrastructure. You'll collaborate directly with machine learning research engineers to enhance model efficiency and reduce training costs. The role involves working with sophisticated tools like NVIDIA NSight and managing thousands of GPUs for our driving simulator.

Wayve's culture emphasizes diversity, inclusion, and mutual respect, creating an environment where everyone's unique perspectives are valued. The company's mission to "create autonomy that propels the world forward" drives innovation in automated driving systems for automakers. They embrace challenging problems and maintain a learning-oriented mindset while pursuing excellence.

This role is perfect for experienced software engineers passionate about infrastructure and ML optimization. You'll have the opportunity to impact the future of autonomous driving while working with state-of-the-art technology and a talented team. The position offers growth opportunities, especially in model optimization and distributed systems, making it an ideal next step for those looking to advance their careers in ML infrastructure.

Last updated 3 minutes 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
  • Develop tooling for GPU resource management and simulation
  • Ensure infrastructure reliability and efficient utilization

Requirements For Software Engineer, Machine Learning Platform

Python
Kubernetes
  • Minimum of 5 years experience in Software Engineering, preferably ML Infrastructure
  • Proficiency in Python
  • Knowledge of software engineering best practices
  • Experience with concurrent, parallel and distributed computing
  • Experience with cloud infrastructure, preferably Azure
  • Experience with Docker, Kubernetes and Terraform
  • Experience with ML frameworks (Pytorch, Tensorflow, ONNX, TensorRT) preferred
  • Experience profiling and optimizing ML models preferred

Benefits For Software Engineer, Machine Learning Platform

Equity
  • Inclusive work environment
  • Career development opportunities
  • Work with cutting-edge technology

Interested in this job?

Jobs Related To Wayve Software Engineer, Machine Learning Platform

Senior Machine Learning Platform Engineer

Senior Machine Learning Platform Engineer optimizing large-scale AI training for autonomous vehicles

Machine Learning Performance Engineer

Wayve seeks a Machine Learning Performance Engineer to optimize large-scale AI training for autonomous vehicles.

Senior Machine Learning Engineer

Senior Machine Learning Engineer role at Wayve, developing cutting-edge AI solutions for autonomous driving in London.

Software Engineer - Anthropic Labs

Software Engineer role at Anthropic Labs focusing on prototyping and evaluating emerging AI capabilities.

AI Engineer

Senior AI Engineer role at Capco focusing on GenAI solutions development and deployment for financial services industry.