Relace Logo

Relace

Machine Learning Engineer

Reposted An Hour Ago
In-Office
San Francisco, CA, USA
Junior
In-Office
San Francisco, CA, USA
Junior
The Machine Learning Engineer role involves optimizing models for performance, utilizing CUDA, GPU scheduling, and collaborating on ML systems design.
The summary above was generated by AI
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.

Similar Jobs

6 Days Ago
Hybrid
Palo Alto, CA, USA
Mid level
Mid level
Financial Services
Design, prototype, and productionize LLM-based architectures for structured and unstructured financial data. Implement pre-training objectives, optimize training throughput, monitor and iterate on deployed models, mentor engineers, and collaborate with product teams to improve data quality, reduce resolution times, and increase consistency across operational workflows.
Top Skills: Attention MechanismsDeepspeedDistributed TrainingFsdpLarge-Scale Compute InfrastructureLlmsMemory Optimization TechniquesPyTorchRopeServing InfrastructureTransformers
6 Days Ago
Hybrid
Mountain View, CA, USA
140K-217K Annually
Mid level
140K-217K Annually
Mid level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Build and productionize agentic AI systems: improve agent behavior, reliability, latency, and quality using LLMs, multimodal models, and vector databases. Design eval benchmarks, manage training and evaluation datasets, implement context engineering, orchestration, multilingual and multimodal support, and partner with platform teams to deliver production-grade, tested ML-driven features.
Top Skills: GoHybrid Vector DatabasesLlmsMultimodal Foundation ModelsPythonVector Databases
6 Days Ago
Hybrid
Mountain View, CA, USA
Senior level
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Design and productionize agentic AI systems and components, improve agent behavior and reliability, create eval benchmarks and high-quality datasets, tune and evaluate LLMs and models, optimize latency and reliability in production, and partner with platform and UI teams to deliver end-to-end ML solutions.
Top Skills: Deep Learning ArchitecturesGoHybrid Vector DatabasesLlmsMultimodal Foundation ModelsPython

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account