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,472May 2, 2025Updated 9 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
- Functional tensors for probabilistic programming☆245Aug 31, 2023Updated 2 years ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆971Jun 14, 2025Updated 8 months ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- ☆30Feb 11, 2022Updated 4 years ago
- A domain specific language to express machine learning workloads.☆1,765Apr 28, 2023Updated 2 years ago
- ONNX Integration Builds☆21May 21, 2018Updated 7 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- Annotating tensor shapes using Python types☆158Jul 7, 2023Updated 2 years ago
- Easy genetic algorithm☆14Mar 5, 2018Updated 7 years ago
- torchbearer: A model fitting library for PyTorch☆641Dec 4, 2023Updated 2 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆2,316Sep 29, 2022Updated 3 years ago
- A utility library around PyTorch☆244Dec 26, 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
- ☆153May 25, 2020Updated 5 years ago
- A Python toolbox for performing gradient-free optimization☆4,151May 30, 2025Updated 9 months ago
- Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses☆194Sep 19, 2023Updated 2 years ago
- Shape annotations for numpy arrays and pytorch/tensorflow tensors.☆14Jul 1, 2022Updated 3 years ago
- Universal Python binding for the LMDB 'Lightning' Database☆13Nov 7, 2017Updated 8 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,720Feb 10, 2026Updated 2 weeks ago
- ☆774Jan 27, 2024Updated 2 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,742Updated this week
- A Keras inspired training utility for PyTorch☆38Sep 13, 2018Updated 7 years ago
- Research language for array processing in the Haskell/ML family☆1,664Jan 5, 2026Updated last month
- Convenient hyperparameter optimization☆14Apr 30, 2024Updated last year
- ATen: A TENsor library for C++11☆717Nov 20, 2019Updated 6 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- Compiler with automatic differentiation☆49Oct 18, 2023Updated 2 years ago
- An intelligent block matrix library for numpy, PyTorch, and beyond.☆310Aug 4, 2024Updated last year
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago
- A language-agnostic interface to TensorBoard☆781Nov 9, 2017Updated 8 years ago