# Machine Learning Engineer

**Company:** [Relace](https://hotfix.jobs/companies/relace)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Experience:** 2+ years
**Skills:** CUDA, Python, C++, Rust, PyTorch, JAX, Deepspeed, Gpu Optimization, Distributed Training, Memory Management
**Posted:** 2025-10-29

> Machine Learning Engineer optimizes ML models for speed and efficiency through low-level CUDA kernel tuning, GPU scheduling, and hardware-aware systems design. Requires 2+ years in ML infrastructure with Python/C++/Rust and distributed frameworks like PyTorch.

## Job Description

## Requirements
- Strong background in systems-level ML engineering.
- Experience with **CUDA**, GPU kernel optimization, and performance tuning.
- Fluency in **Python** and at least one systems language (**C++** or **Rust** preferred).
- Familiarity with distributed training frameworks (e.g., **PyTorch**, **JAX**, **DeepSpeed**, or similar).
- Experience working with large-scale training or inference infrastructure.
- Understanding of memory management, parallelization, and hardware-aware model optimization.
- 2+ years of experience working in ML infrastructure or performance-critical environments.
- Willingness to work in-person from our SF office in FiDi.

## Similar roles

- [Software Engineer, Performance](https://hotfix.jobs/jobs/ff4b3339-6a08-4324-bcb6-635f3698a11d) - Nuro - Mountain View, CA - $153k – $175k/yr
- [Machine Learning Engineer, Relevance and Personalization](https://hotfix.jobs/jobs/e55bff4d-9948-4fd5-8f5e-51f0f5c17db4) - Airbnb - Remote - $166k – $195k/yr
- [Software Engineer](https://hotfix.jobs/jobs/823e27ee-a56e-4d3e-ac4d-dacbce6de23d) - Abridge - San Francisco, CA - $157k – $184k/yr
- [ML Engineer, Generative Video](https://hotfix.jobs/jobs/70497a05-ec28-442a-b182-4bb62ef208eb) - Mirage - New York, NY - $175k – $275k/yr
- [Software Engineer](https://hotfix.jobs/jobs/6d8b2f8e-891e-4d59-b0b0-632d28fa07ac) - Traba - New York, NY - $140k – $200k/yr

**Apply:** https://hotfix.jobs/jobs/8c4736f7-f3b8-46a9-8f67-521caf7c6165
**Canonical:** https://hotfix.jobs/jobs/8c4736f7-f3b8-46a9-8f67-521caf7c6165