tensor

There are 26 repositories under tensor topic.

pytorch / pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
neural-network autograd gpu numpy deep-learning tensor python machine-learning
Language:Python 77845
chenyuntc / pytorch-book
PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
pytorch pytorch-tutorials pytorch-tutorials-cn deep-learning neural-style charrnn gan caption neuraltalk image-classification visdom tensorboard nn tensor autograd jupyter-notebook
Language:Jupyter Notebook 11722
apache / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
compiler deep-learning gpu javascript machine-learning metal opencl performance rocm spirv tensor tvm vulkan
Language:Python 11167
einops
arogozhnikov / einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
chainer cupy deep-learning einops jax keras numpy pytorch tensor tensorflow
Language:Python 7912
cupy / cupy
NumPy & SciPy for GPU
cuda cudnn cublas cusolver nccl python numpy cupy curand cusparse gpu cutensor scipy nvtx nvrtc tensor cusparselt rocm
Language:Python 7774
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.
autodiff autotune concurrency cross-platform deep-learning high-performance kernel-fusion llm machine-learning ndarray neural-network onnx pytorch rust scientific-computing tensor wasm webgpu
Language:Rust 7024
MegEngine / MegEngine
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
autograd deep-learning gpu machine-learning megengine numpy python tensor
Language:C++ 4717
guoguibing / librec
LibRec: A Leading Java Library for Recommender Systems, see
recommender-systems recommendation-algorithms collaborative-filtering matrix-factorization tensor-factorization probabilistic-graphical-models recommender systems factorization matrix tensor collaborative filtering sparse java
Language:Java 3220
mars-project / mars
Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
python numpy tensor pandas machine-learning scikit-learn tensorflow pytorch xgboost lightgbm ray statsmodels joblib dataframe dask
Language:Python 2677
elixir-nx / nx
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
gpu numerical elixir tensor xla pytorch jit
Language:Elixir 2467
DataFrame
hosseinmoein / DataFrame
C++ DataFrame for statistical, Financial, and ML analysis -- in modern C++ using native types and contiguous memory storage
numerical-analysis dataframe data-analysis multidimensional-data cpp large-data heterogeneous-data statistical-analysis financial-data-analysis financial-engineering trading-strategies trading-algorithms statistical machine-learning data-science tensor tensorboard ai pandas polars
Language:C++ 2267
merrymercy / awesome-tensor-compilers
A list of awesome compiler projects and papers for tensor computation and deep learning.
compiler tensor deep-learning machine-learning high-performance-computing code-generation programming-language
2169
tairov / llama2.mojo
Inference Llama 2 in one file of pure 🔥
inference llama llama2 modular mojo parallelize performance simd tensor transformer-architecture vectorization
Language:Mojo 1939
coreylowman / dfdx
Deep learning in Rust, with shape checked tensors and neural networks
autodiff autodifferentiation autograd backpropagation cuda cuda-kernels cuda-support cuda-toolkit cudnn deep-learning deep-neural-networks gpu gpu-acceleration gpu-computing machine-learning neural-network rust rust-lang tensor
Language:Rust 1606
unslothai / hyperlearn
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
data-analysis data-science deep-learning econometrics gpu machine-learning neural-network optimization python pytorch regression-models research scikit-learn statistics statsmodels tensor
Language:Jupyter Notebook 1569
tensorly
tensorly / tensorly
TensorLy: Tensor Learning in Python.
cupy decomposition jax machine-learning mxnet numpy python pytorch regression tensor tensor-algebra tensor-decomposition tensor-factorization tensor-learning tensor-methods tensor-regression tensorflow tensorly
Language:Python 1495
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 nim multidimensional-arrays cuda deep-learning machine-learning cudnn high-performance-computing gpu-computing matrix-library neural-networks parallel-computing openmp linear-algebra ndarray opencl gpgpu iot automatic-differentiation autograd
Language:Nim 1307
tensor-compiler / taco
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
code-generator library linear-algebra sparse tensor tensor-algebra tensor-algebra-compiler
Language:C++ 1208
egison / egison
The Egison Programming Language
egison programming-language functional-programming pattern-matching computer-algebra-system tensor differential-geometry hacktoberfest
Language:Haskell 903
dmlc / dlpack
common in-memory tensor structure
deep-learning operator tensor
Language:Python 851
andrewssobral / lrslibrary
Low-Rank and Sparse Tools for Background Modeling and Subtraction in Videos
rpca matrix-factorization matrix-completion tensor-decomposition tensor matlab matrix subspace-tracking subspace-learning
Language:MATLAB 811
dgasmith / opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
python tensor contraction gpu-acceleration performance einsum tensor-contraction
Language:Python 803
libxsmm / libxsmm
Library for specialized dense and sparse matrix operations, and deep learning primitives.
jit simd avx512 machine-learning sparse blas matrix-multiplication transpose bfloat16 avx2 avx sse vector intel matrix tensor convolution amx fortran
Language:C 798
OAID / AutoKernel
AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。
tensorflow pytorch tvm halide tengine deep-learning auto reinforcement-learning autosearch optimization tensor
Language:C++ 777
OpenMined / TenSEAL
A library for doing homomorphic encryption operations on tensors
cpp cryptography deep-learning docker-image encryption hacktoberfest homomorphic-encryption microsoft-seal python tensor
Language:C++ 744
pytorch / executorch
On-device AI across mobile, embedded and edge for PyTorch
deep-learning embedded gpu machine-learning mobile neural-network tensor
Language:C++ 739
zdevito / ATen
ATen: A TENsor library for C++11
pytorch tensor torch7
Language:C++ 662
norse
norse / norse
Deep learning with spiking neural networks (SNNs) in PyTorch.
autograd deep-learning gpu machine-learning neural-network neuromorphic pytorch pytorch-lightning spiking-neural-networks tensor
Language:Python 611
mcabbott / Tullio.jl
⅀
automatic-differentiation einsum julia notation tensor
Language:Julia 581
DiffSharp / DiffSharp
DiffSharp: Differentiable Functional Programming
autodiff deep-learning dotnet gpu machine-learning neural-network tensor
Language:F# 573
xwying / torchshow
Visualize PyTorch tensors with a single line of code.
visualization tensor image-tensor notebook pytorch
Language:Python 552
raskr / rust-autograd
Tensors and differentiable operations (like TensorFlow) in Rust
rust tensor automatic-differentiation autograd neural-networks deep-learning machine-learning
Language:Rust 480
Grassmann.jl
chakravala / Grassmann.jl
⟨Grassmann-Clifford-Hodge⟩ multilinear differential geometric algebra
geometric-algebra clifford-algebras lie-groups tensor-algebra automatic-differentiation complex-numbers tensor quaternion math computer-algebra differential-geometry linear-algebra topology quantum-computing computational-geometry algebra discrete-exterior-calculus simplicial-complexes manifolds
Language:Julia 449
TensorOperations.jl
Jutho / TensorOperations.jl
Julia package for tensor contractions and related operations
tensor tensor-contraction tensor-permutation tensor-trace tensor-transposition tensor-operations einstein-summation index-notation
Language:Julia 414
LaurentMazare / ocaml-torch
OCaml bindings for PyTorch
artificial-intelligence deep-learning gpu machine-learning neural-network ocaml pytorch tensor
Language:OCaml 399
blitzpp / blitz
Blitz++ Multi-Dimensional Array Library for C++
blitz array-manipulations tensor array vector cpp-library scientific-computing high-performance hpc template-metaprogramming numerics numerical-calculations numerical-computation multi-dimensional-array partial-evaluators
Language:C++ 397