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

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.

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

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