Alternatives and similar repositories for ATen

Users that are interested in ATen are comparing it to the libraries listed below

• pytorch / extension-cpp
  C++ extensions in PyTorch
  ☆1,092Updated 4 months ago
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,040Updated last year
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,306Updated this week
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,002Updated 2 weeks ago
• facebookresearch / TensorComprehensions
  A domain specific language to express machine learning workloads.
  ☆1,759Updated 2 years ago
• dmlc / mshadow
  Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning
  ☆1,111Updated 5 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆452Updated 5 years ago
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,301Updated last month
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,342Updated this week
• baidu-research / baidu-allreduce
  ☆585Updated 7 years ago
• openai / openai-gemm
  Open single and half precision gemm implementations
  ☆381Updated 2 years ago
• pytorch / cppdocs
  PyTorch C++ API Documentation
  ☆228Updated last week
• NVIDIA / cnmem
  A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory
  ☆298Updated 6 years ago
• tbennun / cudnn-training
  A CUDNN minimal deep learning training code sample using LeNet.
  ☆267Updated last year
• warmspringwinds / pytorch-cpp
  Pytorch C++ Library
  ☆367Updated 7 years ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆515Updated 6 years ago
• pytorch / builder
  Continuous builder and binary build scripts for pytorch
  ☆351Updated 3 weeks ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆714Updated 2 years ago
• pytorch / extension-ffi
  Examples of C extensions for PyTorch
  ☆255Updated 2 years ago
• dmlc / nnvm
  ☆1,657Updated 6 years ago
• dmlc / mxnet-memonger
  Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets
  ☆308Updated 7 years ago
• Maratyszcza / NNPACK
  Acceleration package for neural networks on multi-core CPUs
  ☆1,686Updated 11 months ago
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,747Updated last year
• pytorch / elastic
  PyTorch elastic training
  ☆729Updated 2 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 3 years ago
• d2l-ai / d2l-tvm
  Dive into Deep Learning Compiler
  ☆645Updated 2 years ago
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,803Updated last year
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆257Updated 2 years ago
• baidu-research / persistent-rnn
  Fast Recurrent Networks Library
  ☆576Updated 8 years ago
• dlsys-course / dlsys-course.github.io
  Deep learning system course
  ☆215Updated 6 years ago