Staff ML Software Engineer: Robotics & Foundation Models

Icehouseventures; Sunnyvale (CA); On-site; USD 311,000 - 419,000; Full time

Benefits offered by this job

Equity package

Job summary

Wayve in Sunnyvale, CA is seeking an ML Software Engineer to join the MEGA team as a founding member, building foundation models for general‑purpose robots and scalable ML infrastructure. You will design and maintain pipelines, data and training infra, and reusable software that researchers rely on; collaborate with engineers and scientists to translate research into reliable experiments.

Hybrid work with a competitive equity package; salary range from $311,000 to $419,000, dependent on

Qualifications

• Strong software engineering skills.
• Experience building ML pipelines or infrastructure (data ingestion, training, evaluation, or experiments).
• Experience designing reliable, reusable software systems.

Responsibilities

• Design, build, and maintain scalable ML pipelines for data ingestion, model training, evaluation, and related research workflows.
• Build software systems that scale ML workloads to larger datasets, larger models, and more experiments.
• Develop clean and reusable interfaces between data, models, training, evaluation, and downstream robotics workflows.
• Own and improve the health of the MEGA codebase, including architecture, testing, reliability, maintainability, and standards.
• Identify and resolve performance, reliability, and usability bottlenecks across ML workflows.
• Build and maintain infrastructure that supports multiple researchers and ML projects without slowing iteration.
• Work with researchers to translate requirements into practical software solutions.
• Build and use distributed training and data-processing pipelines for large models and multimodal datasets.

Skills

Software engineering

ML pipelines

Software infrastructure

Software design

ML frameworks

Debugging

Testing

Large datasets / models

Collaboration with researchers

Tools

PyTorch

TensorFlow

Job description

Wayve in Sunnyvale, CA is seeking an ML Software Engineer to join the MEGA team as a founding member, building foundation models for general‑purpose robots and scalable ML infrastructure. You will design and maintain pipelines, data and training infra, and reusable software that researchers rely on; collaborate with engineers and scientists to translate research into reliable experiments.

Hybrid work with a competitive equity package; salary range from $311,000 to $419,000, dependent on