Software Engineer, Applied Machine Learning

Home > Python programming jobs

Software Engineer, Applied Machine Learning in USA

fal
Full time
Email

USA
10/14/2026
Report

San Francisco, CA

Responsibilities

Extend image, video, audio, and 3D generative models with new capabilities and modalities using conditioning, generation, editing, training-free methods, and fine-tuning.
Fine-tune generative models and build APIs that allow customers to customize models.
Develop reusable components, abstractions, and building blocks for model capabilities and inference pipelines.
Optimize inference for low latency, high throughput, and efficient GPU utilization.
Build, deploy, and maintain scalable and reliable generative model APIs.
Collaborate directly with customers to develop solutions for generative media use cases.

Requirements

3+ years of professional experience as an Applied ML Engineer, including at least 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 with open-weight model ecosystems, including Hugging Face and Diffusers.
Experience designing training-free extensions for image, video, audio, or 3D generative models.
Experience developing custom model post-training or fine-tuning approaches.
Ability to design scalable, reliable, secure, safe, and performant production ML systems.
Startup or digital media and entertainment industry experience is a nice-to-have.
Demonstrated ability to independently take ambitious ML ideas from concept to production.

Benefits

Health, dental, and vision insurance in the US.
Regular team events and offsites.
Interesting and challenging work with learning and growth opportunities.
Compensation includes $180,000–$230,000 base salary plus equity and comprehensive benefits.

Salary: $180k - $230k/yr

fal

fal is a generative media platform that provides developers with access to the world's best generative image, video, and audio models through a unified API. Trusted by over 2.5 million developers and leading companies, fal offers the fastest inference engine for diffusion models, on-demand server...

Similar positions

AI Applied Scientist

Figma
Full time

USA
10/31/2026
Salary: $153k-$376k

Remote

Sr. Software Engineer - Search & Insights

H1
Full time

Remote
10/31/2026

India

Software Engineer

Xapo Bank
Full time

Remote
10/31/2026

Anywhere

Staff Software Engineer, Backend (Observability)

Affirm
Full time

Canada
10/31/2026
Salary: $181k - $241k/yr

Remote

Staff Software Engineer, Sensing Systems

CHAOS Industries
Full time

USA
10/31/2026
Salary: $211k - $290k/yr

El Segundo, CA