Founding ML Engineer in the Flower Frontier Model Team (all seniority levels welcome) [Germany, UK, Global]
About the role
About the Role
Do you want to push the boundaries of what frontier AI models can be? Join as one of the founding members of the Flower Frontier Model Team, a new group at Flower Labs. We will ship models with superhuman capabilities in domains spanning science, health, finance, drug discovery, and more. This is an opportunity to help invent and build the training paradigms that will define the next decade of AI.
Preference given to candidates with post-training expertise. But any talented individual with a track record of exceptional drive and determination are encouraged to apply regardless of prior experience.
Responsibilities
- Design, implement and optimize core components across data curation, evals, pre-training, and post-training
- Build reliable, maintainable and scalable software stack for SOTA LLMs and foundation models
- Diagnose and resolve GPU/kernel issues, memory/storage bottlenecks, and multi-node failures at scale
- Collaborate on debugging training instabilities and related issues
- Devise surrounding infrastructure, tooling, monitoring, and observability
- Assume technical leadership as the training system scales in complexity and capability
Qualifications
- Familiarity with ML distributed and scaling strategies
- Experience working with GPU clusters for multi-node training
- Ability to adapt to different HPC configurations and GPU architectures (e.g., AMD/NVIDIA)
- Strong engineering discipline and problem-solving skills
- Willingness to learn on the job and work collaboratively
About the Company
Flower Labs is the world-class AI startup behind the most popular open-source framework for training AI on distributed data and compute resources using decentralized and federated methods. Trusted by industry leaders such as Mozilla, JP Morgan, Owkin, Banking Circle and Temenos. A Y Combinator-backed company.