Principal Software Engineer, Stability AI

Architect

$190-290k

+ Stock options

AWS
TypeScript
Python
REST API
Expert level
Remote in US
Stability AI

Open source generative AI company

Open for applications

Stability AI

Open source generative AI company

101-200 employees

B2CB2BArtificial IntelligenceDeep TechMachine LearningAPI

Open for applications

$190-290k

+ Stock options

AWS
TypeScript
Python
REST API
Expert level
Remote in US

101-200 employees

B2CB2BArtificial IntelligenceDeep TechMachine LearningAPI

Company mission

To build the foundation to activate humanity's potential.

Role

Who you are

  • The ideal candidate will have experience in architecting and building REST APIs, hosting AI/ML workflows on HPC clusters, setting up AWS infrastructure and being a mentor for junior developers
  • 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 and working with multiple stakeholders
  • Experienced in root cause analysis and driving operational excellence initiatives of AI/ML services
  • Highly proficient in Python and Typescript

What the job involves

  • We are looking for a Principal Software Engineer (Architect) to be responsible for architecting, designing and developing our next generation of Gen AI API platform that supports multiple modalities like image, video, language, 3D and audio
  • Tech lead to drive architecture, design and develop AI/ML SAAS services and 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 that efficiently uses 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 like frontend team, product managers and technical leadership to implement new features
  • Lead system architecture design & decisions and help 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 micro services that are required to meet the needs of our API customers

Our take

Stability AI is an AI-driven visual art company that designs and implements an open AI tool that creates images based on text input. Through collective intelligence and augmented technology, Stability AI is helping research communities develop cutting-edge AI models for image, language, audio, video, 3D content, biotech and other scientific research.

Working in partnership with Amazon Web Services, Stability AI's latest release, Stable Diffusion is a text-to-image model that utilizes the Ezra-1 UltraCluster (the world's fifth-largest supercomputer). The model quickly proved to be a success, with four of the top ten applications on Apple's App Store being powered by Stable Diffusion after just one month.

Alongside Stable Diffusion, Stability AI offers premium imagining application, DreamStudio, as well as externally built products Lensa, Wonder and NightCafe, which have too amassed great success, with over 40M users and counting. The company is dedicated to investing in more supercomputing power in order to further accelerate its product offerings and is looking to expand its workforce, expecting to grow to over 300 employees by the end of the year.

Freddie headshot

Freddie

Company Specialist

Insights

Top investors

Some candidates hear
back within 2 weeks

324% employee growth in 12 months

Company

Funding (last 2 of 3 rounds)

Jun 2024

$80m

EARLY VC

Oct 2023

$50m

CONVERTIBLE

Total funding: $231m

Company values

  • Pragmatic and Impact-Driven: We are committed to finding practical solutions that make a positive difference.
  • Bold: We constantly push the boundaries of what's possible to create meaningful change and drive progress.
  • Collaborative: We believe that working with our partners, customers, and communities can achieve far more than we could alone.
  • Innovative: We are always exploring new ideas, experimenting with new technologies, and looking for fresh ways to tackle the world's most pressing challenges.
  • Ambitious: We never settle for the status quo. We constantly strive to do better, achieve more, and make a bigger impact in the world.
  • Transparent and Trustworthy: We are committed to being open and honest in all our interactions and building long-term partnerships with our customers, employees, and stakeholders based on mutual respect and trust.

Company HQ

Westminster, London, UK

Leadership

Emad Mostaque

(CEO)

Graduated from Oxford University with a master's degree in mathematics and computer science. Founder of Symmitree, a startup aimed at reducing the cost of technology.

Share this job

View 6 more jobs at Stability AI