Skip to content

@Stability-AI Stability AI

Change the repository type filter

Forks

    Repositories list

    49 repositories

    • deep-learning-containers

      Public
      AWS Deep Learning Containers (DLCs) are a set of Docker images for training and serving models in TensorFlow, TensorFlow 2, PyTorch, and MXNet.
      Python
      •
      Other
      •483•10•0•11•Updated Feb 11, 2025Feb 11, 2025

    • stability-hpc

      Public
      Deploy your HPC Cluster on AWS in 20min. with just 1-Click.
      Shell
      •
      MIT No Attribution
      •44•53•0•0•Updated Jan 24, 2025Jan 24, 2025

    • LLaMA-Factory

      Public
      Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
      Python
      •
      Apache License 2.0
      •5.3k•2•0•0•Updated Jan 17, 2025Jan 17, 2025

    • ComfyUI-Advanced-ControlNet

      Public
      ControlNet scheduling and masking nodes with sliding context support
      Python
      •
      GNU General Public License v3.0
      •71•0•0•0•Updated Oct 31, 2024Oct 31, 2024

    • comfyui_controlnet_aux

      Public
      ComfyUI's ControlNet Auxiliary Preprocessors
      Python
      •
      Apache License 2.0
      •237•3•0•1•Updated Oct 31, 2024Oct 31, 2024

    • truss

      Public
      Serve any model without boilerplate code
      Python
      •
      MIT License
      •83•2•0•11•Updated Oct 25, 2024Oct 25, 2024

    • TransformerEngine

      Public
      A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
      Python
      •
      Apache License 2.0
      •377•2•0•0•Updated Sep 27, 2024Sep 27, 2024

    • lm-evaluation-harness

      Public
      A framework for few-shot evaluation of autoregressive language models.
      Python
      •
      MIT License
      •2.2k•149•4•6•Updated Sep 13, 2024Sep 13, 2024

    • sagemaker-python-sdk

      Public
      A library for training and deploying machine learning models on Amazon SageMaker
      Python
      •
      Apache License 2.0
      •1.2k•13•0•5•Updated Sep 9, 2024Sep 9, 2024

    • rembg

      Public
      Rembg is a tool to remove images background
      Python
      •
      MIT License
      •2k•1•0•0•Updated Aug 13, 2024Aug 13, 2024

    • stability-sdk-ab

      Public
      SDK for interacting with stability.ai APIs (e.g. stable diffusion inference)
      Jupyter Notebook
      •
      MIT License
      •344•1•0•1•Updated Jul 30, 2024Jul 30, 2024

    • llm-leaderboard

      Public
      Project of llm evaluation to Japanese tasks
      Python
      •38•0•0•8•Updated Jul 25, 2024Jul 25, 2024

    • FastChat

      Public
      An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.
      Python
      •
      Apache License 2.0
      •4.7k•45•3•1•Updated Jul 5, 2024Jul 5, 2024

    • Comfyui_Lama

      Public
      Python
      •15•2•0•1•Updated Jun 21, 2024Jun 21, 2024

    • invisible-watermark-gpu

      Public
      python library for invisible image watermark (blind image watermark)
      Python
      •
      MIT License
      •153•59•0•0•Updated Jun 4, 2024Jun 4, 2024

    • k-diffusion

      Public
      Karras et al. (2022) diffusion models for PyTorch
      Python
      •
      MIT License
      •383•17•0•4•Updated May 6, 2024May 6, 2024

    • tensorizer

      Public
      Module, Model, and Tensor Serialization/Deserialization
      Python
      •
      MIT License
      •38•2•0•4•Updated Apr 11, 2024Apr 11, 2024

    • ComfyUI-Impact-Pack

      Public
      Python
      •
      GNU General Public License v3.0
      •225•3•0•0•Updated Feb 9, 2024Feb 9, 2024

    • hyperpod

      Public
      Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
      Python
      •
      MIT No Attribution
      •101•4•0•0•Updated Feb 1, 2024Feb 1, 2024

    • opentelemetry-collector-contrib

      Public
      Contrib repository for the OpenTelemetry Collector
      Go
      •
      Apache License 2.0
      •2.6k•2•0•186•Updated Jan 31, 2024Jan 31, 2024

    • slurm-gcp

      Public
      Slurm on Google Cloud Platform
      HCL
      •
      Apache License 2.0
      •119•0•0•0•Updated Jan 23, 2024Jan 23, 2024

    • stable-vllm

      Public
      A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      •
      Apache License 2.0
      •6.2k•2•0•0•Updated Jan 10, 2024Jan 10, 2024

    • DeepSpeed

      Public
      DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
      Python
      •
      Apache License 2.0
      •4.3k•4•0•0•Updated Dec 15, 2023Dec 15, 2023

    • InternalForkComfyUI

      Public
      The most powerful and modular stable diffusion GUI with a graph/nodes interface.
      Python
      •
      GNU General Public License v3.0
      •7.6k•0•0•0•Updated Dec 11, 2023Dec 11, 2023

    • data-juicer

      Public
      A one-stop data processing system to make data higher-quality, juicier, and more digestible for LLMs! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷为大语言模型提供更高质量、更丰富、更易”消化“的数据！
      Python
      •
      Apache License 2.0
      •216•1•0•0•Updated Dec 4, 2023Dec 4, 2023

    • pycudwt

      Public
      A Python-CUDA implementation of the Wavelet Transform
      C++
      •
      Other
      •11•3•0•0•Updated Nov 30, 2023Nov 30, 2023

    • epochraft-hf-fsdp

      Public
      Example of using Epochraft to train HuggingFace transformers models with PyTorch FSDP
      Python
      •
      MIT License
      •5•0•0•0•Updated Nov 9, 2023Nov 9, 2023

    • mup

      Public
      maximal update parametrization (µP)
      Jupyter Notebook
      •
      MIT License
      •94•0•0•0•Updated Oct 23, 2023Oct 23, 2023

    • epochraft

      Public
      Checkpointable dataset utilities for foundation model training
      Python
      •
      MIT License
      •5•0•0•0•Updated Sep 25, 2023Sep 25, 2023

    • Stable-Grounded-Segment-Anything

      Public
      Grounded-SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
      Jupyter Notebook
      •
      Apache License 2.0
      •1.5k•9•0•0•Updated Sep 5, 2023Sep 5, 2023
    49 repositories found. List is sorted by Last pushed in descending order.