bcaine / nn_cpp
A minimalistic header only C++11 Neural Network library based on Eigen::Tensor
☆20Updated 6 years ago
Related projects ⓘ
Alternatives and complementary repositories for nn_cpp
- flexible-gemm conv of deepcore☆17Updated 4 years ago
- Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions☆21Updated 2 years ago
- Scalable GPU Kernel Fission/Fusion Transformation for Memory-Bound Kernels☆13Updated 9 years ago
- Evaluating different memory managers for dynamic GPU memory☆24Updated 3 years ago
- how to design cpu gemm on x86 with avx256, that can beat openblas.☆66Updated 5 years ago
- RDMA Optimization on MXNet☆14Updated 7 years ago
- GPU Optimization and Memory Abstraction Framework☆32Updated 5 years ago
- This is a tuned sparse matrix dense vector multiplication(SpMV) library☆21Updated 8 years ago
- Emulating DMA Engines on GPUs for Performance and Portability☆34Updated 9 years ago
- A tool for examining GPU scheduling behavior.☆70Updated 3 months ago
- An external memory allocator example for PyTorch.☆13Updated 3 years ago
- Repository holding the code base to AC-SpGEMM : "Adaptive Sparse Matrix-Matrix Multiplication on the GPU"☆28Updated 4 years ago
- Yet another Polyhedra Compiler for DeepLearning☆19Updated last year
- Code for paper "Engineering a High-Performance GPU B-Tree" accepted to PPoPP 2019☆52Updated 2 years ago
- ICML2017 MEC: Memory-efficient Convolution for Deep Neural Network C++实现(非官方)☆17Updated 5 years ago
- study of cutlass☆19Updated 2 weeks ago
- A Distributed Multi-GPU System for Fast Graph Processing☆63Updated 6 years ago
- Sparse-dense matrix-matrix multiplication on GPUs☆14Updated 6 years ago
- ☆31Updated 4 years ago
- Benchmark for matrix multiplications between dense and block sparse (BSR) matrix in TVM, blocksparse (Gray et al.) and cuSparse.☆24Updated 4 years ago
- Artifact of ASPLOS'23 paper entitled: GRACE: A Scalable Graph-Based Approach to Accelerating Recommendation Model Inference☆16Updated last year
- Kernel Fusion and Runtime Compilation Based on NNVM☆69Updated 8 years ago
- play gemm with tvm☆84Updated last year
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆19Updated 9 months ago
- a heterogeneous multiGPU level-3 BLAS library☆45Updated 4 years ago
- An experimental ahead of time compiler for Relay.☆51Updated 4 years ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago
- Learning and practice of high performance computing (CUDA, Vulkan, OpenCL, OpenMP, TBB, SSE/AVX, NEON, MPI, coroutines, etc. )☆57Updated this week
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- An MLIR frontend for tensor expressions☆24Updated 4 years ago