Guocheng's repositories
MNISTDiffusion
Implement a MNIST(also minimal) version of denoising diffusion probabilistic model from scratch.The model only has 4.55MB.
resume-photo-maker
Make a resume photo with a simple python script and two lightweight deep neural networks. | 证件照制作。
SoftmaxFocalLoss
An intuitive pytorch implementation of softmax focal loss which can be used to train classification and segmentation tasks.
yolov5_pruning
Network pruning on yolov5.
kitti_baseline
The "Hello World" approch of KITTI Benchmark
Awesome-Diffusion-Models
A collection of resources and papers on Diffusion Models
carla_leaderboard
carla leaderboard 2.0
cube-studio
cube studio开源云原生一站式机器学习/深度学习AI平台,支持sso登录,多租户/多项目组,数据资产对接,notebook在线开发,拖拉拽任务流pipeline编排,多机多卡分布式算法训练,超参搜索,推理服务VGPU,多集群调度,边缘计算,serverless,标注平台,自动化标注,数据集管理,大模型一键微调,llmops,私有知识库,AI应用商店,支持模型一键开发/推理/微调,私有化部署,支持国产cpu/gpu/npu芯片,支持RDMA,支持pytorch/tf/mxnet/deepspeed/paddle/colossalai/horovod/spark/ray/volcano分布式
duckduckgo_search
Search for words, documents, images, videos, news, maps and text translation using the DuckDuckGo.com search engine. Downloading files and images to a local hard drive.
GPTInterviewer
📚 GPT Interviewer - Practice interview with AI interviewer based on job descriptions and resume
Gymnasium
An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
LAV
(CVPR 2022) A minimalistic mapless end-to-end stack for joint perception, prediction, planning and control for self driving.
llama-recipes
Scripts for fine-tuning Llama2 with composable FSDP & PEFT methods to cover single/multi-node GPUs. Supports default & custom datasets for applications such as summarization & question answering. Supporting a number of candid inference solutions such as HF TGI, VLLM for local or cloud deployment.Demo apps to showcase Llama2 for WhatsApp & Messenger
llama2.c
Inference Llama 2 in one file of pure C
mlops-course
A project-based course on the foundations of MLOps with a focus on intuition and application.
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
open-metric-learning
Library for metric learning pipelines
opencv
Open Source Computer Vision Library
ORB_SLAM3
ORB-SLAM3: An Accurate Open-Source Library for Visual, Visual-Inertial and Multi-Map SLAM
petals
🌸 Run large language models at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading
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
pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
sort
Simple, online, and realtime tracking of multiple objects in a video sequence.
spinningup
An educational resource to help anyone learn deep reinforcement learning.
tinygrad
You like pytorch? You like micrograd? You love tinygrad! ❤️
torchdistill
A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆20 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.
torchmetrics
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
tutorials
This repository contains tutorials and examples for Triton Inference Server
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch