The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
β809Apr 7, 2022Updated 3 years ago
Alternatives and similar repositories for tensor-sensor
Users that are interested in tensor-sensor are comparing it to the libraries listed below
Sorting:
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)β9,415Feb 20, 2026Updated 2 weeks ago
- Lightweight Cluster/Cloud VM Job Management πβ42Aug 27, 2024Updated last year
- HiPlot makes understanding high dimensional data easyβ2,798Jan 10, 2024Updated 2 years ago
- Hummingbird compiles trained ML models into tensor computation for faster inference.β3,531Jul 17, 2025Updated 7 months ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.β1,471May 2, 2025Updated 10 months ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edgesβ63Oct 20, 2022Updated 3 years ago
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networksβ570Jun 13, 2022Updated 3 years ago
- Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domainsβ1,731Oct 8, 2023Updated 2 years ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.β30,906Updated this week
- Model interpretability and understanding for PyTorchβ5,564Feb 26, 2026Updated last week
- A scikit-learn compatible neural network library that wraps PyTorchβ6,151Feb 25, 2026Updated last week
- Hydra is a framework for elegantly configuring complex applicationsβ10,248Feb 7, 2026Updated last month
- Accelerated deep learning R&Dβ3,372Jun 27, 2025Updated 8 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual trβ¦β1,627Mar 25, 2022Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorchβ3,163Mar 22, 2024Updated last year
- Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining theβ¦β2,085Aug 15, 2024Updated last year
- PyTorch, TensorFlow, JAX and NumPy β all of them natively using the same codeβ699Mar 25, 2023Updated 2 years ago
- Version control for machine learningβ1,670Feb 25, 2025Updated last year
- Mathematical operations for JAX pytreesβ209Dec 5, 2024Updated last year
- Aim π« β An easy-to-use & supercharged open-source experiment tracker.β6,017Mar 1, 2026Updated last week
- Debugging, monitoring and visualization for Python Machine Learning and Data Scienceβ3,463Sep 27, 2025Updated 5 months ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networksβ1,726Feb 26, 2026Updated last week
- A python library for decision tree visualization and model interpretation.β3,126Jan 2, 2026Updated 2 months ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.β1,829Jan 18, 2026Updated last month
- JAX-based neural network libraryβ3,193Mar 2, 2026Updated last week
- β927Jan 29, 2026Updated last month
- Latex code for making neural networks diagramsβ24,463Aug 21, 2023Updated 2 years ago
- A Graph Neural Network Library in Jaxβ1,466Mar 18, 2024Updated last year
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networksβ488Jul 1, 2022Updated 3 years ago
- Algorithms for outlier, adversarial and drift detectionβ2,495Dec 11, 2025Updated 2 months ago
- Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposalsβ13,298Updated this week
- π Geometric Computer Vision Library for Spatial AIβ11,104Mar 2, 2026Updated last week
- Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per sβ¦β8,483Mar 1, 2026Updated last week
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.β4,751Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.β7,105Updated this week
- Bayesian active learning library for research and industrial usecases.β922Dec 3, 2025Updated 3 months ago
- A game theoretic approach to explain the output of any machine learning model.β25,079Updated this week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.β1,752Jan 20, 2026Updated last month
- Documentation:β129May 22, 2023Updated 2 years ago