Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆268May 18, 2020Updated 5 years ago
Alternatives and similar repositories for tsalib
Users that are interested in tsalib are comparing it to the libraries listed below
Sorting:
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Nov 9, 2019Updated 6 years ago
- Named Tensor implementation for Torch☆448Jul 29, 2022Updated 3 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,475May 2, 2025Updated 10 months ago
- ☆108Dec 27, 2022Updated 3 years ago
- Jax-based MaxEnt☆17Nov 24, 2019Updated 6 years ago
- Write PyTorch code at the level of individual examples, then run it efficiently on minibatches.☆485Feb 12, 2022Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- Annotating tensor shapes using Python types☆159Jul 7, 2023Updated 2 years ago
- Functional tensors for probabilistic programming☆245Aug 31, 2023Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆974Mar 15, 2026Updated last week
- A domain specific language to express machine learning workloads.☆1,764Apr 28, 2023Updated 2 years ago
- ☆153May 25, 2020Updated 5 years ago
- ONNX Integration Builds☆21May 21, 2018Updated 7 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,437Aug 21, 2025Updated 7 months ago
- A Keras inspired training utility for PyTorch☆38Sep 13, 2018Updated 7 years ago
- A Pytree Module system for Deep Learning in JAX☆212Feb 26, 2023Updated 3 years ago
- Convenient hyperparameter optimization☆14Apr 30, 2024Updated last year
- ☆30Feb 11, 2022Updated 4 years ago
- Easy genetic algorithm☆14Mar 5, 2018Updated 8 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆2,316Sep 29, 2022Updated 3 years ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,752Mar 11, 2026Updated last week
- A utility library around PyTorch☆244Dec 26, 2022Updated 3 years ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,124Apr 20, 2022Updated 3 years ago
- A Python toolbox for performing gradient-free optimization☆4,164Updated this week
- torchbearer: A model fitting library for PyTorch☆641Dec 4, 2023Updated 2 years ago
- A lightweight library for PyTorch training tools and utilities☆1,719Mar 6, 2026Updated 2 weeks ago
- ☆774Jan 27, 2024Updated 2 years ago
- Research language for array processing in the Haskell/ML family☆1,668Jan 5, 2026Updated 2 months ago
- ATen: A TENsor library for C++11☆717Nov 20, 2019Updated 6 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- Named tensors with first-class dimensions for PyTorch☆332Jun 14, 2023Updated 2 years ago
- An intelligent block matrix library for numpy, PyTorch, and beyond.☆310Aug 4, 2024Updated last year
- Tools for PyTorch☆224Aug 10, 2022Updated 3 years ago
- An extended typed Lambda Calculus☆21Aug 5, 2024Updated last year
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆194Sep 19, 2023Updated 2 years ago
- A language-agnostic interface to TensorBoard☆780Nov 9, 2017Updated 8 years ago
- LabNotebook is a tool that allows you to flexibly monitor, record, save, and query all your machine learning experiments.☆528Mar 31, 2018Updated 7 years ago