t-vi

pytorch-tvmisc

Totally Versatile Miscellanea for Pytorch

candlegp

Gaussian Processes in Pytorch

acdl2020

lod2021

PyTorch Tutorial at the LOD2021 conference

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

lit_torchdrift

Lightning module for TorchDrift

kornia

Open Source Differentiable Computer Vision Library for PyTorch

lightning-bolts

Toolbox of models, callbacks, and datasets for AI/ML researchers.

lit-llama

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

minGPT

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

pytorch-lightning

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

bitsandbytes

8-bit CUDA functions for PyTorch

ghstack

Submit stacked diffs to GitHub on the command line

gptq

Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".

incubator-tvm-site

repo for tvm

koalitionsvertrag2021

Coalition agreement between the SPD, Green Party, and FDP as clean PDF file, .docx file, and .txt file

lightning-flash

Collection of tasks for fast prototyping, baselining, finetuning and solving problems with deep learning.

litgpt

Hackable implementation of state-of-the-art open-source LLMs based on nanoGPT. Supports flash attention, 4-bit and 8-bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

nbsphinx

:ledger: Sphinx source parser for Jupyter notebooks

nvfuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

Pillow

The friendly PIL fork (Python Imaging Library)

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

resource-stream

CUDA related news and material links

Tensile

Stretching GPU performance for GEMMs and tensor contractions.

transformers

🤗 Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

tutorials

PyTorch tutorials.

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

tvm-distro

Official autotvm log distro

unsloth

5X faster 60% less memory QLoRA finetuning

zulip

Zulip server and web app—powerful open source team chat