Job Details

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: $125,000 - $200,000
• Equity in a high-growth startup
• Comprehensive health, dental and vision insurance
• 21 days PTO
• Regular team offsites and events

Location:

Hybrid work environment, with offices in:

• Seattle, WA
• Palo Alto, CA
• New York, NY