Taoshu's starred repositories
Pytorch-XNOR-Net
XNOR-Net, with binary gemm and binary conv2d kernels, support both CPU and GPU.
binary-networks-pytorch
Binarize convolutional neural networks using pytorch :fire:
kernel_tuner
Kernel Tuner
NVIDIA_SGEMM_PRACTICE
Step-by-step optimization of CUDA SGEMM
graphlearn-for-pytorch
A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.
neural-compressor
SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX Runtime
ipex-llm
Accelerate local LLM inference and finetuning (LLaMA, Mistral, ChatGLM, Qwen, Baichuan, Mixtral, Gemma, Phi, MiniCPM, etc.) on Intel CPU and GPU (e.g., local PC with iGPU, discrete GPU such as Arc, Flex and Max); seamlessly integrate with llama.cpp, Ollama, HuggingFace, LangChain, LlamaIndex, GraphRAG, DeepSpeed, vLLM, FastChat, Axolotl, etc.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
LLMSpeculativeSampling
Fast inference from large lauguage models via speculative decoding
Megatron-LM
Ongoing research training transformer models at scale
tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
vsc-cec-ide
一个插件,国产化你的VSCode,来源于CEC-IDE,有敏感词检测、防沉迷等功能。
pretrain-gnns
Strategies for Pre-training Graph Neural Networks