shigangli Goto Github PK

akg

AKG (Auto Kernel Generator) is an optimizer for operators in Deep Learning Networks, which provides the ability to automatically fuse ops with specific patterns.

atari

AI research environment for the Atari 2600 games 🤖.

attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".

baselines

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

benchmark_spmv_using_csr5

CSR5-based SpMV on CPUs, GPUs and Xeon Phi

bert

TensorFlow code and pre-trained models for BERT

bert-pytorch

Google AI 2018 BERT pytorch implementation

bigbird

Transformers for Longer Sequences

bigbird-1

Google's BigBird (Jax/Flax & PyTorch) @ 🤗Transformers

bolt

10x faster matrix and vector operations.

brian2

Brian is a free, open source simulator for spiking neural networks.

caffe

Caffe: a fast open framework for deep learning.

chart-doctor

Sample files to accompany the FT's Chart Doctor column

chimera

Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.

clover

Clover: Quantized 4-bit Linear Algebra Library

cnnarm

code-samples

Source code examples from the Parallel Forall Blog

colossalai

Making large AI models cheaper, faster and more accessible

compi

Cache-oblivious MPI all-to-all communications based on Morton order

computelibrary

The Compute Library is a set of computer vision and machine learning functions optimised for both Arm CPUs and GPUs using SIMD technologies.

convnet.pytorch

ConvNet training using pytorch

cosma

Distributed Communication-Optimal Matrix-Matrix Multiplication Algorithm

ctrl

Conditional Transformer Language Model for Controllable Generation

cuassembler

An unofficial cuda assembler, for all generations of SASS, hopefully ：）

cuda-samples

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

cudnn-training

A CUDNN minimal deep learning training code sample using LeNet.

custom-op

dace

DaCe - Data Centric Parallel Programming

daceml

Data-centric machine learning powered by DaCe

dapple

An Efficient Pipelined Data Parallel Approach for Training Large Model