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