tensor

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.

Open Machine Learning Compiler Framework

NumPy & SciPy for GPU

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

The AI search platform

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

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

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

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.

Inference Llama 2 in one file of pure 🔥

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

SIMD-accelerated distances, dot products, matrix ops, geospatial & geometric kernels for 16 numeric types — from 6-bit floats to 64-bit complex — across x86, Arm, RISC-V, and WASM, with bindings for Python, Rust, C, C++, Swift, JS, and Go 📐

TensorLy: Tensor Learning in Python.

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

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

common in-memory tensor structure

Introduction to PyTorch, covering tensor initialization, operations, indexing, and reshaping.

A library for doing homomorphic encryption operations on tensors

⚡️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

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

Visualize PyTorch tensors with a single line of code.

⅀

DiffSharp: Differentiable Functional Programming