ColossalAI

An Open Source Machine Learning Framework for Everyone

Stable Diffusion web UI

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

Open Source Computer Vision Library