Tensors and Dynamic neural networks in Python with strong GPU acceleration
-
Updated
Apr 5, 2025 - Python
#
tensor
Here are 146 public repositories matching this topic...
Open deep learning compiler stack for cpu, gpu and specialized accelerators
javascript machine-learning performance deep-learning metal compiler gpu vulkan opencl tensor spirv rocm tvm
-
Updated
Apr 5, 2025 - Python
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
-
Updated
Feb 9, 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
Mar 2, 2025 - Python
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Mar 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
Apr 2, 2025 - Python
Visualize PyTorch tensors with a single line of code.
-
Updated
Mar 20, 2025 - Python
Machine Learning with Symbolic Tensors
-
Updated
Mar 3, 2025 - Python
Let DI-treetensor help you simplify the structure processing!（树形运算一不小心就逻辑混乱？DI-treetensor快速帮你搞定）
-
Updated
Oct 21, 2024 - Python
Hyper optimized contraction trees for large tensor networks and einsums
-
Updated
Apr 3, 2025 - Python
Abstract your array operations.
-
Updated
Mar 13, 2025 - Python
Framework agnostic python runtime for RWKV models
-
Updated
Aug 24, 2023 - 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
Mar 28, 2025 - 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
Mar 13, 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
Provides a systematic and extensible way to build, train, evaluate, and tune deep learning models using PyTorch.
-
Updated
Jun 17, 2024 - Python
A Python module for compiling PyTorch graphs to C
-
Updated
Feb 2, 2018 - 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."