Tensor

PyTorch - Tensors and Dynamic neural networks in Python with strong GPU accelerationPython

A Python package for tensor computation with GPU acceleration and dynamic neural networks built on a tape-based autograd system.

burnRust

A Rust-based deep learning framework and tensor library optimized for flexibility, efficiency, and cross-platform portability.

einopsPython

A Python library for flexible and readable tensor operations across numpy, PyTorch, JAX, TensorFlow, and other frameworks.

NxElixir

Multi-dimensional arrays (tensors) and numerical definitions for Elixir, enabling machine learning and scientific computing.

MarsPython

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

dfdxRust

A deep learning library in Rust featuring shape-checked tensors and neural networks with compile-time safety.

dfdxRust

A deep learning library for Rust featuring shape-checked tensors and neural networks with compile-time safety.

TensorLyPython

TensorLy: Tensor Learning in Python.

DiffSharpF#

A tensor library for differentiable functional programming in F#, with PyTorch-like APIs and GPU support.

PyTorch bindingsOCaml

OCaml bindings for PyTorch, providing NumPy-like tensor computations with GPU acceleration and automatic differentiation.

UpsurgeSwift

A Swift library providing numpy-like multi-dimensional data structures and operations for numerical computing.

num.crCrystal

A core scientific computing library for Crystal providing n-dimensional tensors, linear algebra, GPU acceleration, and automatic differentiation.

DL4SSwift

A Swift library for accelerated tensor operations and dynamic neural networks with automatic differentiation, supporting all Apple platforms and Linux.

RustTensorRust

A learning-focused, high-performance tensor computation library built from scratch in Rust with automatic differentiation and CPU/CUDA backends.

NDScalaScala

Type-safe n-dimensional arrays (tensors) in Scala 3 with compile-time shape, axis label, and data type validation.

CatniffTypeScript

A Torch-like deep learning framework for JavaScript with direct tensor and autograd operations.

KandleTypeScript

A JavaScript-native machine learning framework with PyTorch-aligned APIs, built from scratch on WebGPU for dynamic graph execution and model interpretability.