Member of Technical Staff, Product Engineering

About the role

We build the fastest GPU compiler in the world. Most compilers have to preserve correctness at every transform, constraining how far they can search, while we prove correctness at the end instead, allowing us to search a far wider space, with agents, with RL, with anything that works and still guarantee the result. It's why we hold #1 on NVIDIA's own kernel benchmark across hundreds of production kernels.

Speed at the kernel layer is only worth what we do with it though, so we're hiring a Member of Technical Staff for Product Engineering to own the surface that the engine reaches the world through. That surface is Model Foundry and it takes a researcher from idea to a running experiment in one commit or one message. When a researcher needs to see the perplexity curve from last night's run or to try an experiment with a new annealing rate, Foundry writes the config, queues the job on whatever silicon makes sense that week, then stream the logs back and versions the whole thing so it can be reproduced six months later.

You'll own it end to end from the interfaces researchers live in all day to the services behind them, making sure the software doesn't just work, but the experience feels right and is enjoyable even when using it 16 hours a day for months on end. The users are close and the loop is short. Foundry is used by both our customers, our internal research team as well as our forward-deployed researchers to train models every hour of the day, so your users either sit next to you or a Slack message away.

The data you're processing is not gentle: you'll be processing telemetry from thousands of GPUs across NVIDIA, AMD, TPU and Trainium from runs that last for weeks to months parsing logs and charts that don't stop and making sure all of the data is accessible to user as well as their agents.

What You'll Do

• Take full ownership of product areas in Model Foundry: scoping, designing, building and iterating on them
• Build clean, fast and polished web interfaces with NextJS that feel almost native
• Design the surfaces where researchers live: run dashboards, live log streaming, experiment comparison, eval results, cluster and storage views
• Build the conversational and commit-driven entry points into Foundry, where a Slack message or commit become a training run
• Obsess over the small things, from loading states to smooth animations and the edge cases that turn a good product into an exceptional one
• Work directly with our forward-deployed researcher and our own training team. They're your users, they sit next to you and their feedback loop is measured in minutes
• Make smart product tradeoffs with little hand-holding, which means knowing when to ship fast and when to spend extra time polishing

What We're Looking For

• Someone with proven experience shipping complete products, ideally something you can show us
• Someone with real taste and a sharp eye for design and motion, which means you notice when an animation eases awkwardly, when padding feels inconsistent or something just feels cheap
• Someone with solid backend chops, which means you can build APIs and troubleshoot production issues
• Someone with enough infrastructure comfort to handle AWS, containers and Kubernetes (you don't need to be a DevOps wizard but shouldn't be scared of a Dockerfile either)

Nice to Have

• Someone with experience working with Bun or an eagerness to dive in
• Someone with a background in developer tools, infrastructure products or building for technical users
• Someone with any exposure to ML training workflows or tools like Weights & Biases, Ray or Slurm
• Someone comfortable using Figma to mock up things yourself when needed

Why Join Us

Most infrastructure companies treat their interface as an afterthought and their users can tell. The engineering and engine behind our product is as good as it gets and the product on top deserves to be just as good. You'll be defining what researchers at frontier labs feel when they train a model and are staring at logs and curves for hours on end with the goal of making the research experience as enjoyable as possible and let the infrastructure get out of the way and fade into the background.

We're a small team operating at frontier scale. We pre-trained foundation models on 4,000 AMD GPUs as a team of three, designed and brought up GB300 NVL72 clusters and designed a TOP500 supercomputer. We believe that hard problems get solved in person and most of our work happens at our office in San Francisco. We offer relocation assistance and, where possible, we'd like you here as often as possible.

The base salary range for this full-time position is $225,000–$275,000, plus meaningful equity and benefits.