All roles

Open role

[Remote] Senior / Staff Machine Learning Infrastracture Engineer

Remote · Japan Full-time

Note: The job is a remote job and is open to candidates in USA. Waabi is a leader in Physical AI, focused on transforming autonomous transportation. The role involves designing and implementing a machine learning platform, collaborating with data scientists, and optimizing model pipelines for performance and efficiency.

Responsibilities

  • Design, develop, and implement the machine learning platform for the continuous deployment and integration of machine learning models
  • Collaborate with data scientists and engineers to understand model requirements and optimize pipeline processes
  • Automate the training, testing and deployment processes for machine learning models
  • Continuously monitor and maintain model pipelines, ensuring optimal performance, accuracy and reliability
  • Optimize machine learning pipelines for scalability, efficiency and cost-effectiveness
  • Ensure compliance with security and data privacy standards in all MLOps activities

Skills

  • 3-5 years of experience supporting machine learning training platforms
  • Bachelor's degree in Computer Science, Data Science or a related field
  • Strong understanding of machine learning principles and model lifecycle management
  • Proficiency in programming languages such as Python, with hands-on experience in machine learning frameworks like TensorFlow or PyTorch
  • Experience with cloud platforms like AWS, Azure, or Google Cloud and their respective machine learning services
  • Familiarity with containerization and orchestration tools such as Kubernetes and Docker
  • Strong problem-solving skills and ability to troubleshoot complex issues
  • Experience with monitoring tools and practices for model performance in production
  • Ability to work collaboratively in cross-functional teams
  • Experience with infrastructure-as-code (IaC) tools such as Terraform or Crossplane
  • Knowledge of big data technologies like Apache Spark or Hadoop
  • Familiarity with data engineering practices and tools
  • Experience with A/B testing and model validation in production environments
  • Relevant MLOps certifications (e.g., AWS Certified Machine Learning – Specialty, DataRobot MLOps Certification) are a plus

Benefits

  • Competitive compensation and equity awards.
  • Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
  • Unlimited Vacation.
  • Flexible hours and Work from Home support.
  • Daily drinks, snacks and catered meals (when in office).
  • Regularly scheduled team building activities and social events both on-site, off-site & virtually.

Company Overview

  • Waabi develops autonomous driving technology and physical artificial intelligence systems for the transportation sector. It was founded in 2021, and is headquartered in Toronto, Ontario, CAN, with a workforce of 51-200 employees. Its website is https://waabi.ai.
  • Company H1B Sponsorship

  • Waabi has a track record of offering H1B sponsorships, with 1 in 2026. Please note that this does not guarantee sponsorship for this specific role.
  • More open positions

    [Remote] Principal Network Engineer - Telecom Focus

    Work from home Full-time role

    [Remote] Financial Analyst

    Work from home Full-time role

    [Remote] POS1368 - PRINCIPAL ANALYST HRIS- Oracle Cloud

    Work from home Full-time role

    [Remote] Accounts Receivable Associate

    Work from home Full-time role

    [Remote] Sr. Product Manager - Sexual Health

    Work from home Full-time role

    Senior Engineer – Remote Aircraft Design & Maintenance Support – Full‑Time – $21/hr – careerzynith

    Work from home Full-time role

    [Remote] AI Engineer (Python, LLM/Agentic AI)

    Work from home Full-time role

    Senior Engineering Manager, Search

    Work from home Full-time role

    Remote Customer Service Representative – Flexible Remote Schedule, Up to $19/hr, No Degree Required, Growth‑Focused Role

    Work from home Full-time role

    Principal Customer Success Executive - Retail and Hospitality

    Work from home Full-time role

    Manager Trauma System Registry

    Work from home Full-time role

    [Work From Home] RN Post Anesthetic Care - $2,354 per week

    Work from home Full-time role

    Experienced Remote Data Entry Specialist – Unlock Your Potential with a Flexible Career Opportunity at careerzynith

    Work from home Full-time role

    Remote Part-Time Data Entry Specialist – Healthcare Records Management | Work From Home Opportunity at $30/Hour

    Work from home Full-time role

    Desenvolvedor PL

    Work from home Full-time role

    Product Manager Pleno - Delivery Intelligence Group | Tech

    Work from home Full-time role

    Registered Nurse- PRN Inpatient Virtual Support (Located Crystal Terrace)

    Work from home Full-time role

    Entry Level Logistics Data Entry Coordinator – Launch Your Career in Global Shipping Operations

    Work from home Full-time role

    Backend Engineer [Mid-Level]

    Work from home Full-time role

    Head of Microgrid Controls

    Work from home Full-time role

    Ingénieur·e en apprentissage automatique, II

    Work from home Full-time role