zejunh

AITemplate

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.

opacus

Training PyTorch models with differential privacy

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

AITemplate

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.

boneDetection

llama2.c

Inference Llama 2 in one file of pure C

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

zejunh/AITemplate

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.

zejunh/boneDetection

zejunh/llama2.c

Inference Llama 2 in one file of pure C

zejunh/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration