Logo
Oumi

ML Research Engineer

Oumi, Palo Alto, California, United States, 94306

Save Job

Overview

Join to apply for the

ML Research Engineer

role at

Oumi We’re looking for a Research Engineer to join our team working on generative AI and LLMs. In this role, you'll bridge research and engineering—designing scalable infrastructure, enabling cutting-edge experiments, and helping open-source the next generation of LLMs. You will collab orate closely with our research team and the open-source community to build tools, run evaluations, and contribute to models that are safe, performant, and accessible. Role Overview

We’re looking for a Research Engineer to join our team working on generative AI and LLMs. In this role, you'll bridge research and engineering—designing scalable infrastructure, enabling cutting-edge experiments, and helping open-source the next generation of LLMs. You will collab orate closely with our research team and the open-source community to build tools, run evaluations, and contribute to models that are safe, performant, and accessible. What you'll do

Design and build systems to support training, fine-tuning, and evaluating large language models. Partner with researchers to define experiments, write reusable code, run benchmarks, and interpret results. Work on LLM alignment and tuning using techniques like reinforcement learning (RLHF), supervised fine-tuning, and prompt optimization. Develop scalable ML pipelines for distributed training (e.g., across multi-GPU and multi-node environments). Contribute to open-source tooling and models to support transparency and community collaboration. Optimize performance across the ML stack—from data loading to deployment. What you’ll bring

Strong experience in machine learning, deep learning, or NLP—especially in generative AI or LLMs. Solid programming skills in Python, and experience with ML frameworks like PyTorch. Experience designing or maintaining ML infrastructure at scale (e.g., cloud-based training, distributed systems). Comfort working in highly collaborative environments with research and engineering teams. Bonus: experience with academic publications, open-source contributions, or LLM alignment work. Share Oumi's values: Beneficial for all, Customer-obsessed, Radical Ownership, Exceptional Teammates, Science-grounded. Benefits

Equity in a high-growth startup Comprehensive health, dental and vision insurance 21 days PTO Location

Hybrid work environment, with offices in: Seattle, WA Palo Alto, CA New York, NY

#J-18808-Ljbffr