Stability AI

StableLM

StableLM: Stability AI Language Models

StableCode

Code Assistance/ Developer Productivity suite of Models

rest-api-support

Stability REST API examples, issues, and discussions | https://api.stability.ai

stability-sdk-go

Golang functions for interacting with Stability API

flash-attention

Fast and memory-efficient exact attention

deep-learning-containers

AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.

stable-audio-demo

Stable-Grounded-Segment-Anything

Grounded-SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything

ComfyUI-Impact-Pack

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

hyperpod

Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.

data-juicer

A one-stop data processing system to make data higher-quality, juicier, and more digestible for LLMs! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷为大语言模型提供更高质量、更丰富、更易”消化“的数据！

iam-credentials-api-proxy

Localhost proxy for IAM Credentials API project

opentelemetry-collector-contrib

Contrib repository for the OpenTelemetry Collector

pycudwt

A Python-CUDA implementation of the Wavelet Transform

stable-3d-gallery

tensorizer

Module, Model, and Tensor Serialization/Deserialization

aws-dlc-examples

Examples for Stability AI Deep Learning Containers on AWS SageMaker

stable-code-docs

stable-piq

Measures and metrics for image2image tasks. PyTorch.

alignment-handbook

Robust recipes for to align language models with human and AI preferences

demos

Resources for demos and training

devcontainer-features

Custom features shared by Stability AI devcontainers

epochraft

Checkpointable dataset utilities for foundation model training

epochraft-hf-fsdp

Example of using Epochraft to train HuggingFace transformers models with PyTorch FSDP

InternalForkComfyUI

The most powerful and modular stable diffusion GUI with a graph/nodes interface.

mup

maximal update parametrization (µP)

slurm-gcp

Slurm on Google Cloud Platform

stabilityai-demos

stable-vllm

A high-throughput and memory-efficient inference and serving engine for LLMs