Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

Tensor library for machine learning

📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.🎉

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

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

The AI-native database built for LLM applications, providing incredibly fast hybrid search of dense vector, sparse vector, tensor (multi-vector), and full-text.

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models

C++ tensors with broadcasting and lazy computing

Simple, safe way to store and distribute tensors

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

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

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

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

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

Pixel IMS: Rootless replacement for Tensor Pixel VoLTE patch

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

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

A library for easy and efficient manipulation of tensor networks.

ML models and internal tensors 3D visualizer

TensorLy: Tensor Learning in Python.

High Performance Computation for N-D Tensors in .NET, similar API to NumPy.

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

creating a tiny tensor library in raw C

Tensors, for human consumption

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

A fast communication-overlapping library for tensor/expert parallelism on GPUs.

A minimal tensor processing unit (TPU), inspired by Google's TPU V2 and V1

Pretty Tensor: Fluent Networks in TensorFlow

Tensors and neural networks in Haskell