Skip to content
CursorCursorSan Francisco, CA

Software Engineer, ML Infrastructure

Builds large-scale ML infrastructure including GPU clusters, training frameworks, and workload schedulers. Requires strong systems engineering in Python, Rust, Golang, with Kubernetes and distributed systems experience.

Salary not listed
On-siteML Engineering

About the role

ML Infrastructure Engineer

The ML Infrastructure team builds large-scale compute, storage, and software infrastructure to support Cursor’s work building the world’s best agentic coding model. We’re looking for strong engineers who are interested in building high-performance infrastructure and the software to support it. This role works closely with ML researchers and engineers to enable their work through improvements to our training framework, systems reliability/performance, and developer experience.

What you might do

  • Collaborate with ML researchers to improve the throughput and reliability of training
  • Work with OEMs, cloud service providers, and others to plan and build cutting-edge GPU infrastructure
  • Improve the density and scalability of compute environments to enable increasingly large RL workloads
  • Create software and systems to automate building, monitoring, and running GPU clusters
  • Build workload scheduling and data movement systems to support Cursor’s growing training footprint

What we’re looking for

  • A strong background in systems and infrastructure-focused software engineering, particularly in Python, Typescript, Rust, and Golang
  • Experience with distributed storage and networking infrastructure, particularly on Linux systems across cloud and bare metal environments
  • Exposure to large-scale systems and their unique challenges, ideally across thousands of nodes with significant resource footprints
  • Production use of infrastructure-as-code and configuration management, across hosts and Kubernetes

Nice to have

  • Operational exposure to Nvidia GPUs with Infiniband or RoCE, particularly with Blackwell and Hopper-class hardware
  • Exposure to Ray, Slurm, or other common compute and runtime schedulers

Skills

PythonTypeScriptRustGoKubernetesLinuxNvidia GpusInfiniBandRaySlurm

Similar roles

ML Engineering jobs
Cloudflare

Software Engineer, AI Agents

CloudflareUnited States

Build and ship production AI agents on Cloudflare's edge platform using Workers, Durable Objects, and AI tools. Requires strong TypeScript/Rust experience, observability expertise, and hands-on LLM tooling for evals, safety, and multi-agent systems.

Salary not listed
On-siteML Engineering
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering