Tensor Open Source Projects

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Open Machine Learning Compiler Framework

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

NumPy & SciPy for GPU

CUDA Templates and Python DSLs for High-Performance Linear Algebra

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

http://torch.ch

TensorFlow tutorials and best practices.

Rust bindings for the C++ api of PyTorch.

Simple examples to introduce PyTorch

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

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

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

A comprehensive list of Deep Learning / Artificial Intelligence and Machine Learning tutorials - rapidly expanding into areas of AI/Deep Learning / Machine Vision / NLP and industry specific areas such as Climate / Energy, Automotives, Retail, Pharma, Medicine, Healthcare, Policy, Ethics and more.

TensorFlow for macOS 11.0+ accelerated using Apple's ML Compute framework.

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

LibRec: A Leading Java Library for Recommender Systems, see

TensorFlow Reinforcement Learning

C++ DataFrame for statistical, financial, and ML analysis in modern C++

Data manipulation and transformation for audio signal processing, powered by PyTorch

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

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

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.

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

Basic Utilities for PyTorch Natural Language Processing (NLP)

Inference Llama 2 in one file of pure 🔥

Drawing Bayesian networks, graphical models, tensors, technical frameworks, and illustrations in LaTeX.

TinyML AI inference library

Deep learning in Rust, with shape checked tensors and neural networks

A library for easy and efficient manipulation of tensor networks.

A domain specific language to express machine learning workloads.

PyTorch Extension Library of Optimized Scatter Operations

PyTorch tutorials and best practices.

TensorLy: Tensor Learning in Python.

Mesh TensorFlow: Model Parallelism Made Easier

Haskell bindings for TensorFlow

This project uses reinforcement learning on stock market and agent tries to learn trading. The goal is to check if the agent can learn to read tape. The project is dedicated to hero in life great Jesse Livermore.

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

A fast, ergonomic and portable tensor library in Nim with a deep learning focus for CPU, GPU and embedded devices via OpenMP, Cuda and OpenCL backends

Deep Learning Zero to All - Pytorch

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs

Machine learning for transportation data imputation and prediction.

A modular framework for neural networks with Euclidean symmetry

common in-memory tensor structure

Pretty Tensor: Fluent Networks in TensorFlow

Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

A library for doing homomorphic encryption operations on tensors

move forward to https://github.com/dmlc/mxnet

Framework-agnostic implementation for state-of-the-art saliency methods (XRAI, BlurIG, SmoothGrad, and more).

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

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

The Egison Programming Language

Low-Rank and Sparse Tools for Background Modeling and Subtraction in Videos

Robust evasion attacks against neural network to find adversarial examples

A lightweight high performance tensor algebra framework for modern C++

Deep learning for Ruby, powered by LibTorch

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

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

Debug PyTorch code using PySnooper