Repository Index

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

• python
• deep-learning
• artificial-intelligence
• ai

Instant neural graphics primitives: lightning fast NeRF and more

• neural-network
• machine-learning
• cuda
• nerf

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

• ai
• artificial-intelligence
• deep-learning
• large-language-models

so-vits-svc fork with realtime support, improved interface and more features.

• sovits
• vits
• voice-conversion
• so-vits-svc

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

• pytorch
• pytorch-lightning
• stable-diffusion
• text-to-image

Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild. Our new online demo is also released at suppixel.ai.

• deep-learning
• diffusion-models
• llava
• sdxl

Lightning fast C++/CUDA neural network framework

• neural-network
• deep-learning
• gpu
• cuda

[EMNLP 2022] An Open Toolkit for Knowledge Graph Extraction and Construction

• knowledge-graph
• relation-extraction
• chinese
• named-entity-recognition

A minimal Python framework for building custom AI inference servers with full control over logic, batching, and scaling.

• ai
• api
• serving
• artificial-intelligence

[NeurIPS 2024] Official code for PuLID: Pure and Lightning ID Customization via Contrastive Alignment

Transform ML models into a native code (Java, C, Python, Go, JavaScript, Visual Basic, C#, R, PowerShell, PHP, Dart, Haskell, Ruby, F#, Rust) with zero dependencies

• machine-learning
• scikit-learn
• statistical-learning
• xgboost

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

• nerf
• neural-radiance-fields
• view-synthesis
• pytorch

Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

• reinforcement-learning
• deep-learning
• pytorch
• pytorch-reinforcement-learning

An Easy To Use PyTorch Computer Vision Library

• computer-vision
• deep-learning
• pytorch
• pytorch-lightning