Popular repositories Loading

1. torchscript-to-tvm torchscript-to-tvm Public

   Cuda 68 17

2. nnvm-vision-demo nnvm-vision-demo Public

   Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM

   Python 49 1

3. tvm-winograd tvm-winograd Public

   Test winograd convolution written in TVM for CUDA and AMDGPU

   Python 41 2

4. tvm-cutlass-eval tvm-cutlass-eval Public

   Python 40 7

5. libflash_attn libflash_attn Public

   C++ 14

6. mxnet-cpp-inference mxnet-cpp-inference Public

   Test MXNet C++ API for doing inference, given a trained model

   C++ 6 1