jspark1105

hopscotch

hopscotch concurrent hashing

block_sparse

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication)

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

tbb_test

benchmark

caffe

Caffe for Sparse Convolutional Neural Networks

caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.

cutlass

CUDA Templates for Linear Algebra Subroutines

d2go

D2Go is a toolkit for efficient deep learning

dlrm

An implementation of a deep learning recommendation model (DLRM)

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

FAI-PEP

Facebook AI Performance Evaluation Platform

fairscale

PyTorch extensions for high performance and large scale training.

fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

gapbs

GAP Benchmark Suite

gloo

Collective communications library with various primitives for multi-machine training.

glow

Compiler for Neural Network hardware accelerators

hpcg

Official HPCG benchmark source code

IterativeSolvers.jl

Implement Arnoldi and Lanczos methods for svds and eigs

libxsmm

Library targeting Intel Architecture (x86) for small, dense or sparse matrix multiplications, and small convolutions.

MatrixMarket.jl

Julia package to read MatrixMarket file format

mlir

"Multi-Level Intermediate Representation" Compiler Infrastructure

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.

pytorch-summary

Model summary in PyTorch similar to `model.summary()` in Keras

splatt

The Surprisingly ParalleL spArse Tensor Toolkit.

TASO

The Tensor Algebra SuperOptimizer for Deep Learning

torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

torchrec

Pytorch domain library for recommendation systems

tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation