YIYANGCAI

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

stable-diffusion

A latent text-to-image diffusion model

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

deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

datasets

🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API

PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

BERT-pytorch

Google AI 2018 BERT pytorch implementation

distiller

Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distiller

Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.

Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

Awesome-Pruning

A curated list of neural network pruning resources.

neural-compressor

SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX Runtime

intel-extension-for-transformers

⚡ Build your chatbot within minutes on your favorite device; offer SOTA compression techniques for LLMs; run LLMs efficiently on Intel Platforms⚡

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

bi-att-flow

Bi-directional Attention Flow (BiDAF) network is a multi-stage hierarchical process that represents context at different levels of granularity and uses a bi-directional attention flow mechanism to achieve a query-aware context representation without early summarization.

StyleGAN_PyTorch

The implementation of StyleGAN on PyTorch 1.0.1

nn_pruning

Prune a model while finetuning or training.

Handwriting-Transformers

Handwriting-Transformers (ICCV21)

mxfont

Official PyTorch implementation of MX-Font (Multiple Heads are Better than One: Few-shot Font Generation with Multiple Localized Experts) ICCV 2021

Model-Compression-Research-Package

A library for researching neural networks compression and acceleration methods.

FsFont

Official PaddlePaddle Implementation of Few-Shot Font Generation by Learning Fine-Grained Local Styles (FsFont)

HALP

prospr

Prospect Pruning: Finding Trainable Weights at Initialization Using Meta-Gradients

hangul-jamo

A library to compose and decompose Hangul syllables using Hangul jamo characters

Eyeudio

pytorch_mpiigaze_demo

Gaze estimation using MPIIGaze and MPIIFaceGaze