Lists (3)
Sort Name ascending (A-Z)
Starred repositories
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
TuShare is a utility for crawling historical data of China stocks
Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]
Code for the manim-generated scenes used in 3blue1brown videos
A concise but complete full-attention transformer with a set of promising experimental features from various papers
Denoising Diffusion Probabilistic Models
[CVPR 2023 Best Paper Award] Planning-oriented Autonomous Driving
LightGlue: Local Feature Matching at Light Speed (ICCV 2023)
Code examples in pyTorch and Tensorflow for CS230
On-device AI across mobile, embedded and edge for PyTorch
Vector (and Scalar) Quantization, in Pytorch
PyTorch native quantization and sparsity for training and inference
TurboDiffusion: 100–200× Acceleration for Video Diffusion Models
A unified library of SOTA model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks …
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)
[pip install medmnist] 18x Standardized Datasets for 2D and 3D Biomedical Image Classification
A batched offline inference oriented version of segment-anything