Pinned Repositories
Books
CuAssembler
An unofficial cuda assembler, for all generations of SASS, hopefully ：）
cuda-samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
cuda_hgemm
Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruction.
DissectingTensorCores
flash-attention
Fast and memory-efficient exact attention
gpu-arch-microbenchmark
Dissecting NVIDIA GPU Architecture
HierarchicalKV
HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of Merlin-KV is to store key-value feature-embeddings on high-bandwidth memory (HBM) of GPUs and in host memory. It also can be used as a generic key-value storage.
KernelCodeGen
GEMM/FMHA CUDA/HIP kernel code generation using MLIR.
My-Leetcode-Solution
leetcode solutions as well as necessary data structure and algorithm
yao-jiashu's Repositories
yao-jiashu/KernelCodeGen
GEMM/FMHA CUDA/HIP kernel code generation using MLIR.
yao-jiashu/Books
yao-jiashu/CuAssembler
An unofficial cuda assembler, for all generations of SASS, hopefully ：）
yao-jiashu/cuda-samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
yao-jiashu/cuda_hgemm
Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruction.
yao-jiashu/DissectingTensorCores
yao-jiashu/flash-attention
Fast and memory-efficient exact attention
yao-jiashu/gpu-arch-microbenchmark
Dissecting NVIDIA GPU Architecture
yao-jiashu/HierarchicalKV
HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of Merlin-KV is to store key-value feature-embeddings on high-bandwidth memory (HBM) of GPUs and in host memory. It also can be used as a generic key-value storage.
yao-jiashu/llvm-project
The LLVM Project is a collection of modular and reusable compiler and toolchain technologies. Note: the repository does not accept github pull requests at this moment. Please submit your patches at http://reviews.llvm.org.
yao-jiashu/My-Leetcode-Solution
leetcode solutions as well as necessary data structure and algorithm
yao-jiashu/tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
yao-jiashu/nvbench
CUDA Kernel Benchmarking Library
yao-jiashu/pdfs
Technically-oriented PDF Collection (Papers, Specs, Decks, Manuals, etc)
yao-jiashu/PPoPP2017_artifact
Third party assembler and GEMM library for NVIDIA Kepler GPU
yao-jiashu/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.