Pinned Repositories
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's Repositories
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