Member of Technical Staff, ML Engineer
Member of Technical Staff, ML Engineer 📍 Cambridge, MA · Hybrid, 3 days in office · Full time 🔬 THE OPPORTUNITY Join an early-stage team building AI systems to discover new physics at scale. Turn research code into reliable production systems and own the training and inference infrastructure behind AI-driven scientific discovery. ⚙️ WHAT YOU WILL OWN * Translate research code into dependable services with clear interfaces, meaningful tests and operational ownership. * Build services for launching experiments, routing across models and delivering results. * Own distributed ML jobs, GPU scheduling and model serving on a shared engineering platform. * Develop deep expertise in one area: distributed training, inference serving, ML platforms, or LLM gateway and agent infrastructure. 👤 WHAT YOU WILL BRING * 3+ years building and operating production ML training or inference infrastructure at meaningful scale, including ownership through failures. * 3+ years of software engineering on systems used by other engineers or researchers. Depth in at least one of: * Distributed training: PyTorch DDP/FSDP, Ray, multi-node jobs, checkpointing and performance. * Inference serving: vLLM, SGLang, Triton, batching, KV cache and quantization. * ML platforms: Kubernetes GPU scheduling, experiment tracking, registries and metering. * Gateways and agents: model routing, agent runtimes, sandboxes and evaluation infrastructure. 🔎 USEFUL EXPERIENCE * Debugging training loops, reward signals or inference behavior alongside researchers. * GPU infrastructure, CUDA or ML performance engineering. * GCP or AWS and infrastructure as code. * Experience turning research prototypes into production systems. 🤝 HOW THE TEAM WORKS High technical standards, simple designs and ownership from specification through deployment and on-call. Expect daily use of agentic coding tools and close collaboration with researchers. 💼 COMPENSATION Competitive salary, benefits and meaningful early-stage equity.