Machine Learning Engineer

Reposted 10 Days Ago
San Francisco, CA, USA
In-Office
Junior
Artificial Intelligence • Machine Learning • Software • Generative AI
The Role
The Machine Learning Engineer role involves optimizing models for performance, utilizing CUDA, GPU scheduling, and collaborating on ML systems design.
Summary Generated by Built In
About Us

Relace is building the models and infrastructure that code agents reach for. We power the fastest model on OpenRouter (10,000 tok/s) and deliver optimized small language models designed for retrieval, application, and core code generation functions.

Our technology supports some of the world’s fastest-moving companies — including Lovable, Figma, and Vercel — as they deploy and scale code generation to hundreds of millions of users. We recently raised our Series A from a16z, and we’re growing quickly.

Our team is made up of mathematicians, physicists, and computer scientists who are deeply passionate about their craft. If you thrive on ambitious technical problems, care about elegant systems design, and want to build the foundation of how code gets written at scale, this is the place for you.

The Role

We’re looking for a Machine Learning Engineer who loves getting close to the metal. This is a hands-on engineering role focused on making models faster, more efficient, and more reliable through low-level optimizations and smart systems design.

The ideal candidate is excited by CUDA kernels, memory layouts, GPU scheduling, and squeezing performance out of complex training and inference workloads. They should be just as comfortable optimizing compute and networking paths as they are working alongside research teams to productionize new architectures.

This is a role for someone who enjoys deep performance tuning, understands the realities of running large-scale ML systems, and thrives in fast-moving, high-leverage environments.

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 Required

  • 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
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
6 Employees
Year Founded: 2022

What We Do

Relace develops AI models and infrastructure for coding agents, offering an end-to-end fine-tuning and inference platform for large language models.

Similar Jobs

General Motors Logo General Motors

Machine Learning Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Remote or Hybrid
Sunnyvale, CA, USA
165000 Employees
182K-251K Annually

General Motors Logo General Motors

Machine Learning Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Remote or Hybrid
Sunnyvale, CA, USA
165000 Employees
189K-321K Annually

Block Logo Block

Machine Learning Engineer

Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In-Office or Remote
8 Locations
12000 Employees
277K-415K Annually

Block Logo Block

Machine Learning Engineer

Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In-Office or Remote
8 Locations
12000 Employees
277K-415K Annually

Similar Companies Hiring

Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
LTX Thumbnail
Conversational AI • Generative AI
Jerusalem, Israel
360 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account