• 2
• 0
• 0

tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

• 2
• 0
• 0

pytorch-micro-benchmarking

• 2
• 0
• 0

DeepSpeedExamples

Example models using DeepSpeed

• 2
• 0
• 0

pyro

Deep universal probabilistic programming with Python and PyTorch

• 2
• 0
• 0

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

• 2
• 0
• 0

onnxruntime

ONNX Runtime: cross-platform, high performance scoring engine for ML models

• 2
• 0
• 0

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

• 2
• 0
• 0

iGEMMgen

Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)

• 2
• 0
• 0

dlrm

An implementation of a deep learning recommendation model (DLRM)

• 2
• 0
• 0

roctracer

ROCm Tracer Callback/Activity Library for Performance tracing AMD GPU's

• 2
• 0
• 0

models

Models and examples built with TensorFlow

• 2
• 0
• 0

rccl-tests

RCCL Performance Benchmark Tests

• 2
• 0
• 0

examples

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

• 2
• 0
• 0

rocSPARSE

Next generation SPARSE implementation for ROCm platform

• 2
• 0
• 0

rocSOLVER

Next generation LAPACK implementation for ROCm platform

• 2
• 0
• 0

horovod

Distributed training framework for TensorFlow, Keras, and PyTorch.

• 2
• 0
• 0

rccl

ROCm Communication Collectives Library (RCCL)

• 2
• 0
• 0

rocFFT

Next generation FFT implementation for ROCm

• 2
• 0
• 0

rocALUTION

Next generation library for iterative sparse solvers for ROCm platform

• 2
• 0
• 0

rocBLAS

Next generation BLAS implementation for ROCm platform