1h 31m ago
Senior Machine Learning Engineer
pytorch
🏢On-site
|United States of AmericaSenior Machine Learning Engineer
Comfy | Artificial intelligence software
1h 31m agoOn-site
Full Time
IC
United States of America
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
If you've worked with diffusion/LLM models before or built custom nodes for ComfyUI, that's awesome
Required Skills
pytorch
Required Languages
🇬🇧 English