tensor

There are 30 repositories under tensor topic.

pytorch / pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
autograd deep-learning gpu machine-learning neural-network numpy python tensor
Language:Python 85847
chenyuntc / pytorch-book
PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
autograd caption charrnn deep-learning gan image-classification jupyter-notebook neural-style neuraltalk nn pytorch pytorch-tutorials pytorch-tutorials-cn tensor tensorboard visdom
Language:Jupyter Notebook 12270
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 11929
cupy / cupy
NumPy & SciPy for GPU
cublas cuda cudnn cupy curand cusolver cusparse cusparselt cutensor gpu nccl numpy nvrtc nvtx python rocm scipy tensor
Language:Python 9683
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 9411
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 8647
MegEngine / MegEngine
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
autograd deep-learning gpu machine-learning megengine numpy python tensor
Language:C++ 4778
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 3245
mars-project / mars
Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
dask dataframe joblib lightgbm machine-learning numpy pandas python pytorch ray scikit-learn statsmodels tensor tensorflow xgboost
Language:Python 2709
elixir-nx / nx
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
elixir gpu jit numerical pytorch tensor xla
Language:Elixir 2688
DataFrame
hosseinmoein / DataFrame
C++ DataFrame for statistical, Financial, and ML analysis -- in modern C++ using native types and contiguous memory storage
ai cpp data-analysis data-science dataframe financial-data-analysis financial-engineering heterogeneous-data large-data machine-learning multidimensional-data numerical-analysis pandas polars statistical statistical-analysis tensor tensorboard trading-algorithms trading-strategies
Language:C++ 2564
merrymercy / awesome-tensor-compilers
A list of awesome compiler projects and papers for tensor computation and deep learning.
code-generation compiler deep-learning high-performance-computing machine-learning programming-language tensor
2460
pytorch / executorch
On-device AI across mobile, embedded and edge for PyTorch
deep-learning embedded gpu machine-learning mobile neural-network tensor
Language:C++ 2395
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 2110
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 1902
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 1766
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 1584
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
autograd automatic-differentiation cuda cudnn deep-learning gpgpu gpu-computing high-performance-computing iot linear-algebra machine-learning matrix-library multidimensional-arrays ndarray neural-networks nim opencl openmp parallel-computing tensor
Language:Nim 1343
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++ 1267
dmlc / dlpack
common in-memory tensor structure
tensor operator deep-learning
Language:Python 930
egison / egison
The Egison Programming Language
computer-algebra-system differential-geometry egison functional-programming hacktoberfest pattern-matching programming-language tensor
Language:Haskell 913
dgasmith / opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
contraction einsum gpu-acceleration performance python tensor tensor-contraction
Language:Python 873
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++ 866
libxsmm / libxsmm
Library for specialized dense and sparse matrix operations, and deep learning primitives.
amx avx avx2 avx512 bfloat16 blas convolution fortran intel jit machine-learning matrix matrix-multiplication simd sparse sse tensor transpose vector
Language:C 856
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 845
OAID / AutoKernel
AutoKernel 是一个简单易用，低门槛的自动算子优化工具，提高深度学习算法部署效率。
auto autosearch deep-learning halide optimization pytorch reinforcement-learning tengine tensor tensorflow tvm
Language:C++ 781
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 689
zdevito / ATen
ATen: A TENsor library for C++11
tensor torch7 pytorch
Language:C++ 689
xwying / torchshow
Visualize PyTorch tensors with a single line of code.
image-tensor notebook pytorch tensor visualization
Language:Python 649
mcabbott / Tullio.jl
⅀
automatic-differentiation einsum julia notation tensor
Language:Julia 617
DiffSharp / DiffSharp
DiffSharp: Differentiable Functional Programming
machine-learning tensor dotnet autodiff gpu deep-learning neural-network
Language:F# 592
raskr / rust-autograd
Tensors and differentiable operations (like TensorFlow) in Rust
autograd automatic-differentiation deep-learning machine-learning neural-networks rust tensor
Language:Rust 488
TensorOperations.jl
Jutho / TensorOperations.jl
Julia package for tensor contractions and related operations
einstein-summation index-notation tensor tensor-contraction tensor-operations tensor-permutation tensor-trace tensor-transposition
Language:Julia 486
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 computer-algebra differential-geometry linear-algebra topology quantum-computing computational-geometry algebra discrete-exterior-calculus simplicial-complexes manifolds geometry-processing
Language:Julia 476
basalt
basalt-org / basalt
A Machine Learning framework from scratch in Pure Mojo 🔥
ai autograd deep-learning deep-neural-networks machine-learning ml mojo neural-network tensor
Language:Mojo 434
LaurentMazare / ocaml-torch
OCaml bindings for PyTorch
artificial-intelligence deep-learning gpu machine-learning neural-network ocaml pytorch tensor
Language:OCaml 417