Senior ML Engineer — Ultra-Fast Inference & Memory

comfy-org; San Francisco (CA); On-site; USD 100,000 - 140,000; Full time

Job summary

comfy-org is seeking an AI Optimization Engineer in San Francisco to enhance model inference performance for ComfyUI. The role entails building and optimizing the core inference engine, ensuring models run faster and use less memory.

Ideal candidates are passionate about model optimization, have production experience in PyTorch, and enjoy tackling technical challenges in the visual AI domain. This position offers a unique opportunity to influence cutting-edge technology.

Qualifications

• Experience optimizing model inference and memory management.
• Proficiency in production PyTorch code.
• Passion for exploring how models function internally.

Responsibilities

• Build and optimize the core inference engine for ComfyUI.
• Enhance model performance and reduce memory usage.
• Collaborate with the core team on feature development.
• Address technical challenges in visual AI.
• Influence the direction of the technology.

Skills

Model inference optimization

Torch optimizations

Memory management

Production PyTorch code

Visual AI problems

Job description

comfy-org is seeking an AI Optimization Engineer in San Francisco to enhance model inference performance for ComfyUI. The role entails building and optimizing the core inference engine, ensuring models run faster and use less memory.

Ideal candidates are passionate about model optimization, have production experience in PyTorch, and enjoy tackling technical challenges in the visual AI domain. This position offers a unique opportunity to influence cutting-edge technology.