C
Comfy Org

Senior/Staff ML Engineer, Performance Optimization

San Francisco, CA Posted 2025-05-29
Type
Full-time
Experience
8+ yr
Source
Ashby
The Role

We're looking for someone who loves optimizing model inference to join us in building the core of ComfyUI - the most complex and bleeding-edge part of our engine. You'll be working on making AI models run faster and more efficiently than anyone thought possible.

You are a good fit if this describes you:

- You geek out about model inference, torch optimizations, and memory management

- You've written production PyTorch code that pushes performance boundaries

- You love diving deep into how models actually work under the hood

- You get excited about making insanely optimized code that just works

- You think the current state of ML deployment could be way better

What you'll do:

- Build and optimize the core inference engine that powers ComfyUI

- Make massive models run faster and use less memory than anyone else

- Work directly with our core team on architecting new features

- Tackle the hardest technical problems in the visual AI space

- Help shape where we take this technology next

Bonus: If you've worked with diffusion/LLM models before or built custom nodes for ComfyUI, that's awesome
PyTorchLLM
Comfy Org is hiring for the senior/staff ml engineer, performance optimization role. NewJob aggregates active openings directly from Comfy Org's applicant tracking system, so this listing is current. More jobs at Comfy Org →
Apply on company site