ForgeApply · Job listing
Member of Technical Staff - ML Performance
Modal
See all 27 open roles at Modal →
Tailor your resume for this Modal job in about a minute.
ForgeApply rewrites your resume for this exact posting, then autofills the application on Modal's site with it. You review everything before it's sent. Free trial, no card required.
About this role
ABOUT US:
AI needs a new infrastructure layer. We're building it at Modal.
Every era of computing brought new workloads that previous infrastructure couldn't support: mainframes, databases, and the cloud. Each time, the company that rebuilt the layer underneath defined the decade. AI is no different, except it touches everything instead of one slice, and the window to build the layer underneath it is open right now.
Our customers include category-defining companies like Lovable https://modal.com/blog/lovable-case-study, Ramp https://modal.com/blog/how-ramp-built-a-full-context-background-coding-agent-on-modal, Cognition, DoorDash, and Suno. They rely on Modal for instant GPU access, sub-second container starts, and native storage, so it's simple to serve low-latency inference, fine-tune models, and access production-ready sandboxes at scale.
We recently raised a $355M Series C https://modal.com/blog/modal-series-c at a $4.65B valuation, led by General Catalyst and Redpoint Ventures. We've crossed $300M+ ARR and grown fivefold since September.
Our team includes creators of popular open-source projects (e.g.,Seaborn https://github.com/mwaskom/seaborn,Luigi https://github.com/spotify/luigi), academic researchers, international olympiad medalists, and experienced engineering and product leaders with decades of experience.
THE ROLE:
We are looking for strong engineers with experience in making ML systems performant at scale. If you are interested in contributing to open-source projects and Modal’s container runtime to push language and diffusion models towards higher throughput and lower latency, we’d love to hear from you!
REQUIREMENTS:
- 5+ years of experience writing high-quality, high-performance code.
- Experience working with torch, high-level ML frameworks, and inference engines (vLLM or TensorRT).
- Familiarity with Nvidia GPU architecture and CUDA.
- Experience with ML performance engineering (tell us a story about boosting GPU performance — debugging SM occupancy issues, rewriting an algorithm to be compute-bound, eliminating host overhead, etc).
- Nice-to-have: familiarity with low-level operating system foundations (Linux kernel, file systems, containers, etc).
Salary insight
The midpoint of this range ($275k) is about 72% above the median disclosed salary for New York roles listed on ForgeApply ($160k across 9,985 jobs).
Based on live postings with disclosed pay on ForgeApply; refreshed daily. Not an estimate of this employer's offer.
Tailor your resume for this Modal role before you apply.
Tailor my resume for this jobSimilar jobs
- Member of Technical Staff, ML Performance — Odysseyml · Palo Alto
- Member of Technical Staff, ML Systems — Netpreme · Santa Clara, CA
- Member of Technical Staff, ML Engineer — Psi · Boston
- Member of ML Technical Staff — Pragmatike · San Francisco
- Senior/Staff Software Engineer, ML Performance Optimization — Zoox · Foster City, CA
- Software Engineer, ML Performance Optimization — Zoox · Foster City, CA
- Member of Technical Staff - ML Systems & Inference — Gimlet · San Francisco, CA
- Member of Technical Staff — ML Research, Planning — Causal · San Francisco
Free ATS checker · How to Tailor Your Resume to a Job Description (Step by Step)