Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Sep 1, 2026 - Python
#
tensor
Here are 947 public repositories matching this topic...
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
rust machine-learning deep-learning metal cross-platform neural-network vulkan cuda wasm pytorch scientific-computing ndarray tensor webgpu rocm autodiff onnx kernel-fusion
-
Updated
Sep 1, 2026 - Rust
Open Machine Learning Compiler Framework
javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm
-
Updated
Sep 1, 2026 - Python
PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)
deep-learning jupyter-notebook nn pytorch autograd caption gan image-classification tensorboard tensor neural-style visdom pytorch-tutorials pytorch-tutorials-cn charrnn neuraltalk
-
Updated
Dec 24, 2023 - Jupyter Notebook
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
-
Updated
Aug 26, 2026 - Python
The AI search platform
search java search-engine machine-learning big-data ai server vector tensor vespa serving-recommendation rag vector-search vector-database
-
Updated
Sep 1, 2026 - Java
On-device AI across mobile, embedded and edge for PyTorch
-
Updated
Sep 1, 2026 - Python
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
-
Updated
Oct 24, 2024 - C++
LibRec: A Leading Java Library for Recommender Systems, see
java matrix collaborative-filtering matrix-factorization systems sparse recommender tensor-factorization collaborative tensor factorization filtering recommender-systems recommendation-algorithms probabilistic-graphical-models
-
Updated
Jul 13, 2023 - Java
C++ DataFrame for statistical, financial, and ML analysis in modern C++
data-science machine-learning ai cpp statistical-analysis data-analysis tensorboard trading-strategies trading-algorithms tensor statistical dataframe numerical-analysis financial-engineering multidimensional-data financial-data-analysis large-data
-
Updated
Aug 31, 2026 - C++
A list of awesome compiler projects and papers for tensor computation and deep learning.
programming-language machine-learning deep-learning compiler high-performance-computing code-generation tensor
-
Updated
Oct 19, 2024
Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
python machine-learning tensorflow numpy scikit-learn pandas pytorch xgboost lightgbm tensor dask ray dataframe statsmodels joblib
-
Updated
Jan 2, 2024 - Python
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
python data-science machine-learning statistics research deep-learning neural-network gpu optimization scikit-learn pytorch econometrics data-analysis tensor regression-models statsmodels
-
Updated
Nov 19, 2024 - Jupyter Notebook
Inference Llama 2 in one file of pure 🔥
performance modular mojo inference simd llama tensor vectorization parallelize transformer-architecture llama2
-
Updated
Feb 9, 2026 - Mojo
Deep learning in Rust, with shape checked tensors and neural networks
rust machine-learning deep-neural-networks deep-learning neural-network gpu cuda autograd rust-lang gpu-acceleration cuda-kernels tensor gpu-computing backpropagation cudnn cuda-toolkit cuda-support autodiff autodifferentiation
-
Updated
Jul 23, 2024 - Rust
SIMD-accelerated distances, dot products, matrix ops, geospatial & geometric kernels for 16 numeric types — from 6-bit floats to 64-bit complex — across x86, Arm, RISC-V, and WASM, with bindings for Python, Rust, C, C++, Swift, JS, and Go 📐
javascript swift rust golang information-retrieval cpp metrics numpy assembly simd matrix-multiplication scipy blas tensor arm-neon vector-search
-
Updated
Sep 1, 2026 - C
TensorLy: Tensor Learning in Python.
python machine-learning mxnet tensorflow numpy regression pytorch decomposition tensor-factorization tensor tensor-algebra tensorly tensor-learning tensor-decomposition cupy tensor-methods jax tensor-regression
-
Updated
Nov 16, 2025 - Python
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
iot machine-learning nim deep-learning opencl linear-algebra automatic-differentiation openmp parallel-computing cuda autograd gpgpu neural-networks high-performance-computing ndarray tensor gpu-computing multidimensional-arrays cudnn matrix-library
-
Updated
May 26, 2026 - Nim
Add this topic to your repo
To associate your repository with the tensor topic, visit your repo's landing page and select "manage topics."