jianyuh

HowToOptimizeGemm

about

asmjit

Complete x86/x64 JIT and AOT Assembler for C++

blislab

BLISlab: A Sandbox for Optimizing GEMM

cpu_gpu_profiling

CS378_PfCandP

CS378 Programming for Correctness and Performance

cutlass

CUDA Templates for Linear Algebra Subroutines

effectivepython

Effective Python: Second Edition — Source Code and Errata for the Book

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

flash-attention

Fast and memory-efficient exact attention

friendLunarBirthday

generate "csv" format data of my friends' Chinese lunar Birthday for Google Calendar import

glow

Compiler for Neural Network hardware accelerators

hub

me

about page for my personal website

neon

Nervana's python based Deep Learning Framework

param

PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for evaluation of training and inference platforms.

pytext

A natural language modeling framework based on PyTorch

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

sparse-ads-baselines

Strassen

sudonohup.github.com

tblis

TBLIS is a library and framework for performing tensor operations, especially tensor contraction, using efficient native algorithms.

tblis-strassen

torchrec-1

Pytorch domain library for recommendation systems

torchrec-3

Pytorch domain library for recommendation systems

TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.

tutorials

PyTorch tutorials.

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

vimbackup

vim plugin backup

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.