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