A general fine-tuning kit geared toward diffusion models.

OneDiff: An out-of-the-box acceleration library for diffusion models.

Various AI scripts. Mostly Stable Diffusion stuff.

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

SkyPilot: Run LLMs, AI, and Batch jobs on any cloud. Get maximum savings, highest GPU availability, and managed execution—all with a simple interface.

An open source implementation of CLIP.

NVIDIA GPU metrics exporter for Prometheus leveraging DCGM

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Music Style Transfer with Transformers

This repo contains code and a pre-trained model for clothes segmentation.

RetinaFace: Deep Face Detection Library for Python

Taming Transformers for High-Resolution Image Synthesis