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

yourfirm GmbH logo
yourfirm GmbH

Senior Fullstack Developer for AI-Driven Mission Technologies

Seeking a Senior Fullstack Developer for AI-driven mission technologies, focusing on Java, JavaScript, Python, and C++. Remote work available.

Catalyze Group logo
Catalyze Group

Full Stack Developer with AI and API Expertise

Join Catalyze Group as a Full Stack Developer to build AI-powered grant-writing tools. Work with React, Django, and more in Amsterdam.

Gorgias logo
Gorgias

Senior Full-Stack Engineer ReactJS/NodeJS

Join Gorgias as a Senior Full-Stack Engineer specializing in ReactJS and NodeJS, enhancing AI-powered ecommerce solutions.

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.

Reaktor logo
Reaktor

Lead Developer with DevOps and Functional Programming

Join Reaktor as a Lead Developer in Amsterdam, focusing on DevOps, Functional Programming, and JavaScript in a hybrid work environment.

RightCrowd logo
RightCrowd

Full Stack Engineer with Node.js and React

Join RightCrowd as a Full Stack Engineer to develop cloud-native applications using Node.js and React. Work remotely with cutting-edge technology.

doctari group logo
doctari group

Senior Full-Stack Engineer - TypeScript, React, Node.js

Join us as a Senior Full-Stack Engineer to develop a super app for medical professionals using TypeScript, React, and Node.js.

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.

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.

APPRAID TECH logo
APPRAID TECH

Software Architect with Rust and C++ Experience

Seeking a Software Architect with expertise in Rust and C++ for automotive safety-critical systems in Munich.

WECHEER logo
WECHEER

Full Stack Team Leader .Net

Lead a remote full-stack team with .NET and Angular expertise, focusing on technical leadership and hands-on development.

Lekker Code Company logo
Lekker Code Company

Fullstack Developer (C#, Angular or React)

Join Lekker Code Company as a Fullstack Developer specializing in C#, Angular, or React. Work remotely with a focus on innovative solutions.

Persona logo
Persona

LLM Backend Developer

Join Persona as a LLM Backend Developer, work remotely, and develop AI-driven backend systems for top startups.

Verimi logo
Verimi

Senior Software Architect (Java, Spring)

Join Verimi as a Senior Software Architect to design and implement digital identity solutions using Java, Spring, and cloud technologies.

ANWB logo
ANWB

Full-stack JavaScript Developer

Join ANWB as a Full-stack JavaScript Developer to enhance our online platform using JavaScript, TypeScript, and AWS technologies.

Brenntag logo
Brenntag

Senior API Platform Engineer

Join Brenntag as a Senior API Platform Engineer in Amsterdam to drive API platform development using Kubernetes, Istio, and AWS EKS.

FoodLabs logo
FoodLabs

Senior C++ Computer Vision Engineer

Join a cutting-edge AI-DeepTech startup in Berlin as a Senior C++ Computer Vision Engineer. Work on world-class on-device AI technology.

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.

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.

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.

Binance logo
Binance

Senior Backend Developer (Node.js) / SRE

Join Binance as a Senior Backend Developer (Node.js) / SRE to develop monitoring systems for high-load production environments.

iubenda logo
iubenda

Chief Technology Officer (CTO) - Ruby on Rails & JavaScript

Join iubenda as CTO to lead tech strategy in a SaaS scale-up. Expertise in Ruby on Rails, JavaScript, and cloud infrastructure required.

Instapro Group logo
Instapro Group

Senior Backend Engineer - Payments

Join Instapro Group as a Senior Backend Engineer in Berlin, focusing on PHP and payment systems in a hybrid work environment.

Carbon13 logo
Carbon13

Cofounder - Full Stack Developer/Data Scientist for Climatech Startup

Join Carbon13 as a cofounder in climate tech, leveraging AI, data science, and software development to combat climate change.