Great software doesn’t happen on its own. It takes great people. That just happens to be our forte. With nearly 20 years of matching top engineering talent with preeminent and innovative brands, we look for people who are inquisitive, resourceful, dedicated to their craft, and driven to help companies build great software. If this sounds like you, read on.
You will join a highly technical team building a custom diffusion-based generative model and the surrounding training and data infrastructure. This is not fine-tuning an existing model; it is a ground-up implementation of a new architecture designed for high-end creative production.
Requirements
Must haves:
- This role will be a Team Lead and requires prior leadership experience
- Expert-level PyTorch
- Proven experience training custom diffusion models
- Deep understanding of attention mechanisms, U-Nets, VAEs, schedulers
- Ability to implement modular model components from architectural specs
- Training optimization (mixed precision, batching, checkpointing)
- Clean coding practices and strong documentation
- Ability to evaluate and improve visual model outputs
Nice to haves:
- Experience with HuggingFace Diffusers
- Work in image/video generation
- Knowledge of control nets, inpainting, style conditioning
We offer
- Work your way — anywhere, anytime. Our remote-first approach lets you choose where and how you work best!
- Experience working with diverse teams and gaining international expertise
- A friendly, supportive team and an enjoyable work environment where your ideas matter
- A chance to work on exciting, challenging projects using cutting-edge technologies that make a real impact
- Comprehensive health insurance, corporate psychologist access, and partial sports activity coverage
- Free training programs, reimbursement for certifications, and access to online learning platforms to fuel your growth
- Paid vacation, public holidays, and sick leave are fully covered by Forte Group
- Referral bonuses, regular performance reviews, and full support for business trips
- Corporate events and holiday presents
Join a team that invests in your well-being, growth, and success!