Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Software Engineer, ML Infra

$180k - $225k

NewsBreak

Machine Learning Infrastructure Engineer

We're hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the models behind our Ads and Recommendations products. You'll join a small, high-ownership team that ships platform improvements end-to-end—partnering with product and data teams, reducing latency and cost, and shortening the path from an idea to a safely launched model.

You'll work across the ML lifecycle: making training faster and more reliable, improving model serving performance, and strengthening our feature/embedding platform so models stay fresh and consistent between offline and online use. We're looking for someone who can take real ownership, finish what's started, and raise the bar on stability and developer experience.

If you like building reliable systems that make ML teams move faster—and you enjoy turning complexity into simple, durable solutions—we'd love to talk.

Responsibilities
  • Design and develop machine learning infrastructure.
  • Own and enhance core components of the ML infrastructure, including systems for offline and online model training, model pipeline health monitoring, model serving, feature authoring, and feature serving.
  • Proactively address ML infrastructure issues that may impact production.
  • Collaborate with ML engineers to build robust model pipelines utilizing the ML infrastructure.
Requirements
  • Education: Bachelor's degree in a related field with 5+ years of relevant experience, or MS/PhD in a related field with 3+ years of relevant experience
  • Experience delivering production systems that support ML use cases (training or serving).
  • Proven track record in building and maintaining large-scale distributed backend systems
  • Proficient in Python, with a strong understanding of object-oriented languages such as C++ or Java
  • Strong problem solving skills with good teamwork and communication skills
Preferred Qualifications
  • Experience leading cross-team projects from design through stable rollout.
  • Experience iterating on ML models and shipping them to production.
Benefits

We offer competitive benefits package:

  • Health, dental, and vision care for you and your family
  • Top-tier 401(K) plan with company matching
  • Paid time off and paid holidays
  • Paid parental leave
  • FSA and commuter benefits programs
  • Team activity budget

The US base salary range for this full-time position is listed below. Pay may vary based on a number of factors including job-related skills, level, experience, geographic location and relevant education or training. At NewsBreak, we design our overall rewards package to attract top talents. Depending on the position, the role may also be eligible for discretionary bonus and options. Your recruiter can share more details during the hiring process.

Annual Base Pay Range

$180,000 - $225,000 USD

Vacancy posted 3 hours ago
Similar jobs that could be interesting for youBased on the Software Engineer, ML Infra in Mountain View, CA vacancy
  • $180k - $225k

     ...visit About the role:We’re hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the...  ...from an idea to a safely launched model.You’ll work across the ML lifecycle: making training faster and more reliable, improving... 
    Suggested
    Full time
    Local area
    Work from home

    News Break

    Mountain View, CA
    4 days ago
  • $147k - $210k

     ...computing concepts, consensus algorithms (e.g., Paxos).Google's software engineers develop the next-generation technologies that change how...  ...performant bulk data loading pathways(essential for emerging AI/ML workloads), introducing advanced mutation semantics and data life... 
    Suggested

    Google

    Sunnyvale, CA
    4 days ago
  • $272k - $431.25k

    We are seeking a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to join our Hardware Infrastructure team. As an Engineer, you will have a pivotal role in enhancing efficiency for our researchers by implementing progressions throughout the entire stack... 
    Suggested
    Full time

    Nvidia

    Santa Clara, CA
    1 day ago
  • $207k - $300k

     ...Dev and AI Infra Software Engineer, Release and Test Infra, XR Share Dev and AI Infra Software Engineer, Release and Test Infra, XR corporate...  ..., development, and architecture. ~ Experience applying AI/ML techniques to software engineering problems. Preferred... 
    Suggested

    Google LLC

    San Jose, CA
    1 day ago
  • $165.2k - $223.6k

     ...responsible for delivering high quality and fresh ML model training data, and providing...  ...station.Key job responsibilitiesAs an ML Engineer you will:Lead development of services and...  ...3+ years of non-internship professional software development experience- 2+ years of non-... 
    Suggested
    Internship
    Local area
    Worldwide
    Flexible hours

    Amazon

    Palo Alto, CA
    17 hours ago
  • $275.8k - $340.5k

     ...self-driving vehicles, join us. About the team: The AV ML Infra team at GM builds ML infrastructure designed to meet the unique...  ...efficient ML experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting-edge ML techniques. Our... 
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    6 hours ago
  • $147k - $210k

     ...Implement solutions in one or more specialized ML areas, utilize ML infrastructure, and...  ...accessible technologies.Google's software engineers develop the next-generation technologies...  ...businesses at Google. The YouTube Ads ML Infra team empowers YouTube Ads ML advancement... 

    Google

    Mountain View, CA
    3 days ago
  • $193.93k - $291.15k

     ...team of high-output generalists where ML and systems engineering converge. This is not a "run the models...  ...eval autonomy path, across ML and infra boundaries.About YouBS in Computer Science...  ...area.4+ years of industry software engineering experience with Python fluency... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $193.3k - $261.5k

     ...Senior Software Development Engineer, AWS Infra Supply Chain Automation Job ID: 10489640 | Amazon.com Services LLC AWS Infrastructure Services owns the design, planning, delivery, and operation of all AWS global infrastructure. In other words, we’re the people who... 
    Internship
    Local area
    Flexible hours

    Amazon

    Sunnyvale, CA
    17 hours ago
  •  ...role and team We are looking for senior software engineers to build the backbone of Neural...  ...generation, and log augmentation, along with the ML pipelines used to train and validate...  ...distributed services Collaborate closely with Infra, Autonomy, Research and other product... 
    Full time
    For contractors
    For subcontractor
    Casual work
    Work at office
    Remote work
    Day shift

    Decisive Point

    Sunnyvale, CA
    2 days ago
  • $180k - $210k

     ...work across distributed systems, production observability, and ML infra integrations to deliver an experience that feels instant,...  ...most impactful roadmap investments. You are: ~3+ years of software engineering experience building production distributed systems or cloud-... 
    Work at office
    Home office
    Flexible hours

    Glean.info

    Mountain View, CA
    4 days ago
  • $209k - $313k

     ...and other digital services.Snap Engineering teams build fun and...  ...critical role in scaling our ML Infrastructure, optimizing AI...  ...impactful.We’re looking for a Software Engineer to join the ML Platform...  ...deprecation. With the significant infra cost of this area, our team also... 
    Live in
    Work at office
    Local area

    Snapchat

    Palo Alto, CA
    17 hours ago
  • $206.5k - $258.1k

     ...future generations. Role Summary Rivian Autonomy is building an ML Infrastructure team to give hundreds of ML engineers a training platform they can trust at fleet scale. We are seeking a Staff Software Engineer to help design and build the platform that trains and... 
    Full time
    Contract work
    Temporary work
    Part time
    Local area
    Shift work

    Rivian

    Palo Alto, CA
    17 hours ago
  • $70 per hour

     ...and tens of billions in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous driving technology...  ...us build the data platform and infrastructure to support ML launch evals for the Waymo Driver. Waymo interns partner with... 
    Hourly pay
    Full time
    Internship
    Summer internship
    Relocation package

    Waymo

    Mountain View, CA
    1 day ago
  • $174k - $252k

     ...visualization workflows.Support new and exciting ML paradigms (such as horizontal scaling for...  ...experience.5 years of experience with software development in one or more programming...  ...open-source projects.Google's software engineers develop the next-generation technologies... 

    Google

    Sunnyvale, CA
    2 days ago
  • $174k - $252k

     ...Senior Software Engineer, Performance, SOC Architect Share Senior Software Engineer, Performance, SOC Architect corporate_fare Google place...  ...and a passion for developer productivity. ~ Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML... 

    Google LLC

    Sunnyvale, CA
    4 days ago
  • $164k - $282k

     ...workflow service at scale.We are looking for a visionary Staff Software Engineer to lead the architectural design and implementation of our...  ...always on for our customers.AI-Driven Automation: Integrate AI/ML models to automate system functionality, including predictive... 
    Temporary work

    Coupang

    Mountain View, CA
    2 days ago
  • $120k - $195k

     ...’s AI model training, feature engineering and serving with hundreds of billions...  ...scale feature engineering infra for all AI use cases from...  ...frameworks, data infra, compute software, and hardware to harness the power...  ...with hundreds of new ML models per quarter using thousands... 
    For contractors
    Work experience placement
    Work at office
    Flexible hours

    Linkedin

    Mountain View, CA
    1 day ago
  • $117k - $234k

     ...Role Summary:We are seeking a talented and passionate Senior, Software Engineer (Backend) for our team. You will be part of Catalog Spec Engineering...  ..., and problem solving.Experience working with generated AI/ML technologies.Experience with NoSQL databases like Cosmos DB,... 
    Full time
    Temporary work
    Part time
    Work at office

    Walmart

    Sunnyvale, CA
    2 days ago
  •  ...page, in any of our markets, a decision engine works behind the scenes to choose the right...  ...Airflow) that feed our algorithms and AI/ML models. Partner with data scientists...  ...ll Bring: ~5+ years building production software, including 3+ years in Java and/or Python... 
    Full time
    Remote work

    Walmart Inc.

    Sunnyvale, CA
    3 days ago
  •  ...Full Stack Engineer – Digital Payments About Gen: Gen is a global company dedicated to powering Digital Freedom through its trusted...  ...build intelligent integrations is a strong plus. Exposure to AI/ML concepts and the ability to evaluate and integrate AI‑driven... 
    Work experience placement
    Flexible hours

    Gen Digital Inc.

    Mountain View, CA
    17 hours ago
  • $125k - $175k

     ...hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains,...  ...launched model. You'll work across the ML lifecycle: making training faster and...  ...related discipline. Familiarity with software development processes, including version... 
    Full time
    Work experience placement
    Local area

    NewsBreak

    Mountain View, CA
    28 days ago
  •  ...you, we'd love to talk. Role Overview: As a Full Stack Software Engineer, you will be a pivotal force in developing, deploying, and...  ...expertise in designing, scaling, and optimizing enterprise-grade ML or data-intensive systems. Preferred Qualifications:... 

    Nace AI

    Palo Alto, CA
    4 days ago
  • $85.4k - $192.9k

     ...designing, developing, testing and maintaining software applications and products to meet...  ...customer requirements. Full-stack software engineering roles, who can develop all components of...  ...MS SQL Server, PostgreSQL, and MySQL. ~ ML Training: Google Vertex AI (Studio, Training... 
    Immediate start
    Flexible hours

    Ford Motor Company

    Palo Alto, CA
    3 days ago
  • $150k

     ...-class researchers, data scientists, and engineers, tackling the most fundamental and impactful...  ...to collaborate with Researchers and ML Engineers to produce AI applications and...  ...functional team that's responsible for the full software development life cycle, from conception... 

    Institute of Foundation Models

    Sunnyvale, CA
    3 days ago
  • $147k - $210k

     ...Implement solutions in one or more specialized ML areas, utilize ML infrastructure, and...  ...:Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science,...  ...launching features.2 years of experience with software development in one or more programming languages... 
    Local area

    Google

    Mountain View, CA
    2 days ago
  •  ...system-level results into clear feedback for engineering and leadership, and help accelerate...  ...analyses to introspect autonomous driving software performance at subsystem interfaces across...  ...Propose and develop new statistical and ML methods to quantify performance and identify... 
    Full time
    Local area
    Work from home

    General Motors

    Sunnyvale, CA
    1 day ago
  • $130k - $260k

     ...shaping the long-term technical vision, architecture, and engineering excellence of our enterprise-scale AI/ML systems and platforms. You will work across teams to...  ...building and operating large-scale, distributed software systems. Proven track record as a senior individual... 
    Full time
    Contract work
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    3 days ago
  • $162k - $260k

     ...efficient and accessible for all. We’re searching for a Senior Software Engineer, Perception to join our autonomous driving team. In this role,...  ...testing to produce robust, well-tested systems.Scale ML Engineering Pipelines: Build and scale robust ML pipelines to... 
    Work at office
    Local area
    3 days per week

    Aurora Innovation

    Mountain View, CA
    2 days ago
  • $174k - $252k

     ...minimal latency.Collaborate with cross-functional product and engineering teams to integrate features—like seamless cross-platform video...  ...working with algorithms, data structures, complexity analysis and software design.Experience with Android system-level development or... 
    Worldwide

    Google

    Mountain View, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Software Engineer, ML Infra. Be the first to apply!