The HPC toolbox: fused matrix multiplication, convolution, data-parallel strided tensor primitives, OpenMP facilities, SIMD, JIT Assembler, CPU detection, state-of-the-art vectorized BLAS for floats and integers

✭ 191

nim assembler deep-learning parallel simd jit high-performance-computing tensor convolution openmp blas

Mars

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

✭ 2,308

python javascript cython C++shell c machine-learning pytorch tensorflow numpy pandas scikit-learn xgboost tensor dataframe lightgbm dask ray statsmodels joblib

Tinytpu

Implementation of a Tensor Processing Unit for embedded systems and the IoT.

✭ 153

assembly linux tensorflow iot verilog fpga internet-of-things vhdl embedded-systems tensor

Tensorflow Cheatsheet

My personal reference for Tensorflow

✭ 147

tensorflow tutorial neural-network cheatsheet tensor

Hptt

High-Performance Tensor Transpose library

✭ 141

high-performance-computing tensor tensors multidimensional-arrays

l2 is a fast, Pytorch-style Tensor+Autograd library written in Rust

✭ 126

rust machine-learning ai tensor

Mtensor

A C++ Cuda Tensor Lazy Computing Library

✭ 115

cuda numpy tensor eigen

Pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

✭ 52,811

python C++Cuda c CMake Objective-C++deep-learning machine-learning neural-network gpu numpy tensor autograd

Tensorflow Gpu Macosx

Unoffcial NVIDIA CUDA GPU support version of Google Tensorflow for MAC OSX

✭ 103

python gpu osx mac tensor

Deepnet

Deep.Net machine learning framework for F#

✭ 99

fsharp deep-learning machine-learning dotnet gpu cuda tensor gpu-computing gpu-acceleration

Pytorch Wrapper

Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.

✭ 92

python deep-learning machine-learning pytorch data-science neural-network tensor

Pytorch2c

A Python module for compiling PyTorch graphs to C

✭ 86

python deep-learning pytorch graph tensor

Hyperlearn

50% faster, 50% less RAM Machine Learning. Numba rewritten Sklearn. SVD, NNMF, PCA, LinearReg, RidgeReg, Randomized, Truncated SVD/PCA, CSR Matrices all 50+% faster

✭ 1,204

python jupyter-notebook deep-learning machine-learning pytorch data-science neural-network gpu statistics optimization data-analysis scikit-learn research tensor

Pytorch Book

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

✭ 9,546

Jupyter Notebook python shell jupyter-notebook deep-learning pytorch gan image-classification tensor tensorboard neural-style pytorch-tutorials nn autograd caption visdom pytorch-tutorials-cn charrnn neuraltalk

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

✭ 1,133

elixir pytorch gpu jit tensor

Mtt

MATLAB Tensor Tools

✭ 61

matlab tensor

Cloud Volume

Read and write Neuroglancer datasets programmatically.

✭ 63

python serverless cloud numpy big-data s3 matrix mesh tensor

Numeric

N-dimensional matrix class for Rust

✭ 51

rust matrix linear-algebra tensor

Qtorch

qTorch (Quantum Tensor Contraction Handler) https://arxiv.org/abs/1709.03636 -> for quantum simulation using tensor networks

✭ 43

c simulation tensor quantum

Tensorly

TensorLy: Tensor Learning in Python.

✭ 977

python machine-learning pytorch tensorflow numpy mxnet tensor cupy

Aiscm

Guile numerical arrays and tensor extension

✭ 34

scheme debian tensor

Drlkit

A High Level Python Deep Reinforcement Learning library. Great for beginners, prototyping and quickly comparing algorithms

✭ 29

python deep-learning machine-learning pytorch tensorflow reinforcement-learning gpu numpy deep-reinforcement-learning tensor gym

Imagenet Classifier Tensorflow

Image recognition and classification using Convolutional Neural Networks with TensorFlow

✭ 13

python tensorflow cuda tensor cudnn

Taco

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

✭ 846

library code-generator linear-algebra tensor

Egison

The Egison Programming Language

✭ 800

haskell hacktoberfest functional-programming tensor pattern-matching

Tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

✭ 7,494

python C++rust c CMake shell deep-learning machine-learning performance compiler gpu vulkan opencl metal tensor spirv rocm tvm

Arraymancer

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

✭ 793

nim deep-learning machine-learning iot neural-networks cuda opencl linear-algebra parallel-computing gpgpu high-performance-computing tensor gpu-computing automatic-differentiation openmp cudnn multidimensional-arrays

Lrslibrary

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

✭ 625

matlab matrix tensor matrix-factorization

Aten

ATen: A TENsor library for C++11

✭ 578

pytorch tensor torch7

Cupy

NumPy & SciPy for GPU

✭ 5,625

python cython C++c Cuda Dockerfile gpu cuda numpy tensor scipy cupy cudnn cublas rocm cusolver nccl curand cusparse nvrtc cutensor nvtx cusparselt

Libxsmm

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

✭ 518

c fortran machine-learning matrix vector simd jit intel sse tensor avx2 convolution avx512 avx blas

Awesome Tensor Compilers

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

✭ 490

deep-learning machine-learning compiler code-generation high-performance-computing tensor

Autokernel

AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。

✭ 485

deep-learning pytorch tensorflow reinforcement-learning optimization tensor auto

Opt einsum

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

✭ 397

python performance tensor gpu-acceleration

Dlpack

RFC for common in-memory tensor structure and operator interface for deep learning system

✭ 398

deep-learning operator tensor

Diffsharp

DiffSharp: Differentiable Functional Programming

✭ 365

deep-learning machine-learning dotnet neural-network gpu tensor

Niutensor

NiuTensor is an open-source toolkit developed by a joint team from NLP Lab. at Northeastern University and the NiuTrans Team. It provides tensor utilities to create and train neural networks.

✭ 337

deep-learning machine-learning artificial-neural-networks tensor

Ocaml Torch

OCaml bindings for PyTorch

✭ 308

ocaml deep-learning machine-learning pytorch neural-network artificial-intelligence gpu tensor

Grassmann.jl

⟨Leibniz-Grassmann-Clifford⟩ differential geometric algebra / multivector simplicial complex

✭ 289

julia math linear-algebra quantum-computing tensor computational-geometry algebra automatic-differentiation computer-algebra topology complex-numbers

Thinking In Tensors Writing In Pytorch

Thinking in tensors, writing in PyTorch (a hands-on deep learning intro)