Tensors and Dynamic neural networks in Python with strong GPU acceleration

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.

Animation engine for explanatory math videos

The simplest, fastest repository for training/finetuning medium-sized GPTs.

GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.

The best ChatGPT that $100 can buy.

The official Meta Llama 3 GitHub site

Open-Sora: Democratizing Efficient Video Production for All

Graph Neural Network Library for PyTorch

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

The open source developer platform to build AI agents and models with confidence. Enhance your AI applications with end-to-end tracking, observability, and evaluations, all in one integrated platform.

State-of-the-Art Text Embeddings

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

This project aim to reproduce Sora (Open AI T2V model), we wish the open source community contribute to this project.

A command-line productivity tool powered by AI large language models like GPT-4, will help you accomplish your tasks faster and more efficiently.

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

An open access book on scientific visualization using python and matplotlib

PyTorch package for the discrete VAE used for DALL·E.

🐍 Geometric Computer Vision Library for Spatial AI

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

An Easy-to-use, Scalable and High-performance RLHF Framework based on Ray (PPO & GRPO & REINFORCE++ & vLLM & Ray & Dynamic Sampling & Async Agentic RL)

🔥 2D and 3D Face alignment library build using pytorch

[NeurIPS 2024] Depth Anything V2. A More Capable Foundation Model for Monocular Depth Estimation

Code release for ConvNeXt model

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

The Enterprise-Grade Production-Ready Multi-Agent Orchestration Framework. Website: https://swarms.ai