Oumi is looking to solve the problem of making frontier AI truly open for all by developing an all-in-one platform to build state-of-the-art AI models and open foundation models in collaboration with academic collaborators and the open community.
Requirements
- 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.
Responsibilities
- 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.
Other
- Share Oumi's values: Beneficial for all, Customer-obsessed, Radical Ownership, Exceptional Teammates, Science-grounded.
- Hybrid work environment, with offices in: Seattle, WA, Palo Alto, CA, New York, NY
- Equity in a high-growth startup
- Comprehensive health, dental and vision insurance
- 21 days PTO
