hlu1 Goto Github PK

aitemplate_public

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.

cpuinfo

CPU INFOrmation library (x86/ARM, Linux/Mach/NaCl)

cutlass

CUDA Templates for Linear Algebra Subroutines

dmlc-core

A common bricks library for building scalable and portable distributed machine learning.

keepingyouawake

Prevents your Mac from going to sleep.

mingpt

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

models

A repository for storing pre-trained Caffe2 models.

onnx

Open Neural Network Exchange

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

qnnpack

Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators

taso

A Tensor Algebra SuperOptimizer for Deep Learning

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

tvm-samples