[Remote] Senior / Staff Machine Learning Infrastracture Engineer

Note: The job is a remote job and is open to candidates in USA. Waabi is a leader in Physical AI, focused on unlocking the next era of autonomous transportation. The role involves designing, developing, and implementing a machine learning platform for continuous deployment and integration of machine learning models, while collaborating with data scientists and engineers to optimize processes.ResponsibilitiesDesign, develop, and implement the machine learning platform for the continuous deployment and integration of machine learning modelsCollaborate with data scientists and engineers to understand model requirements and optimize pipeline processesAutomate the training, testing and deployment processes for machine learning modelsContinuously monitor and maintain model pipelines, ensuring optimal performance, accuracy and reliabilityOptimize machine learning pipelines for scalability, efficiency and cost-effectivenessEnsure compliance with security and data privacy standards in all MLOps activitiesSkills3-5 years of experience supporting machine learning training platformsBachelor's degree in Computer Science, Data Science or a related fieldStrong understanding of machine learning principles and model lifecycle managementProficiency in programming languages such as Python, with hands-on experience in machine learning frameworks like TensorFlow or PyTorchExperience with cloud platforms like AWS, Azure, or Google Cloud and their respective machine learning servicesFamiliarity with containerization and orchestration tools such as Kubernetes and DockerStrong problem-solving skills and ability to troubleshoot complex issuesExperience with monitoring tools and practices for model performance in productionAbility to work collaboratively in cross-functional teamsExperience with infrastructure-as-code (IaC) tools such as Terraform or CrossplaneKnowledge of big data technologies like Apache Spark or HadoopFamiliarity with data engineering practices and toolsExperience with A/B testing and model validation in production environmentsRelevant MLOps certifications (e.g., AWS Certified Machine Learning – Specialty, DataRobot MLOps Certification) are a plusBenefitsCompetitive 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 OverviewWaabi 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 SponsorshipWaabi has a track record of offering H1B sponsorships, with 1 in 2026, 8 in 2025, 6 in 2024, 6 in 2023, 4 in 2022. Please note that this does not guarantee sponsorship for this specific role.