YongfeiYan

ollama

Get up and running with Llama 3.1, Mistral, Gemma 2, and other large language models.

okcoin-leeks-reaper

OKCoin韭菜收割机

lightweight-charts

Performant financial charts built with HTML5 canvas

machine-learning-for-trading

Code for Machine Learning for Algorithmic Trading, 2nd edition.

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

stylegan

StyleGAN - Official TensorFlow Implementation

gaussian-splatting

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

MVSNet

MVSNet (ECCV2018) & R-MVSNet (CVPR2019)

PromptPapers

Must-read papers on prompt-based tuning for pre-trained language models.

awesome-knowledge-distillation

Awesome Knowledge Distillation

A-Survey-on-Generative-Diffusion-Model

unidiffuser

Code and models for the paper "One Transformer Fits All Distributions in Multi-Modal Diffusion"

open-diffusion

Simple large-scale training of stable diffusion with multi-node support.

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.

accelerate

🚀 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

flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework

diffusion-deploy

stable-diffusion

Latent Text-to-Image Diffusion

Torch-Pruning

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

Diff-Pruning

[NeurIPS 2023] Structural Pruning for Diffusion Models

pytorch-mobilenet-v2

A PyTorch implementation of MobileNet V2 architecture and pretrained model.

linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

torchview

torchview: visualize pytorch models

taming-transformers

Taming Transformers for High-Resolution Image Synthesis

Paddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）

CodeSamples

Code samples for the CUDA tutorial "CUDA and Applications to Task-based Programming"

spotinfo

CLI for exploring AWS EC2 Spot inventory. Inspect AWS Spot instance types, saving, price, and interruption frequency.

py-spotcheck

A simple CLI tool to check the spot prices of AWS instances

bazel-libtorch-cpp

instant-nerf-pytorch

NeRF trained SUPER FAST in pytorch