Skip to content
RelaceRelaceSan Francisco, CA

Machine Learning Engineer

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.

Salary not listed
On-site2+ YOEML Engineering

About the role

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.

Skills

CUDAPythonC++RustPyTorchJAXDeepspeedGpu OptimizationDistributed TrainingMemory Management

Similar roles

ML Engineering jobs
Nuro

Software Engineer, Performance

NuroMountain View, CA

New grad software engineer optimizing performance of autonomous vehicle software. Profile, debug, and reduce latency of C++ systems running on x86, ARM, and GPUs while building tools for memory management and high-performance code.

153k – 175k/yr
On-siteEntry levelML Engineering
Airbnb

Machine Learning Engineer, Relevance and Personalization

AirbnbUnited States

Build and productionize cutting-edge ML models and ranking algorithms for Airbnb's search, recommendation, and personalization systems at scale. Requires PhD (new grad) or 2+ years applied ML experience with strong programming and ML systems skills.

166k – 195k/yr
Remote2+ YOEML Engineering
Abridge

Software Engineer

AbridgeSan Francisco, CA +1

Early-career Software Engineer building and iterating on agentic LLM systems, RAG pipelines, evaluation frameworks, and backend/frontend components for AI-powered clinical documentation at Abridge. Requires CS degree or equivalent plus hands-on GenAI experience from projects, internships or coursework.

157k – 184k/yr
On-siteEntry levelML Engineering
Mirage

ML Engineer, Generative Video

MirageNew York, NY

Build and scale video generation models at Mirage. Optimize training/inference for low-latency, real-time performance using PyTorch, CUDA, and distributed systems. Requires 2+ years industry experience in deep learning infrastructure.

175k – 275k/yr
On-site2+ YOEML Engineering
Traba

Software Engineer

TrabaNew York, NY

Build and own production AI agent systems (harnesses, evals, orchestration) on frontier LLMs for industrial supply chain workflows at Traba. Requires 1+ years shipping LLM/agent features to production, strong Python/TS skills, and high-agency in ambiguous customer environments.

140k – 200k/yr
On-site1+ YOEML Engineering