ComfyUI · 3 days ago
Senior/Staff Applied Machine Learning Engineer
San Francisco, CA
Full-time
Onsite
Senior Level, Lead/StaffComfyUI is the world’s leading visual AI platform, and they are seeking a Senior/Staff Applied Machine Learning Engineer to optimize model inference and enhance the performance of AI models. The role involves building and optimizing the core inference engine, tackling technical challenges in visual AI, and collaborating with the core team on new features.
Computer Software
Responsibilities
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
Qualification
Required
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
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
Preferred
If you've worked with diffusion/LLM models before or built custom nodes for ComfyUI, that's awesome
Company
ComfyUI
Building ComfyUI
Founded in 2024
11-50 employees
H1B Sponsorship
ComfyUI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2024 (2)
2022 (8)
2021 (4)
2020 (1)
Funding
Current Stage
Early StageTotal Funding
$17M2025-09-16Series Unknown· $17M
Company data provided by crunchbase