Machine Learning Engineer, Stability AI
Computer Vision
Open source generative AI company
Be an early applicant
Open source generative AI company
101-200 employees
Be an early applicant
101-200 employees
Company mission
To activate humanity's potential through generative AI
Role
Who you are
- MS or PhD in Computer Science
- 1+ years of experience working on machine learning projects
- Familiar with diffusion models
- Proficiency with Python scientific stack, PyTorch, creating Jupyter/Colab notebooks
- Strong coding skill and comfortable with coding in production platforms
- Ability to diagnose technical problems, debug code, and automate manual routine tasks
- Ability to communicate machine learning concepts and results effectively through writing and visualization
- Track record of publication in top-tier machine learning conferences
What the job involves
- We are looking for Machine Learning engineers who are passionate about generative models and the creative applications of AI to join our Applied ML team
- We look for highly innovative ML engineers with a strong blend of research and engineering skills, motivated to push the boundaries of image, video, and 3D foundation models and their applications
- In this role, you will collaborate with research and platform teams to build, improve, and optimize workflows that power state-of-the-art application solutions based on diffusion models
- You will have access to high-performance computing resources and work alongside top researchers and engineers, making a real impact in the rapidly growing field of generative AI
- Lead efforts to drive the design, develop, prototype and productionize ML systems that addresses the customer needs
- Collaborate with the research team on developing the next generation of models, where you may assist with areas such as optimization of model training, model fine-tuning, dataset engineering, tooling, open-source efforts, and etc
- Work on the commercial side, productionizing generative model-based solutions, by partnering with platform and research teams building the infrastructure to serve them at scale
- Write production-quality code and influence the next generation of Stability AI’s machine learning solutions
- Advocate for engineering best practices and upskill the team
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.
The company is also focused on expanding its creative sector offerings, recently hiring legendary filmmaker James Cameron to bring an artistic perspective to its suite of storytelling tools. This exciting development will 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
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 8 more jobs at Stability AI