Machine Learning Engineer - Infrastructure

About the Role

As our Machine Learning Infrastructure Engineer , you'll design and scale the platforms that power cutting-edge AI : from high-performance inference engines to the underlying agent technologies and large-scale compute clusters that keep everything running.

You'll collaborate closely with researchers and product engineers to push the limits of inference performance, build reliable foundations for AI agents, and advance the next generation of training and post-training pipelines.

Responsibilities

Speed up research development, help researchers explore SOTA and new techniques on day one

Build and optimize model training pipeline including data collection, data loading, SFT and RL

Optimize a high-performance inference platform on top of both open-source and proprietary inference engines

Develop and scale technologies for large-scale cluster scheduling, high-performance distributed training, and AI networking

Build a strong engineering discipline across observability and reliability at scale

Collaborate with research and product teams to translate breakthroughs into robust, production-ready infrastructure

Qualifications

Expertise in one or more of : inference engines, GPU optimization, cluster scheduling, or cloud-native infra

Familiarity with modern ML frameworks (PyTorch, vLLM, Verl, etc.)

Startup-ready mindset (adaptable, fast-moving, high-ownership)

What makes us interesting

Small, elite team of ex-founders, researchers from top AI Labs, top CS grads, and engineers from top companies

True ownership You will not be blocked by bureaucracy, shipping meaningful work within weeks rather than months

Serious momentum We\'re well-funded by top investors, moving fast, and focused on execution

What we do

Ship consumer products powered by cutting-edge AI research, and

Build infrastructure that facilitates research and product, and

Innovate cutting-edge research that will open up new consumer product forms

The Details

Full-time, onsite role in Menlo Park

Startup hours apply

Generous salary, with additional benefits to be discussed during the hiring process

J-18808-Ljbffr