About the job
The Opportunity
Join us at ComfyOrg as a Senior/Staff Applied Machine Learning Engineer! We are on the hunt for a passionate innovator who is enthusiastic about optimizing model inference. You will play a pivotal role in developing the heart of ComfyUI, our cutting-edge visual AI platform. Your expertise will help us push the limits of AI model performance, making them run faster and more efficiently than ever before.
Are You a Match?
You are fascinated by model inference, memory management, and torch optimizations.
You possess experience in writing production-level PyTorch code that challenges performance standards.
You have a passion for understanding the inner workings of AI models.
You thrive on developing highly optimized code that consistently delivers results.
You believe that the current landscape of ML deployment holds significant room for improvement.
Your Responsibilities:
Develop and enhance the core inference engine that drives ComfyUI.
Optimize large models for speed and memory efficiency.
Collaborate with our core team to architect new features.
Tackle complex technical challenges within the visual AI domain.
Contribute to the future direction of our technology.
Experience with diffusion or LLM models, as well as creating custom nodes for ComfyUI, is highly beneficial.

