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
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆28Updated 9 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- SqueezeNet Generator☆31Updated 7 years ago
- Move to https://github.com/apache/incubator-tvm-site☆26Updated 4 years ago
- ☆16Updated 7 years ago
- ☆23Updated 8 years ago
- ☆18Updated 7 years ago
- Simple, efficient and flexible vision toolbox for mxnet framework.☆31Updated 7 years ago
- ☆47Updated 5 years ago
- Distributed Learning by Pair-Wise Averaging☆52Updated 7 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆41Updated 6 years ago
- Torch FFI-bindings for NNPACK☆30Updated 7 years ago
- Tools to convert Caffe models to neon's serialization format☆39Updated 2 years ago
- Direct C++ Interface to PyTorch☆80Updated 7 years ago
- Tensorflow implementation of SGD with Coupled Adaptive Batch Size (CABS)☆43Updated 8 years ago
- Python Binding to NVRTC☆79Updated 7 months ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- ShaResNet: reducing residual network parameter number by sharing weights☆32Updated 8 years ago
- Training neural networks with 8-bit computations☆28Updated 9 years ago
- Evaluating efficiency of several types of convolutions☆30Updated 7 years ago
- Fast binary matrix product on CPU☆10Updated 9 years ago
- Deep exponential family models in MXNet/Gluon. Layers o' latents 💤☆17Updated 7 years ago
- auto-tuning momentum SGD optimizer☆23Updated 7 years ago
- The Operator Vectorization Library, or OVL, is a python productivity library for defining high performance custom operators for the Tenso…☆68Updated 8 years ago
- Deep learning study in Gluon 2nd edition☆24Updated 6 years ago
- OpenCL Inference Engine for pytorch☆51Updated 7 years ago
- train on AWS☆75Updated 6 years ago
- ☆57Updated 7 years ago