yinghai Goto Github PK

abseil-cpp

ainotes

Some notes that I wrote while studying machine learning

aitemplateoss

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

caffe2

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

cutlass

CUDA Templates for Linear Algebra Subroutines

fbgemm

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

fbthrift

Facebook's branch of Apache Thrift, including a new C++ server.

folly

An open-source C++ library developed and used at Facebook.

foxi

ONNXIFI with Facebook Extension

functorch

functorch is a prototype of JAX-like composable function transforms for PyTorch.

glow

Compiler for Neural Network hardware accelerators

ideep

julia

The Julia Language: A fresh approach to technical computing.

mkl-dnn

Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)

onnx

Open Neural Network Exchange

onnx-fb-universe

ONNX Integration Builds

onnx-tensorrt

ONNX-TensorRT: TensorRT backend for ONNX

pytext

A natural language modeling framework based on PyTorch

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

tensorcomprehensions

A domain specific language to express machine learning workloads.

tensorflow

Computation using data flow graphs for scalable machine learning

tensorrt

TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.

torch-tensorrt

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

tutorials

Tutorials for using ONNX

tvm

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

tvm-1

TVM integration into PyTorch

xformers

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