Software Engineer, Applied Machine Learning

The role

You will take technical ownership of the model layer for a generative media platform. You will extend SOTA open-source models and help maintain a fleet of generative model APIs.

What you will do

• Extend SOTA image, video, audio, and 3D models with additional capabilities and modalities.
• Develop novel approaches to model conditioning, generation, and editing.
• Fine-tune generative models using a GPU fleet.
• Build and maintain fine-tuning APIs for customer customization.
• Develop reusable components, abstractions, and building blocks for model capabilities and inference pipelines.
• Apply state-of-the-art inference techniques for efficiency, low latency, and high throughput.
• Build, deploy, and maintain scalable, reliable generative model APIs.
• Work directly with customers to develop novel solutions for generative media needs.

Required qualifications

• 3+ years of professional experience as an Applied ML Engineer.
• 1–2 years focused on generative media or computer vision.
• Expert-level proficiency in Python and PyTorch.
• Deep practical understanding of diffusion and flow-based generative models.
• Hands-on experience working with open-weight model ecosystems, including Hugging Face, Diffusers, and related tooling.
• Ability to anticipate and solve challenges that arise when deploying ML models to production.
• Strong engineering judgment in designing systems that are scalable, reliable, secure, safe, and performant.

Nice to have

• Experience designing and implementing training-free extensions to image, video, audio, or 3D generative models.
• Experience developing and executing custom post-training or fine-tuning approaches to extend generative models.
• Track record of working in fast-paced startup environments or digital media and entertainment industries.
• Demonstrated ability to independently take ambitious ML ideas from concept to production.

Benefits

• Interesting and challenging work
• A lot of learning and growth opportunities
• Health, dental, and vision insurance (US)
• Regular team events and offsites

About the company

fal is the generative media ecosystem powering the next generation of AI products. They build infrastructure, tools, and model access for developers and enterprises.