Senior / Staff Machine Learning Infra Engineer

Company: Basarako
Location: San Francisco
Posted on: February 1, 2025

Job Description:

Senior / Staff Machine Learning Infra EngineerPosition Title: Senior / Staff Machine Learning Infrastructure EngineerCompany Name: Waabi (US) Inc.Company Overview:Waabi (US) Inc. is a cutting-edge AI company founded by Raquel Urtasun, a pioneer and visionary in the field. Our team is dedicated to developing the next generation of self-driving technology, utilizing the power of AI to safely navigate real-world environments. With offices in Toronto, San Francisco, and Dallas, we are rapidly expanding and seeking diverse, innovative, and collaborative candidates to join our team. Learn more about us at .Position Overview:As a Senior / Staff Machine Learning Infrastructure Engineer at Waabi (US) Inc., you will work alongside a team of multidisciplinary engineers and research scientists to enable safe self-driving at scale. You will collaborate with cross-functional teams to understand the growing need for cloud usage and propose strategies for compute and data usage in training and simulation workloads. You will design and implement scalable and resilient cloud infrastructure, promote best practices for cloud usage, and oversee cloud strategies across the company. This role offers the opportunity to make a positive impact on the world through innovative solutions and collaboration.Key Responsibilities:

• Work with a team to enable safe self-driving at scale, using an AI-first approach.
• Collaborate with cross-functional teams to understand the growing need for cloud usage and propose strategies for compute and data usage in 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 usage in training and simulation environments, overseeing cloud strategies across the company.
• Read and develop production-quality software.
• Schedule jobs and allocate resources.
• Monitor and improve system performance.
• Work with containers and container orchestration.
• Provide support for Linux systems.
• Develop platform services for other teams to use.
• Work in an Agile/Scrum environment.Qualifications:
• Bachelor's or Master's/PhD in Computer Science or a related field, 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 with public cloud platforms, preferably AWS.
• Experience with infrastructure as code systems, preferably Terraform.
• Experience with job scheduling and resource allocation.
• Experience with containers and container orchestration, such as Docker, ECS, or Kubernetes.
• Comfortable working with Linux systems.
• Experience building platform services for other teams to use.
• Open-minded and collaborative team player with a willingness to help others.
• Passion for self-driving technologies, solving tough 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.
• Familiarity with common ML tools, workflows, and frameworks, such as Kubeflow or MLFlow.
• Understanding of system performance tuning at the software, hardware, and network levels.
• Knowledge of GPUs and accelerators in ML training and inference use cases.Compensation:The annual salary range for this role is $158,000 - $269,000 USD, in addition to competitive perks and benefits. Waabi (US) Inc. determines yearly salary ranges based on various factors and compensation practices. The salary base range reflects the minimum and maximum target for new hire salaries for this position across all US locations. Please note that the company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus.Perks and Benefits:
• Competitive compensation and equity awards.
• Health and wellness benefits, including medical, dental, and vision coverage (for full-time employees).
• Unlimited vacation.
• Flexible hours and work from home support.
• Daily drinks, snacks, and catered meals (when in the office).
• Regularly scheduled team building and social events, both on-site, off-site, and virtual.
• As we grow, this list will continue to evolve.Equal Opportunity Employer:Waabi (US) Inc. is an equal opportunity employer. We celebrate diversity and are committed to creating a supportive, inclusive, and accessible workplace for all our employees. We welcome applicants from all backgrounds and identities, including 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 also provide workplace accommodations for qualified individuals with disabilities, as required by applicable law. If you require reasonable accommodation during the job application or interview process, please contact our recruiting team.Join Our Team:At Waabi (US) Inc., we are building technologies to revolutionize the way the world moves. If you are passionate about making an impact and being a part of the future, join our talented team and help us make a difference.Apply link
  #J-18808-Ljbffr

Keywords: Basarako, Santa Rosa , Senior / Staff Machine Learning Infra Engineer, Engineering , San Francisco, California