wxthu

Pinned Repositories

tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Language:Python11.7k 376 3.4k3.5k
ColossalAI
Making large AI models cheaper, faster and more accessible
Language:Python38.7k 383 1.7k4.3k
onnxruntime
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
Language:C++14.4k 245 6.5k2.9k
TensorRT
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
Language:C++10.7k 157 3.7k2.1k
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python82.9k 1.7k 45.6k22.3k
custom_matmul_kernels
Customized matrix multiplication kernels
Language:Jupyter Notebook0 0 00
mDNN_inference
Language:C++0 2 00
mmGraph
research only
Language:C++0 1 00
multi-yolov5
This project is only for scientific research
Language:Python0 1 00
new_yaxin
Language:Python0 1 00

wxthu's Repositories

wxthu/custom_matmul_kernels
Customized matrix multiplication kernels
Language:Jupyter Notebook0 0 00
wxthu/debug
Language:Cuda0 1 00
wxthu/mDNN_inference
Language:C++0 2 00
wxthu/mmGraph
research only
Language:C++0 1 00
wxthu/multi-yolov5
This project is only for scientific research
Language:Python0 1 00
wxthu/new_yaxin
Language:Python0 1 00
wxthu/onnx
Open standard for machine learning interoperability
Language:C++0 0 00