Popular GitHub Repositories for Tensor Computation

Tensors and Dynamic neural networks in Python with strong GPU acceleration

aka "Bayesian Methods for Hackers": An introduction to Bayesian methods + probabilistic programming with a computation/understanding-first, mathematics-second point of view. All in pure Python ;)

Tensor library for machine learning

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

? Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

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

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Gorgonia is a library that helps facilitate machine learning in Go.

Unified embedding generation and search engine. Also available on cloud - cloud.marqo.ai

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

Relax! Flux is the ML library that doesn't make you tensor

? ? ? ? GraphScope: A One-Stop Large-Scale Graph Computing System from Alibaba | 一站式图计算系统

Hummingbird compiles trained ML models into tensor computation for faster inference.

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

Learning in infinite dimension with neural operators.

C++ DataFrame for statistical, Financial, and ML analysis -- in modern C++ using native types and contiguous memory storage

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

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

A unified framework for privacy-preserving data analysis and machine learning