Demo

Senior / Staff Machine Learning Infra Engineer

Waabi
San Francisco, CA Full Time
POSTED ON 1/17/2025
AVAILABLE BEFORE 4/15/2025

Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving technology. With a world class team and an innovative approach that unleashes the power of AI to "drive" safely in the real world, Waabi is bringing the promise of self-driving closer to commercialization than ever before. Waabi is backed by best-in-class investors across the technology, logistics and the Canadian innovation ecosystem.

With offices in Toronto, San Francisco, and Dallas, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit : www.waabi.ai

You will...

  • Work alongside a team of multidisciplinary Engineers and Research Scientists using an AI-first approach to enable safe self-driving at scale.
  • Collaborate with cross-functional teams in the company to understand the growing need and pain points in cloud usage.
  • Propose cloud strategies around compute and data usages for training and simulation workloads.
  • Design and implement scalable and resilient cloud infrastructure optimized for long term reliability and adaptability.
  • Devise and promote best practices for cloud usages in training and simulation environments, oversee cloud strategies and usages across the whole company.

Qualifications :

  • BS, MS / PhD in Computer Science or similar technical field of study or equivalent practical experience.
  • 5 years of relevant industry experience.
  • Experience in reading and developing production quality software.
  • Deep understanding of Cloud compute and data storage for distributed training and inference workloads.
  • Familiarity with Python, GO, Rust or C ecosystems.
  • Experience working with public cloud platforms (AWS preferred).
  • Experience with infrastructure as code systems (Terraform preferred).
  • Experience in job scheduling and resource allocation.
  • Experience with containers and container orchestration (i.e., Docker, ECS, Kubernetes).
  • Experience and high level of comfort working with Linux systems.
  • Experience with building platform services that enable other teams to do their best work.
  • Open-minded and collaborative team player with the willingness to help others.
  • Passionate about self-driving technologies, solving hard problems, and creating innovative solutions.
  • Experience working in an Agile / Scrum environment.
  • Bonus / nice to have :

  • Experience with on-premise servers, network equipment and scale-out storage systems.
  • Experience with CI / CD pipelines and release management.
  • Experience in common ML tools, workflows and frameworks (i.e. systems like Kubeflow or MLFlow).
  • Understand system performance tuning at software, hardware, and network levels.
  • Have good understanding of GPUs and accelerators in ML training and inference use cases.
  • The US yearly salary range for this role is : $158,000 - $269,000 USD in addition to competitive perks & benefits. Waabi (US) Inc.'s yearly salary ranges are determined based on several factors in accordance with the Company's compensation practices. The salary base range is reflective of the minimum and maximum target for new hire salaries for the position across all US locations. Note : The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus.

    Perks / 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.
  • As we grow, this list continues to evolve!
  • Waabi is a technology start-up building technologies to transform the way the world moves. Join our talented team to be a part of the future and to make an impact!

    Waabi is an equal opportunity employer. We celebrate diversity and are committed to creating a supportive, inclusive, and accessible workplace for all our employees. We seek applicants of all backgrounds and identities, across race, color, ethnicity, national origin or ancestry, age, citizenship, religion, sex, sexual orientation, gender identity or expression, military or veteran status, marital status, pregnancy or parental status, caregiver status, disability, or any other characteristic protected by law. We make workplace accommodations for qualified individuals with disabilities as required by applicable law. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know.

    Salary : $158,000 - $269,000

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Senior / Staff Machine Learning Infra Engineer?

    Sign up to receive alerts about other jobs on the Senior / Staff Machine Learning Infra Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $129,363 - $167,316
    Income Estimation: 
    $145,845 - $177,256
    Income Estimation: 
    $147,836 - $182,130
    Income Estimation: 
    $154,597 - $194,610
    Income Estimation: 
    $86,891 - $130,303
    Income Estimation: 
    $129,363 - $167,316
    Income Estimation: 
    $145,845 - $177,256
    Income Estimation: 
    $147,836 - $182,130
    Income Estimation: 
    $154,597 - $194,610
    Income Estimation: 
    $86,891 - $130,303
    Income Estimation: 
    $81,253 - $112,554
    Income Estimation: 
    $89,966 - $112,616
    Income Estimation: 
    $95,407 - $122,738
    Income Estimation: 
    $103,114 - $138,258
    Income Estimation: 
    $86,891 - $130,303
    Income Estimation: 
    $154,597 - $194,610
    Income Estimation: 
    $172,688 - $210,712
    Income Estimation: 
    $170,589 - $211,671
    Income Estimation: 
    $178,619 - $225,190
    Income Estimation: 
    $86,891 - $130,303
    Income Estimation: 
    $103,114 - $138,258
    Income Estimation: 
    $118,163 - $145,996
    Income Estimation: 
    $120,777 - $151,022
    Income Estimation: 
    $129,363 - $167,316
    Income Estimation: 
    $86,891 - $130,303
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Waabi

    Waabi
    Hired Organization Address San Francisco, CA Intern
    Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving...
    Waabi
    Hired Organization Address San Francisco, CA Full Time
    Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving...
    Waabi
    Hired Organization Address San Francisco, CA Full Time
    Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving...
    Waabi
    Hired Organization Address San Francisco, CA Full Time
    Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving...

    Not the job you're looking for? Here are some other Senior / Staff Machine Learning Infra Engineer jobs in the San Francisco, CA area that may be a better fit.

    Staff Machine Learning Engineer

    Bobbie's Learning Center LLC, San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!