Founding ML Engineer – Flower Frontier Model Team

Job not on LinkedIn

🕒 December 17, 2025

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Flower Labs

WebsiteLinkedInAll Job Openings

11 - 50 employees

Founded 2023

🤖 Artificial Intelligence

📚 Education

Artificial Intelligence • Software • Education

Flower Labs is a company that develops a friendly federated AI framework called Flower, which enables users to implement federated learning across various machine learning frameworks and programming languages. Flower aims to simplify the process of federated learning, allowing efficient distribution and execution of workloads across a multitude of clients. The platform is designed for scalability and usability, supporting diverse devices and cloud systems for research and production purposes.

📋 Description

• Join as one of the founding members of the Flower Frontier Model Team, a new group at Flower Labs charged with building category-defining models. • Build SOTA LLMs and foundation models within a small, high-impact team. • Design, implement and optimize core components across the full spectrum of stages relevant to frontier model building: data curation, evals, pre-training, post-training. • Collaborate on the debugging of training instabilities and related issues. • Devise surrounding infrastructure, tooling, monitoring, and observability for large-scale LLM development.

🎯 Requirements

• Exceptional software engineering skills (Python, deep learning frameworks, testing, profiling, refactoring, reproducibility) • Expertise with modern ML training stacks: PyTorch, JAX or equivalent; experience implementing model architectures from scratch and working within libraries like DeepSpeed, Megatron or equivalent • Ability to tune, debug, and profile large-scale training runs • Hands-on experience working with large GPU clusters, including job orchestration, scheduling, multi-node runs, NCCL/RDMA issues, and GPU performance optimization • Ability to collaborate effectively with both research-oriented and engineering-oriented colleagues; comfortable turning research ideas into robust, maintainable implementations • Good engineering hygiene: modular design, code reviews, documentation, reproducibility, versioning of data/models/configurations • Familiarity with common tools (Linux command line, git, Docker, …) • Openness to adopting new tooling • Solid understanding of distributed systems and networking • Strong written English • Open, honest and transparent communication skills

🏖️ Benefits

• Professional development opportunities • Flexible working hours