Logo
Oumi

ML Research Engineer

Oumi, Palo Alto, California, United States, 94306

Save Job

About Oumi

Why we exist:

Oumi is on a mission to make frontier AI truly open for all. We are founded on the belief that AI will have a transformative impact on humanity, and that developing it collectively, in the open, is the best path forward to ensure that it is done efficiently and safely. What we do:

Oumi provides an all-in-one platform to build state-of-the-art AI models, end to end, from data preparation to production deployment, empowering innovators to build cutting-edge models at any scale. Oumi also develops open foundation models in collaboration with academic collaborators and the open community. Our Approach:

Oumi is fundamentally an open-source first company, with open-collaboration across the community as a core principle. Our work is: Open Source First: All our platform and core technology is open source

Research-driven: We conduct and publish original research in AI, collaborating with our community of academic research labs and collaborators

Community-powered: We believe in the power of open-collaboration and welcome contributions from researchers and developers worldwide

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 collbaborate 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

Competitive salary: $140,000 - $220,000

Equity in a high-growth startup

Comprehensive health, dental and vision insurance

21 days PTO

Regular team offsites and events

#J-18808-Ljbffr