spcl / ucudnn
Accelerating DNN Convolutional Layers with Micro-batches
☆63Updated 5 years ago
Alternatives and similar repositories for ucudnn:
Users that are interested in ucudnn are comparing it to the libraries listed below
- A prototype implementation of AllReduce collective communication routine.☆19Updated 6 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆41Updated 6 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Experimental toolchain to compile and run Chainer models☆112Updated 4 years ago
- DNN Inference with CPU, C++, ONNX support: Instant☆56Updated 6 years ago
- Greentea LibDNN - a universal convolution implementation supporting CUDA and OpenCL☆135Updated 8 years ago
- Library for fast image convolution in neural networks on Intel Architecture☆29Updated 7 years ago
- Symbolic Expression and Statement Module for new DSLs☆205Updated 4 years ago
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- Python bindings for NVTX☆66Updated last year
- Add-on package for ONNX format support in Chainer☆85Updated 5 years ago
- TVM stack: exploring the incredible explosion of deep-learning frameworks and how to bring them together☆64Updated 6 years ago
- Intel® Optimization for Chainer*, a Chainer module providing numpy like API and DNN acceleration using MKL-DNN.☆170Updated last week
- nGraph™ Backend for ONNX☆42Updated 2 years ago
- High Efficiency Convolution Kernel for Maxwell GPU Architecture☆134Updated 7 years ago
- CNNs in Halide☆23Updated 9 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- ☆71Updated 6 months ago
- Sparse matrix computation library for GPU☆56Updated 4 years ago
- Chainer x TensorRT☆34Updated 6 years ago
- This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.☆35Updated last year
- portDNN is a library implementing neural network algorithms written using SYCL☆113Updated 11 months ago
- A CUDA Extension of Neural Network Libraries☆92Updated 10 months ago
- npcomp - An aspirational MLIR based numpy compiler☆51Updated 4 years ago
- CUDA Tensor Transpose (cuTT) library☆51Updated 7 years ago
- Tests and benchmarks for cudnn (and in the future, other nvidia libraries)☆53Updated 4 years ago
- Efficient Sparse-Winograd Convolutional Neural Networks (ICLR 2018)☆191Updated 5 years ago
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 7 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Efficient forward propagation for BCNNs☆50Updated 7 years ago