ForgeApply · Job listing

Senior/Staff ML Engineer, Performance Optimization

Comfy-org

San Francisco, USonsiteOther

Apply in about a minute — without sacrificing quality.

ForgeApply autofills this application and tailors your resume to this exact posting. You review everything before it's sent. Free trial, no card required.

Apply with ForgeApply View original posting →

About this role

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