Stability AI logo

Lead Architect - Gen AI API Platform

Stability AI

About the Role

We are seeking a Lead Architect to spearhead the architecture, design, and development of our next-generation Gen AI API platform. This platform supports multiple modalities, including image, video, language, 3D, and audio. The ideal candidate will have extensive experience in architecting and building REST APIs, hosting AI/ML workflows on HPC clusters, setting up AWS infrastructure, and mentoring junior developers.

Responsibilities

  • Serve as the technical lead to drive architecture, design, and development of AI/ML SaaS services.
  • Set up inference on an HPC cluster for multiple modalities on a common Gen AI platform.
  • Build robust application backends on AWS infrastructure that support highly available AI/ML services at high scale, efficiently utilizing GPU clusters.
  • Define comprehensive API specifications and documentation.
  • Deliver customer-facing services, including account management, identity, single-sign-on, subscription billing, and self-service support tools, integrating with existing internal systems where necessary.
  • Collaborate with stakeholders such as the frontend team, product managers, and technical leadership to implement new features.
  • Lead system architecture design and decisions, helping drive consensus.
  • Manage large compute clusters for ML inference and development.
  • Deliver and manage our developer and researcher productivity tools, including CI/CD pipelines for deploying new machine learning models, orchestration, continuous/progressive deployments, test environments, feature flags, and GitHub.
  • Own the orchestration, deployments, middleware, and any other microservices required to meet the needs of our API customers.

Qualifications

  • 10+ years of experience in building REST APIs and backend infrastructure on AWS.
  • Experienced in designing and building AI/ML infrastructure and working with large GPU clusters, preferably in multiple modalities like image, video, audio, and language.
  • Distributed system architecture design knowledge and experience with delivering high traffic and highly available SaaS type services.
  • Well-versed in data structures, data modeling, and database management systems, billing and metering, as well as object and file storage systems.
  • Experienced in mentoring other engineers and collaborating with multiple stakeholders.
  • Experienced in root cause analysis and driving operational excellence initiatives of AI/ML services.
  • Highly proficient in Python and Typescript.

Compensation

The salary range for this role is between $190,000 and $250,000. Total compensation also includes stock options and benefits.

Equal Employment Opportunity

We are an equal opportunity employer and do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability, or other legally protected statuses.

Benefits
Extracted with AI

  • Stock options
  • Remote work flexibility

Similar jobs

Last update: 23 minutes ago

Aiven logo
Aiven

Staff Software Engineer

Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.

doctari group logo
doctari group

Technical Lead - SuperApp (Typescript React Node.js)

Join us as a Technical Lead to build a next-gen healthcare platform using TypeScript, React, and AWS. Lead a team in a hybrid work environment.

n8n logo
n8n

Senior Software Engineer (Node.js & TypeScript)

Join n8n as a Senior Software Engineer to build AI applications using Node.js and TypeScript. Remote role within Europe.

Together AI logo
Together AI

Senior Backend Engineer - Java, Rust, Go

Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.

9am logo
9am

Senior Solution Architect AWS (Freelance)

Seeking a Senior Solution Architect AWS for a freelance role in Frankfurt, Germany. Remote work available. Extensive AWS experience required.

Climatiq logo
Climatiq

Senior Full Stack Engineer - Climate Tech - Rust & TypeScript

Join Climatiq as a Senior Full Stack Engineer to develop climate tech solutions using Rust and TypeScript. Remote work available.

Cere Network logo
Cere Network

Principal AI Engineer

Join Cere Network as a Principal AI Engineer to drive AI innovation in Web3. Requires 10+ years in AI/ML, NLP, and software development.

Labelbox logo
Labelbox

Full-Stack Engineer with Angular and React.js

Join Labelbox as a Full-Stack Engineer to develop scalable systems using Angular, React.js, and GraphQL. Work remotely in a dynamic AI-driven environment.

LEGALFLY logo
LEGALFLY

Back End Engineer with Node.js and TypeScript

Join LegalFly as a Back End Engineer to revolutionize legal AI with Node.js and TypeScript in a hybrid role in Ghent.

dataroots logo
dataroots

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

Attio logo
Attio

Senior Product Engineer [Rust & Typescript]

Join Attio as a Senior Product Engineer working with Rust & TypeScript to build innovative CRM features. Remote work available.

ZAUBAR logo
ZAUBAR

Senior Fullstack & Unity Developer (Gen AI, AR)

Join ZAUBAR as a Senior Fullstack & Unity Developer in Berlin to create immersive AR experiences with GenAI. Work on cutting-edge technology in a dynamic team.

BCG X logo
BCG X

AI Engineer

Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.

Nebius AI logo
Nebius AI

Senior Backend Engineer (Go)

Join Nebius as a Senior Backend Engineer (Go) to develop fault-tolerant cloud services in a hybrid work environment.

DeepL logo
DeepL

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

Motius logo
Motius

Senior Backend Developer

Join Motius as a Senior Backend Developer to work on cutting-edge R&D projects using AWS, Docker, GraphQL, and more in a hybrid work environment.

NN Group logo
NN Group

Senior Full-stack Engineer (Angular, Node.js, TypeScript)

Join NN Group as a Senior Full-stack Engineer, leading software architecture and development with Angular, Node.js, and TypeScript.

OnHires logo
OnHires

Senior Full Stack Engineer (PHP, Angular, React)

Seeking a Senior Full Stack Engineer with PHP, Angular, React expertise for remote work in the EU. 6+ years experience required.

Aleph logo
Aleph

Frontend Engineer, AI

Join Aleph as a Frontend Engineer focusing on AI to develop innovative features using React.js and AI technologies in a remote role.

Pass App logo
Pass App

Senior Software Engineer (Backend)

Join Pass App as a Senior Software Engineer (Backend) to build scalable web3 solutions using microservices, TypeScript, and Node.js.

osapiens logo
osapiens

Senior Full-Stack Developer - TypeScript

Join osapiens as a Senior Full-Stack Developer using TypeScript to build sustainable SaaS solutions.

Basetime BV logo
Basetime BV

Senior Python Developer with AWS Experience

Join Basetime BV as a Senior Python Developer to develop and maintain AWS cloud solutions. Hybrid work, competitive salary, and growth opportunities.

UKG logo
UKG

Lead AI Full Stack Developer

Lead AI Full Stack Developer role in Alpharetta, GA, focusing on AI-driven applications using GCP, full-stack development, and MLOps.

Computer Futures logo
Computer Futures

Cloud Data Engineer

Seeking a Cloud Data Engineer with expertise in AWS, Python, and CI/CD for a hybrid role in Hannover. Join our dynamic team!