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 WordPress Developer, Remote Job

    Work from home Full-time role

    Remote Data Entry Amazon Specialist – Part-Time, Work From Home, No Experience Required, Full Training Provided

    Work from home Full-time role

    Creative Designer (Graphic Design, Video Editing & AI Tools)

    Work from home Full-time role

    [Remote] Key Account Manager – Power Transmission

    Work from home Full-time role

    Part-Time Remote Data Entry Associate – Home‑Based Data Management & Accuracy Specialist at careerzynith

    Work from home Full-time role

    Experienced Customer Relationship Representative – Remote Seasonal (Full & Part Time positions available) for a Leading eCommerce Company

    Work from home Full-time role

    Remote Loan Signing Agent Columbus, OH

    Work from home Full-time role

    Overnight Online Assistant | $25–$35/hr | Global Remote Work Night Shift Opportunities

    Work from home Full-time role

    Marketing Coordinator: Content & Brand Campaigns (Remote)

    Work from home Full-time role

    Experienced Remote Office Clerk/Data Entry Specialist – Operations Support

    Work from home Full-time role

    MakeUseOf - Features Writer

    Work from home Full-time role

    [Remote] Healthcare Associate Attorney

    Work from home Full-time role

    Shopify Developer (E-commerce Focused) – Full Time (Remote)

    Work from home Full-time role

    Cisco Engineer

    Work from home Full-time role

    Service Mesh Engineer (Istio / Linkerd)

    Work from home Full-time role

    Virtual Critical Care Registered Nurse - Pulse Center - Full Time - days

    Work from home Full-time role