Pinned

1. CUDA_gemm CUDA_gemm Public

   A simple high performance CUDA GEMM implementation.

   Cuda 292 34

2. openxla/xla openxla/xla Public

   A machine learning compiler for GPUs, CPUs, and ML accelerators

   C++ 2.3k 342

3. KgeN KgeN Public

   A TVM-like CUDA/C code generator.

   Python 8

4. microsoft/nni microsoft/nni Public

   An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

   Python 13.8k 1.8k

5. microsoft/nnfusion microsoft/nnfusion Public

   A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.

   C++ 932 155

6. Pyflow Pyflow Public

   A toy Pytorch implementation.

   Python 2