ROCm / nnvm-rocm
NNVM for ROCm Examples
☆19Updated 7 years ago
Alternatives and similar repositories for nnvm-rocm:
Users that are interested in nnvm-rocm are comparing it to the libraries listed below
- Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM☆49Updated 7 years ago
- MXNet Model Serving☆25Updated 7 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 8 years ago
- Tools to convert Caffe models to neon's serialization format☆39Updated 2 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- PyTorch development for onnx☆21Updated 7 years ago
- This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.☆35Updated last year
- Python Binding to NVRTC☆79Updated 5 months ago
- TensorFlow util for building memory usage timeline from LOG_MEMORY messages☆64Updated 7 years ago
- ☆23Updated 8 years ago
- ☆57Updated 6 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- ☆16Updated 7 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- ☆18Updated 7 years ago
- SqueezeNet Generator☆31Updated 6 years ago
- ☆35Updated 7 years ago
- Torch FFI-bindings for NNPACK☆30Updated 7 years ago
- TFStage: TensorFlow Project Scaffolding☆62Updated 5 years ago
- ☆45Updated 2 years ago
- Bugfixing fork of Python bindings for the NVIDIA GPU Management Library☆51Updated 7 years ago
- train on AWS☆75Updated 6 years ago
- Efficient layer normalization GPU kernel for Tensorflow☆110Updated 7 years ago
- Fast binary matrix product on CPU☆10Updated 9 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- OpenCL Inference Engine for pytorch☆51Updated 7 years ago
- Deep exponential family models in MXNet/Gluon. Layers o' latents 💤☆17Updated 7 years ago