Juncheng'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.
openfold
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
Language:PythonApache-2.0000