Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
Fast and memory-efficient exact attention
CUDA Templates for Linear Algebra Subroutines