ROCm / nnvm-rocm
NNVM for ROCm Examples
☆19Updated 7 years ago
Related projects ⓘ
Alternatives and complementary repositories for nnvm-rocm
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆29Updated 8 years ago
- 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
- ☆16Updated 7 years ago
- Just-in-time Dynamic Batching with MXNet Gluon.☆52Updated 4 years ago
- MXNet - nGraph integration☆34Updated 2 years ago
- Tools to convert Caffe models to neon's serialization format☆39Updated last year
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- Training deep neural networks with low precision multiplications☆63Updated 9 years ago
- Torch FFI-bindings for NNPACK☆30Updated 7 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- Python Binding to NVRTC☆79Updated last month
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- The Operator Vectorization Library, or OVL, is a python productivity library for defining high performance custom operators for the Tenso…☆68Updated 7 years ago
- OpenCL Inference Engine for pytorch☆51Updated 6 years ago
- PyTorch development for onnx☆21Updated 6 years ago
- Boda: A C++ Framework for Efficient Experiments in Computer Vision☆63Updated 5 years ago
- SqueezeNet Generator☆32Updated 6 years ago
- Fast binary matrix product on CPU☆10Updated 8 years ago
- Generating Families of Practical Fast Matrix Multiplication Algorithms☆12Updated 7 years ago
- ONNX Integration Builds☆20Updated 6 years ago
- Flattened convolutional neural networks (1D convolution modules for Torch nn)☆61Updated 9 years ago
- Deep exponential family models in MXNet/Gluon. Layers o' latents 💤☆17Updated 7 years ago
- Direct C++ Interface to PyTorch☆80Updated 6 years ago
- ☆47Updated 4 years ago
- Training neural networks with 8-bit computations☆29Updated 8 years ago
- ☆58Updated 6 years ago
- An Example of MXNet Models Comilation and Deployment with NNVM in C++☆15Updated 6 years ago
- ☆23Updated 8 years ago