tensor
There are 632 repositories under tensor topic.
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
chenyuntc/pytorch-book
PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
apache/tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
cupy/cupy
NumPy & SciPy for GPU
tracel-ai/burn
Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
MegEngine/MegEngine
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
guoguibing/librec
LibRec: A Leading Java Library for Recommender Systems, see
mars-project/mars
Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
elixir-nx/nx
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
hosseinmoein/DataFrame
C++ DataFrame for statistical, Financial, and ML analysis -- in modern C++ using native types and contiguous memory storage
merrymercy/awesome-tensor-compilers
A list of awesome compiler projects and papers for tensor computation and deep learning.
pytorch/executorch
On-device AI across mobile, embedded and edge for PyTorch
tairov/llama2.mojo
Inference Llama 2 in one file of pure 🔥
unslothai/hyperlearn
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
coreylowman/dfdx
Deep learning in Rust, with shape checked tensors and neural networks
tensorly/tensorly
TensorLy: Tensor Learning in Python.
mratsim/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
tensor-compiler/taco
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
egison/egison
The Egison Programming Language
dmlc/dlpack
common in-memory tensor structure
dgasmith/opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
libxsmm/libxsmm
Library for specialized dense and sparse matrix operations, and deep learning primitives.
OpenMined/TenSEAL
A library for doing homomorphic encryption operations on tensors
andrewssobral/lrslibrary
Low-Rank and Sparse Tools for Background Modeling and Subtraction in Videos
OAID/AutoKernel
AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。
zdevito/ATen
ATen: A TENsor library for C++11
norse/norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
xwying/torchshow
Visualize PyTorch tensors with a single line of code.
DiffSharp/DiffSharp
DiffSharp: Differentiable Functional Programming
raskr/rust-autograd
Tensors and differentiable operations (like TensorFlow) in Rust
Jutho/TensorOperations.jl
Julia package for tensor contractions and related operations
chakravala/Grassmann.jl
⟨Grassmann-Clifford-Hodge⟩ multilinear differential geometric algebra
basalt-org/basalt
A Machine Learning framework from scratch in Pure Mojo 🔥
LaurentMazare/ocaml-torch
OCaml bindings for PyTorch