Senior/Staff AI Cloud Infra Engineer

Comfy Orgabout 2 months ago
San Francisco, CA, United States
Hybrid
Full-time
Junior Level (1-3 years)

Job Description

The Role:

We're looking for a Cloud Infrastructure Engineer who thrives on building and scaling large-scale GPU compute platforms fast. You'll be instrumental in developing and managing the foundational infrastructure that powers our AI workloads. Our core infrastructure relies heavily on Python, Kubernetes (K8s), Terraform, and Ansible, but we care more about your ability to learn, adapt, and ship robust solutions than whether you've used these exact tools before.

You are a good fit if this describes you:

  • You excel at building and managing distributed compute platforms, especially those involving GPU resources.
  • You have deep expertise in backend systems that orchestrate complex workloads efficiently, managing capacity and resource constraints.
  • You possess a strong understanding of foundational cloud infrastructure (AWS/GCP/Azure) and Linux provisioning/management tools.
  • You know how to design for reliability and scale with minimal operational overhead.
  • You learn new technologies rapidly because you're excited by solving hard infrastructure challenges.
  • You've scaled infrastructure before and understand the tradeoffs that matter.
  • You think most infrastructure moves too slowly and could be way better automated and optimized.
  • You're comfortable diving into unfamiliar systems and making them work reliably.
  • You are a self-starter who executes quickly, takes ownership, and constantly seeks improvement.

What you'll do:

  • Develop and maintain our core Python platform for routing requests, orchestrating AI workloads, managing GPU server capacity, observability, and more.
  • Develop and maintain our infrastructure layer using Terraform, Ansible, and cloud provider APIs to manage our fleet of GPU workers across cloud and potentially bare metal environments.
  • Own and operate the technologies underpinning our platform, potentially including K8s, FluxCD, Nomad, Prometheus, Thanos, Grafana, Loki, distributed networking/storage, etc.
  • Architect and implement solutions that directly impact the performance and availability of services for millions of ComfyUI users.
  • Work closely with our core engineering team to design and build new infrastructure systems.
  • Help create the vision and lay the foundation for where our infrastructure should go in the next 1/2/5 years.
  • Help shape our technical direction and infrastructure best practices as we grow.

Requirements:

  • Deep experience building and managing distributed compute platforms, preferably using Python.
  • Strong foundation in managing cloud infrastructure (AWS, GCP, or Azure). Experience with bare metal is a plus.
  • Solid understanding of container orchestration (Kubernetes preferred) and CI/CD principles and tools.
  • Excellent communication skills.
  • Proven ability to learn fast and ship quality infrastructure code and configurations.

Nice to have:

  • You have excelled at a fast-paced, high-growth tech startup before or are extremely excited about being in one.
  • Experience specifically with GPU management, scheduling, and monitoring in a large-scale environment.
  • Experience with specific observability tools (Prometheus, Grafana, Loki, Thanos).

Skills:

Amazon Web Services (AWS), Ansible, Application Programming Interface (API), Artificial Intelligence (AI), Best Practices, Capacity Management, Cloud Computing, Communication Skills, Computer Systems, Continuous Deployment/Delivery, Continuous Integration, Diving, Fleet Management, GCP (Good Clinical Practices), GPU (Graphics Processing Unit), Linux Operating System, Microsoft Windows Azure, Python Programming/Scripting Language, Server Capacity, Software Engineering, Startup, Systems Administration/Management

undefined:

Comfy Org

Required Skills

Systems Administration/Management
Cloud Computing
Continuous Deployment/Delivery
Fleet Management
Computer Systems
Capacity Management
Amazon Web Services (AWS)
Microsoft Windows Azure
Continuous Integration
Best Practices
GCP
Communication Skills
Python Programming/Scripting Language
GPU (Graphics Processing Unit)
Linux Operating System
Software Engineering
Application Programming Interface (API)
Diving
Artificial Intelligence (AI)
Startup
Ansible
Server Capacity