Machine Learning Engineer, Stability AI
Audio
Open source generative AI company
Job no longer available
Open source generative AI company
101-200 employees
Job no longer available
101-200 employees
Company mission
To activate humanity's potential through generative AI.
Role
Who you are
- 5+ years working on machine learning projects, including inference and pipeline development
- Solid knowledge of Python scientific stack, PyTorch and at least one high-performance inference framework (e.g. TensorRT)
- Experience profiling and optimizing deep neural networks, including knowledge of GPU profiling tools such as NVIDIA Nsight
- Experience with Python audio processing libraries such as librosa, torchaudio, or similar
- Experience with cloud orchestration systems such as Kubernetes and cloud providers such as AWS, GCP, and Azure
- Ability to write robust and maintainable client-server architectures and APIs
- Ability to rapidly prototype solutions and iterate on them with tight product deadlines
- Experience with training and/or deploying ML models with Amazon AWS (Sagemaker a plus) or Google Cloud
- Strong communication, collaboration, and documentation skills
- Experience building interactive web demos that serve generative ML models
- Experience with the open-source ML ecosystem (HuggingFace, W&B, etc.)
- Experience with Linux and command line tools
- Evidence of interest in music / audio projects is valued
What the job involves
- Lead efforts to drive the design, development and production of customer-facing ML music, speech and audio generation systems, with specific reference to inference and API environments
- Work with the Audio, Platform and Inference teams on building pipelines for the next generation of models, where you may assist with areas such as optimization, model tuning and deployment, HPC clusters, and tooling
- Be a strategic thought partner for leaders across the organization on driving business impact through machine learning
- Work on the commercial side - productionizing generative models, and building the infrastructure to serve them at scale
- Produce events and metrics in our data warehouse so that we can analyze critical business metrics like cost, performance, reliability, etc
- Be part of the team that brings new Stability audio models and pipelines into existence for API customers
- Prototype and productionize inference platform improvements and new features
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 Electronic Arts (EA), Stability AI is co-developing next-generation generative AI model and workflows designed to empower artists, designers, and developers across interactive entertainment. By embedding its research teams directly alongside EA's creative talent, Stability AI is advancing image-to-3D, text-to-3D, and volumetric media tools that accelerate world-building, rapid prototyping, and visual storytelling. This collaboration highlights Stability AI's growing leadership i multimodal and 3D generative AI, and its mission to put powerful, creator-centric tools in the hands of global creative industries.
The company is also focused on expanding its creative sector offerings, hiring legendary filmmaker James Cameron to bring an artistic perspective to its suite of storytelling tools. This exciting development will further help Stability AI push the boundaries of AI-driven media and help on its mission to transform visual storytelling.
Freddie
Company Specialist at Welcome to the Jungle
Company
Funding (last 2 of 3 rounds)
Jun 2024
$80m
EARLY VC
Oct 2023
$50m
CONVERTIBLE
Total funding: $231m
Company HQ
Notting Hill, London, UK
Share this job
View 7 more jobs at Stability AI