PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

An open-source efficient deep learning framework.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Development repository for the Triton language and compiler

C++ IPC Library: A high-performance inter-process communication using shared memory on Linux/Windows.

A list of awesome compiler projects and papers for tensor computation and deep learning.

Eclipse iceoryx™ - true zero-copy inter-process-communication

Guide: Finetune GPT2-XL (1.5 Billion Parameters) and finetune GPT-NEO (2.7 B) on a single GPU with Huggingface Transformers using DeepSpeed

cricket is a virtualization solution for GPUs

A library that provides an embeddable, persistent key-value store for fast storage.

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…

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

CVNets: A library for training computer vision networks

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

NVIDIA Linux open GPU kernel module source

Automatic virtualization of (general) accelerators.

Lucid: A Non-Intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs

cluster data collected from production clusters in Alibaba for cluster management research