Jobs · Engineering · California

Senior/Staff Backend Engineer - Distributed System

Zettabyte · Palo Alto, CA · 2 mo ago
HybridEngineeringFull-time

About Us

At Zettabyte, we’re on a mission to make AI compute ubiquitous, seamless, and limitless. We’re building a cloud where AI just works—anywhere, anytime. “AI Power. Everywhere.” Be part of the team designing the infrastructure for the AI-first world.

Why this role exists

We need a Backend Engineer to build the systems that orchestrate GPU clusters for AI workloads. You'll create APIs that handle GPU allocation, memory management, compute scheduling, and multi-tenant isolation—challenges unique to AI infrastructure that go far beyond typical backend engineering.

What you’ll do

  • Design APIs that abstract complex GPU operations into simple developer experiences
  • Build scheduling algorithms that maximize GPU utilization while ensuring SLA compliance
  • Develop resource management systems for GPU lifecycle—provisioning, allocation, scheduling, and release
  • Create usage tracking and billing systems for GPU-hours, memory usage, and compute utilization
  • Implement monitoring for GPU-specific metrics, health checks, and automatic failure recovery
  • Create multi-tenancy systems with resource isolation, quota management, and fair scheduling
  • Optimize cold starts for model serving and implement efficient model loading strategies
  • Collaborate with frontend engineers to expose complex infrastructure through intuitive interfaces
  • Leverage AI-assisted coding tools (GitHub Copilot, Claude Code, Cursor IDE, etc.) to boost productivity and code quality

You’ll thrive here if you:

  • Have 5+ years backend engineering experience with distributed systems
  • Are proficient in Go, Python, or similar backend languages
  • Have experience with resource scheduling, orchestration, and API design (REST, GraphQL, gRPC)
  • Understand hardware constraints and system optimization
  • Have Linux systems knowledge and containerization experience (Docker, Kubernetes)
  • Are comfortable working with expensive resources where efficiency directly impacts costs
  • Are excited about solving novel problems in AI infrastructure (not just another CRUD app)
  • Have a startup mindset—comfortable with ambiguity and rapid iteration

Bonus Qualifications

  • Have GPU or HPC cluster management experience
  • Understand ML/AI workload patterns and requirements
  • Have experience with high-value resource allocation systems
  • Have background in performance optimization for compute-intensive workloads
  • Have familiarity with GPU virtualization and sharing technologies
  • Have experience building billing or metering systems

Similar jobs