Staff Technical Lead for Inference & ML Performance

fal • San Francisco, CA

19h

About The Position

fal is the generative media ecosystem powering the next generation of AI products. We build the infrastructure, tools, and model access that teams need to move from idea to production, and do it at scale without compromise. For developers and enterprises, fal is the foundation that makes generative media not just possible, but practical: a unified platform where high-performance inference, orchestration, and observability come together to unlock new categories of AI-native products.
As generative media reshapes industries across a market projected to grow by hundreds of billions over the next decade, fal is becoming the ecosystem that ambitious teams build on.
Why this role matters
You’ll shape the future of fal’s inference engine and ensure our generative models achieve best-in-class performance. Your work directly impacts our ability to rapidly deliver cutting-edge creative solutions to users, from individual creators to global brands.

Requirements

• Deeply experienced in ML performance optimization. You've optimized inference for large-scale generative models in production environments.
• Understand the full ML performance stack. From PyTorch, TensorRT, TransformerEngine, Triton to CUTLASS kernels, you’ve navigated and optimized them all.
• Know inference inside-out. Expert-level familiarity with advanced inference techniques: quantization, kernel authoring, compilation, model parallelism (TP, context/sequence parallel, expert parallel), distributed serving and profiling.
• Lead from the front. You're a respected IC who enjoys getting hands-on with the toughest problems, demonstrating excellence to inspire your team.
• Thrive in cross-functional collaboration. Comfortable interfacing closely with applied ML teams, researchers, and stakeholders.

Nice To Haves

• Experience building inference engines specifically for diffusion and generative media models
• Track record of industry-leading performance improvements (papers, open-source contributions, benchmarks)
• Leadership experience in scaling technical teams

Responsibilities

• Set technical direction. Guide your team (kernels, applied performance, ML compilers, distributed inference) to build high-performance inference solutions.
• Hands-on IC leadership. Personally contribute to critical inference performance enhancements and optimizations.
• Collaborate closely with research & applied ML teams. Influence model inference strategies and deployment techniques.
• Drive advanced performance optimizations. Implement model parallelism, kernel optimization, and compiler strategies.
• Mentor and scale your team. Coach and expand your team of performance-focused engineers.

Benefits

• One of the highest impact roles at one of the fastest growing companies (revenue is growing 40% MoM, we are 60x+ RR compared to last year, raised Series A/B/C within the last 12 months) with a world changing vision: hyperscaling human creativity.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed