Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Dec 16, 2025 - Python
#
tensor
Here are 163 public repositories matching this topic...
Open Machine Learning Compiler Framework
javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm
-
Updated
Dec 16, 2025 - Python
On-device AI across mobile, embedded and edge for PyTorch
-
Updated
Dec 16, 2025 - Python
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
-
Updated
Nov 24, 2025 - Python
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
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
Deep learning with spiking neural networks (SNNs) in PyTorch.
machine-learning deep-learning neural-network gpu pytorch autograd spiking-neural-networks tensor neuromorphic pytorch-lightning
-
Updated
Sep 23, 2025 - Python
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Jun 14, 2025 - Python
Read and write Neuroglancer datasets programmatically.
python cloud big-data serverless numpy matrix s3 gcs mesh tensor electron-microscopy connectomics microscopy skeletons biomedical-image-processing volumetric-data microscopy-images neuroglancer large-image chunked-image
-
Updated
Dec 16, 2025 - Python
Hyper optimized contraction trees for large tensor networks and einsums
-
Updated
Dec 15, 2025 - Python
Phase-Amplitude Coupling under Python
parallel-computing neuroscience pac tensor filtering coupling phase amplitude oscillations phase-amplitude-coupling event-related
-
Updated
Jul 30, 2024 - Python
Tensor Train Toolbox
data-driven dynamical-systems als tensor dmrg tensor-decomposition scikit mandy tensor-train dynamic-mode-decomposition quantum-simulation tensor-network mals slim-decomposition extended-dynamic-mode-decomposition
-
Updated
Oct 24, 2025 - Python
Keras implementation of Neural Graph Fingerprints as proposed by Duvenaud et al., 2015
atom graph graph-algorithms gpu keras molecule fingerprint neural-networks tensor differentiable-models morgan-fingerprints
-
Updated
Jan 25, 2019 - Python
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
deep-neural-networks deep-learning pytorch tensor tensor-contraction tensor-learning tensor-networks factorized-cnns tensor-convolution-networks tensor-regression deep-tensor-learning tensorized-networks factorized-convolution tensor-regression-layers factorized-networks
-
Updated
Jun 9, 2024 - Python
Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.
-
Updated
Jun 17, 2024 - Python
Framework agnostic python runtime for RWKV models
-
Updated
Aug 24, 2023 - Python
Machine Learning with Symbolic Tensors
-
Updated
Nov 20, 2025 - Python
depth map computation
map image images structure depth stereo tensor disparity epi sgm lightfield light-field block-matching light-fields lightfields epipolar sgbm semi-global-matching
-
Updated
Jan 5, 2023 - Python
Visualize PyTorch tensors with a single line of code.
-
Updated
Aug 29, 2025 - Python
Improve this page
Add a description, image, and links to the tensor topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the tensor topic, visit your repo's landing page and select "manage topics."