Scale AI · 1 day ago
AI Infrastructure Engineer, Core Infrastructure
Scale AI is a company focused on developing reliable AI systems for critical decisions. They are seeking an AI Infrastructure Engineer to design and build foundational systems that power ML infrastructure, optimizing workloads across various compute environments and enhancing reliability and cost efficiency.
Artificial Intelligence (AI)Data Collection and LabelingGenerative AIImage RecognitionMachine Learning
Responsibilities
Design and maintain fault-tolerant, cost-efficient systems that manage compute allocation, scheduling, and autoscaling across clusters and clouds
Build common abstractions and APIs that unify job submission, telemetry, and observability across serving and training workloads
Develop systems for usage metering, cost attribution, and quota management, enabling transparency and control over compute budgets
Improve reliability and efficiency of large-scale GPU workloads through better scheduling, bin-packing, preemption, and resource sharing
Partner with ML engineers and API teams to identify bottlenecks and define long-term architectural standards
Lead projects end-to-end — from requirements gathering and design to rollout and monitoring — in a cross-functional environment
Qualification
Required
4+ years of experience building large-scale backend or distributed systems
Strong programming skills in Python, Go, or Rust, and familiarity with modern cloud-native architecture
Experience with containers and orchestration tools (Kubernetes, Docker) and Infrastructure as Code (Terraform)
Familiarity with schedulers or workload management systems (e.g., Kubernetes controllers, Slurm, Ray, internal job queues)
Understanding of observability and reliability practices (metrics, tracing, alerting, SLOs)
A track record of improving system efficiency, reliability, or developer velocity in production environments
Preferred
Experience with multi-tenant compute platforms or internal PaaS
Knowledge of GPU scheduling, cost modeling, or hybrid cloud orchestration
Familiarity with LLM or ML training workloads, though deep ML expertise is not required
Benefits
Comprehensive health, dental and vision coverage
Retirement benefits
A learning and development stipend
Generous PTO
A commuter stipend
Company
Scale AI
Scale’s mission is to develop reliable AI systems for the world’s most important decisions.
H1B Sponsorship
Scale AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (82)
2024 (54)
2023 (29)
2022 (17)
2021 (10)
2020 (10)
Funding
Current Stage
Late StageTotal Funding
$15.9BKey Investors
MetaAccelTiger Global Management
2025-06-10Corporate Round· $14.3B
2025-06-04Series Unknown
2024-05-21Series F· $1B
Recent News
Crunchbase News
2026-01-07
Company data provided by crunchbase