Popular GitHub Repositories for Tensor

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

On-device AI across mobile, embedded and edge for PyTorch

A list of awesome compiler projects and papers for tensor computation and deep learning.

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

Inference Llama 2 in one file of pure ?

Library for specialized dense and sparse matrix operations, and deep learning primitives.

AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。

Deep learning with spiking neural networks (SNNs) in PyTorch.

DiffSharp: Differentiable Functional Programming

Tensors and differentiable operations (like TensorFlow) in Rust

A Machine Learning framework from scratch in Pure Mojo ?

OCaml bindings for PyTorch

NiuTensor is an open-source toolkit developed by a joint team from NLP Lab. at Northeastern University and the NiuTrans Team. It provides tensor utilities to create and train neural networks.

a c++/cuda template library for tensor lazy evaluation

My personal reference for Tensorflow

A language and compiler for irregular tensor programs.