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

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

You like pytorch? You like micrograd? You love tinygrad! ❤️

The V Tensor Library

Tiny Differentiable Simulator is a header-only C++ and CUDA physics library for reinforcement learning and robotics with zero dependencies.

A Julia library for efficient tensor computations and tensor network calculations

OpenBLAS is an optimized BLAS library based on GotoBLAS2 1.13 BSD version.

BLAS-like Library Instantiation Software Framework

Google Research

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

Basic Tensor Algebra Subroutines

TBLIS is a library and framework for performing tensor operations, especially tensor contraction, using efficient native algorithms.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.

CUDA Templates for Linear Algebra Subroutines

Bayesian Modeling and Probabilistic Programming in Python

Development repository for the Triton language and compiler

An easily integrable Cholesky solver on CPU and GPU