Zejun H.'s 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.
Language:PythonApache-2.0000
Language:Jupyter Notebook000
llama2.c
Inference Llama 2 in one file of pure C
Language:PythonMIT000
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:PythonNOASSERTION000