sunway513

awesome-cpp

A curated list of awesome C++ (or C) frameworks, libraries, resources, and shiny things. Inspired by awesome-... stuff.

bert

TensorFlow code and pre-trained models for BERT

benchmarks

A benchmark framework for Tensorflow

climate-seg-benchmark

Reference implementation for the climate segmentation benchmark, based on the Exascale Deep Learning for Climate Analytics work

cookbook

Deep learning for dummies. All the practical details and useful utilities that go into working with real models.

DeepLearningExamples

Deep Learning Examples

deploy-code-server

detectron2

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

dlrm

An implementation of a deep learning recommendation model (DLRM)

gloo

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

HIP

HIP : Convert CUDA to Portable C++ Code

Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Megatron-LM

Ongoing research training transformer language models at scale, including: BERT & GPT-2

misc

nccl-rccl-parser

Tool to run rccl-tests/nccl-tests based on from an application

nmt

TensorFlow Neural Machine Translation Tutorial

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

rocBLAS

Next generation BLAS implementation for ROCm platform

ROCm

ROCm - Open Source Platform for HPC and Ultrascale GPU Computing

rocm-recipes

Recipes for rocm

ROCm.github.io

ROCm Website

ROCm_Documentation

ROCm Software Platform Documentation

ROCmValidationSuite

The ROCm Validation Suite is a system administrator’s and cluster manager's tool for detecting and troubleshooting common problems affecting AMD GPU(s) running in a high-performance computing environment, enabled using the ROCm software stack on a compatible platform.

Tensile

Stretching GPU performance for GEMMs and tensor contractions.

tensorflow-large-model-support

Large Model Support in Tensorflow

tensorflow-upstream

TensorFlow ROCm port

transformers

🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.

vision

Datasets, Transforms and Models specific to Computer Vision

YOLOv3_TensorFlow

Complete YOLO v3 TensorFlow implementation. Support training on your own dataset.

yolov5

YOLOv5 in PyTorch > ONNX > CoreML > TFLite