Stability AI est une bibliothèque de modèles de génération open source offrant des fonctionnalités d'entraînement, d'inférence et d'application pour divers modèles génératifs. Elle prend en charge l'entraînement de nombreux modèles génératifs, y compris l'entraînement basé sur PyTorch Lightning, et propose de nombreuses options de configuration ainsi qu'une architecture modulaire. Les utilisateurs peuvent l'employer pour entraîner des modèles génératifs, puis effectuer des inférences et des applications grâce aux modèles fournis. La bibliothèque inclut également des exemples de configurations d'entraînement et des fonctionnalités de traitement de données pour faciliter la prise en main et la personnalisation.