# Senior Software Engineer - ML Infrastructure

**Company:** [Applied Intuition](https://hotfix.jobs/companies/applied-intuition)
**Location:** Sunnyvale, CA
**Role:** ML Engineering
**Salary:** $153k – $222k/yr
**Experience:** 3+ years
**Skills:** PyTorch, TensorFlow, Airflow, Flyte, Kubernetes, GPU, Distributed Training, Machine Learning Pipelines, Torchserve, Nvidia Triton
**Posted:** 2026-04-21

> Builds distributed ML infrastructure including GPU training, end-to-end pipelines, and deployment platforms. Requires 3+ years experience in production ML systems, strong software engineering, and familiarity with open-source tools.

## Job Description

## Responsibilities
- Design and implement distributed cloud GPU training approaches for deep learning model training and evaluation
- Build end-to-end machine learning pipelines and integrate them into core product workflows
- Encourage change, especially in support of ML engineering best practices, and maintain a high standard of excellence
- Collaborate with engineers across the entire company to solve complex data problems at scale

## Requirements
- Bachelor's degree in Computer Science, Software Engineering, or equivalent
- 3+ years of professional experience
- Experience with building software components to address production, full-stack machine learning challenges
- Opinions about building a company-wide platform for ML training, evaluation, and deployment
- Knowledge of the open source landscape with judgment on when to choose open source versus build in-house
- Excellent analytical and problem-solving skills

## Nice to Have
- Experience with developing, running, and managing orchestration systems like Airflow and Flyte that non-engineers can use to build data pipelines
- Experience with ML modeling frameworks (PyTorch, Tensorflow, etc.), and model serving platforms (TorchServe, TensorFlow Serving, NVIDIA Triton inference server, etc.)

## Compensation
- Base salary range: $153,000 - $222,000 USD annually
- Equity, comprehensive health/dental/vision insurance, 401k with employer match, learning/wellness stipends, paid time off

## Similar roles

- [Senior AI Engineer](https://hotfix.jobs/jobs/fdf9ce54-1a65-4442-a904-d175b0df9243) - Grafana Labs - Remote - $154k – $185k/yr
- [Senior Software Engineer - AI and Automation, Data & Analytics | USA | Remote](https://hotfix.jobs/jobs/d182728b-ee48-41c1-8d1f-b440bbbc7c0c) - Grafana Labs - Remote - $154k – $185k/yr
- [Machine Learning Engineer - Simulation Framework](https://hotfix.jobs/jobs/f38163d6-88a9-4ed1-9c29-b1f4e610f320) - Zoox - Foster City, CA - $151k – $257k/yr
- [Sr. Software Engineer, Machine Learning](https://hotfix.jobs/jobs/19067112-c9ce-4c49-8706-9f960d168a9c) - Pinterest - Remote - $156k – $320k/yr
- [Sr. Software Engineer, tvScientific](https://hotfix.jobs/jobs/82f201f6-0d12-4205-9a11-8a912b57fef9) - Pinterest - Remote - $156k – $320k/yr

**Apply:** https://hotfix.jobs/jobs/ce708525-34f1-4474-9516-673a3457e146
**Canonical:** https://hotfix.jobs/jobs/ce708525-34f1-4474-9516-673a3457e146