Senior/Staff ML Engineer, Performance Optimization

Posted 6 Months Ago
San Francisco, CA, USA
In-Office
Senior level
Information Technology • Software
The Role
Join ComfyUI to optimize model inference and build a core engine for AI models, ensuring efficiency and performance in visual AI workflows.
Summary Generated by Built In

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

Skills Required

  • Experience with production-grade PyTorch code
  • Deep understanding of model inference, optimizations, and memory management
  • Experience with diffusion/LLM models
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
40 Employees
Year Founded: 2024

What We Do

Building ComfyUI

Similar Jobs

Zoox Logo Zoox

Staff Software Engineer

Artificial Intelligence • Machine Learning • Robotics • Software • Transportation • Design • Manufacturing
In-Office or Remote
2 Locations
2900 Employees
242K-389K Annually

VSCO Logo VSCO

Community Manager

Digital Media • Mobile • Productivity • Social Media • Software
Easy Apply
In-Office
San Francisco, CA, USA
110 Employees
115K-125K Annually

Atlassian Logo Atlassian

Senior Manager, AI & Digital Natives

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
In-Office or Remote
San Francisco, CA, USA
11000 Employees
134K-209K Annually
Hybrid
10 Locations
1100 Employees
264K-423K Annually

Similar Companies Hiring

Golden Pet Brands Thumbnail
Digital Media • eCommerce • Information Technology • Marketing Tech • Pet • Retail • Social Media
El Segundo, California
178 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account