Arsmart123

bayesian-tensor-rank-determination

Benchmark_SpGEMM_using_CSR

CSR-based SpGEMM on nVidia and AMD GPUs

CUDA_gemm

A simple high performance CUDA GEMM implementation.

MegEngine

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch

BERT-pytorch

Google AI 2018 BERT pytorch implementation

Compute-continuous-moments-de-ned-in-a-rectangular-region-using-CUDA-and-some-applications

YHs_Sample

Yinghan's Code Sample

IT5007_Project_Spark-Tok

This is the repository containing souce code of our IT5007 Project - Spark Tok

maxas

Assembler for NVIDIA Maxwell architecture

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

ParsiMoNe

Parallel Construction of Module Networks

acrotensor

A C++ library for computing large scale tensor contractions.

ttpy

Python implementation of the TT-Toolbox

how-to-optimize-gemm

row-major matmul optimization

code-samples

Source code examples from the Parallel Forall Blog

FasterTransformer

Transformer related optimization, including BERT, GPT

cutlass

CUDA Templates for Linear Algebra Subroutines

MSplitGEMM

Large matrix multiplication in CUDA

SGEMM-Implementation-and-Optimization

:pencil: Some source code about matrix multiplication implementation on CUDA

matrix-cuda

matrix multiplication in CUDA

optimizing-matrix-multiplication-examples

Here's optimizing matrix multiplication examples.

NVIDIA_SGEMM_PRACTICE

Step-by-step optimization of CUDA SGEMM

CUDA-Programming-with-Python

关于书籍CUDA Programming使用了pycuda模块的Python版本的示例代码

CUDA-Programming

Sample codes for my CUDA programming book

extension-cpp

C++ extensions in PyTorch

pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

cuda_accelerate

使用c++以及cuda加速神经网络样例（实现矩阵加法和矩阵乘法）

CUDA_Freshman

tensorly-notebooks

Tensor methods in Python with TensorLy