# Member of Technical Staff, AI Training Infrastructure

**Company:** [Fireworks AI](https://hotfix.jobs/companies/fireworks-ai)
**Location:** San Mateo, CA
**Role:** DevOps / SRE
**Salary:** $175k – $220k/yr
**Skills:** Python, C++, Kubernetes, Distributed Systems, Gpu Programming, PyTorch, TensorFlow, MLOps, AWS, GCP
**Posted:** 2026-03-05

> Builds and optimizes scalable infrastructure for AI model training on large GPU clusters. Requires expertise in distributed systems, Python/C++, and ML frameworks.

## Job Description

## Responsibilities

- Design, build, and maintain scalable infrastructure for AI model training pipelines.
- Optimize distributed training systems for large-scale GPU clusters.
- Collaborate with AI researchers and engineers to deploy training workflows efficiently.

## Requirements

- Strong experience with distributed systems and high-performance computing.
- Proficiency in Python, C++, and infrastructure-as-code tools.
- Deep knowledge of AI training frameworks and GPU optimization.

## Nice-to-Haves

- Experience with Kubernetes and cloud platforms like AWS or GCP.
- Background in machine learning operations (MLOps).

## Similar roles

- [Senior/Staff Site Reliability Engineer](https://hotfix.jobs/jobs/176c9f9e-3a92-449a-b4cb-39cf88da222b) - Sage - New York, NY - $175k – $230k/yr
- [Member of Technical Staff, Performance Optimization](https://hotfix.jobs/jobs/71bacc0c-4460-4a8e-870b-9c1de45763ec) - Fireworks AI - San Mateo, CA - $175k – $220k/yr
- [Member of Technical Staff, Cloud Infrastructure](https://hotfix.jobs/jobs/5d9ce5b3-e68f-4a25-80bc-3a35d2c6fcae) - Fireworks AI - New York, NY - $175k – $220k/yr
- [Staff Software Engineer, Infrastructure](https://hotfix.jobs/jobs/bd4b210b-ee5c-467a-877b-7095aa86057e) - Rad AI - Remote - $175k – $230k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/3a484da9-09cb-408e-a132-e6f664620a16) - Grafana Labs - Remote - $175k – $210k/yr

**Apply:** https://hotfix.jobs/jobs/3b60d4a2-35a8-4163-8505-a2811ac7ceae
**Canonical:** https://hotfix.jobs/jobs/3b60d4a2-35a8-4163-8505-a2811ac7ceae