Apply on Employer Site

StackAI · 4 months ago

AI Infrastructure Engineer

SF Office - 171 2nd, 4th floor

Full-time

Onsite

Mid Level

4+ years exp

Stack-ai is a fast-growing Series A startup focused on shaping the future of AI in enterprise workflows. They are seeking an AI Infrastructure Engineer to design and implement scalable backend architectures for their AI platform, manage distributed job orchestration, and collaborate with ML engineers to enhance production systems.

AppsArtificial Intelligence (AI)Developer PlatformSoftware

H1B Sponsor Likely

Responsibilities

Design and implement scalable backend architectures tailored for AI workloads (inference, orchestration, monitoring)

Manage and optimize distributed job orchestration using Celery, Temporal, RabbitMQ, and Celery Beat

Enhance data pipelines and caching strategies using Redis

Collaborate closely with ML engineers to integrate models into production, ensuring scalability and reliability

Build robust monitoring, observability, retry, and fault tolerance systems around job execution

Support infrastructure management and incident response to ensure uptime and performance

Contribute to platform infrastructure and tooling to support Stack AI’s rapid growth trajectory

Qualification

Backend engineeringDistributed systemsJob orchestrationRedisCeleryRabbitMQCI/CD practicesDockerKubernetesTerraformMLOps pipeline

Required

4+ years of backend engineering experience (Python preferred)

Deep expertise in task queues, job orchestration, and distributed systems

Hands-on experience with Redis, Celery, RabbitMQ, Celery Beat, and ideally Temporal

Experience scaling systems at a startup or in fast-paced environments

Strong understanding of deploying, monitoring, and optimizing AI/ML systems in production (infrastructure toolsets and CI/CD practices)

Familiarity with containerization (Docker, Kubernetes), IaC (Terraform), or MLOps pipeline

Company

StackAI

Build and deploy Enterprise-Grade AI Agents.

Founded in 2023

Cambridge, Massachusetts, USA

51-200 employees

https://stack-ai.com

H1B Sponsorship

StackAI 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 (4)