Alternatives and similar repositories for ATen:

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

• dmlc / dlpack
  common in-memory tensor structure
  ☆983Updated 3 weeks ago
• pytorch / gloo
  Collective communications library with various primitives for multi-machine training.
  ☆1,294Updated this week
• pytorch / extension-cpp
  C++ extensions in PyTorch
  ☆1,083Updated 3 months ago
• openai / blocksparse
  Efficient GPU kernels for block-sparse matrix multiplication and convolution
  ☆1,039Updated last year
• facebookresearch / TensorComprehensions
  A domain specific language to express machine learning workloads.
  ☆1,759Updated 2 years ago
• pytorch / tvm
  TVM integration into PyTorch
  ☆452Updated 5 years ago
• NVIDIA / cnmem
  A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory
  ☆297Updated 6 years ago
• pytorch / cppdocs
  PyTorch C++ API Documentation
  ☆224Updated this week
• tensor-compiler / taco
  The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
  ☆1,293Updated 3 weeks ago
• pytorch / extension-ffi
  Examples of C extensions for PyTorch
  ☆255Updated 2 years ago
• tbennun / cudnn-training
  A CUDNN minimal deep learning training code sample using LeNet.
  ☆266Updated last year
• NVIDIA / cub
  [ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
  ☆1,745Updated last year
• Maratyszcza / NNPACK
  Acceleration package for neural networks on multi-core CPUs
  ☆1,687Updated 10 months ago
• openai / openai-gemm
  Open single and half precision gemm implementations
  ☆381Updated 2 years ago
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,313Updated this week
• warmspringwinds / pytorch-cpp
  Pytorch C++ Library
  ☆367Updated 6 years ago
• dmlc / mshadow
  Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning
  ☆1,110Updated 5 years ago
• google / gemmlowp
  Low-precision matrix multiplication
  ☆1,800Updated last year
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆511Updated 6 years ago
• pytorch / builder
  Continuous builder and binary build scripts for pytorch
  ☆352Updated last month
• pytorch / QNNPACK
  Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators
  ☆1,540Updated 5 years ago
• szagoruyko / pyinn
  CuPy fused PyTorch neural networks ops
  ☆273Updated 7 years ago
• inducer / loopy
  A code generator for array-based code on CPUs and GPUs
  ☆602Updated last week
• intel / ideep
  Intel® Optimization for Chainer*, a Chainer module providing numpy like API and DNN acceleration using MKL-DNN.
  ☆170Updated last week
• NVIDIA-developer-blog / code-samples
  Source code examples from the Parallel Forall Blog
  ☆1,283Updated 9 months ago
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆710Updated 2 years ago
• dmlc / nnvm
  ☆1,658Updated 6 years ago
• libxsmm / libxsmm
  Library for specialized dense and sparse matrix operations, and deep learning primitives.
  ☆869Updated last week
• NVIDIA / nvvl
  A library that uses hardware acceleration to load sequences of video frames to facilitate machine learning training
  ☆685Updated 6 years ago
• baidu-research / baidu-allreduce
  ☆582Updated 7 years ago