Ethan He's repositories
channel-pruning
Channel Pruning for Accelerating Very Deep Neural Networks (ICCV'17)
epipolar-transformers
Epipolar Transformers (best paper award, CVPR 2020 workshop)
softer-NMS
Bounding Box Regression with Uncertainty for Accurate Object Detection (CVPR'19)
Modern-Compiler-Implementation-in-C
book and codes for Modern Compiler Implementation in C
resnet-cifar10-caffe
ResNet-20/32/44/56/110 on CIFAR-10 with Caffe
resnet-imagenet-caffe
train resnet on imagenet from scratch with caffe
3D-reconstruction
two view structure from motion
Estimated-Depth-Map-Helps-Image-Classification
Depth estimation with neural network, and learning on RGBD images
Xception-caffe
Xception implemented with caffe
mini-ImageNet
Generate mini-ImageNet with ImageNet for fewshot learning
Objects-Detection-with-YOLO-on-Artwork-Dataset
Objects Detection with YOLO on Artwork Dataset ( you only look once system)
yihui-he.github.io
my personal website
lip-tracking-with-snake-active-contour-and-particle-filter
object tracking from scratch
ChatGPT-Next-Web
One-Click to deploy well-designed ChatGPT web UI on Vercel. 一键拥有你自己的 ChatGPT 网页服务。
idleminer.py
mine crypto when GPU is idle
TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Megatron-LM
Ongoing research training transformer models at scale
NeMo
NeMo: a toolkit for conversational AI
NeMo-Megatron-Launcher-1
NeMo Megatron launcher and tools
scattermoe
Triton-based implementation of Sparse Mixture of Experts.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.