ML Research Engineer

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 allinone platform to build stateoftheart AI models, end to end, from data preparation to production deployment, empowering innovators to build cuttingedge 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 opensource first company, with opencollaboration across the community as a core principle. Our work is:

• Open Source First: All our platform and core technology is open source

• Researchdriven: We conduct and publish original research in AI, collaborating with our community of academic research labs and collaborators

• Communitypowered: We believe in the power of opencollaboration 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, youll bridge research and engineering—designing scalable infrastructure, enabling cuttingedge experiments, and helping opensource the next generation of LLMs. You will collbaborate closely with our research team and the opensource community to build tools, run evaluations, and contribute to models that are safe, performant, and accessible.

  What youll do:

  • Design and build systems to support training, finetuning, 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 finetuning, and prompt optimization.

  • Develop scalable ML pipelines for distributed training (e.g., across multiGPU and multinode environments).

  • Contribute to opensource 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., cloudbased training, distributed systems).

    • Comfort working in highly collaborative environments with research and engineering teams.

    • Bonus: experience with academic publications, opensource contributions, or LLM alignment work.

    • Share Oumis values: Beneficial for all, Customerobsessed, Radical Ownership, Exceptional Teammates, Sciencegrounded.

    • Benefits

      • Competitive salary: $140,000 $200,000

      • Equity in a highgrowth startup

      • Comprehensive health, dental and vision insurance

      • 21 days PTO

      • Regular team offsites and events