ofnote / tsalibLinks
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆266Updated 5 years ago
Alternatives and similar repositories for tsalib
Users that are interested in tsalib are comparing it to the libraries listed below
Sorting:
- Named Tensor implementation for Torch☆445Updated 2 years ago
- ☆153Updated 5 years ago
- Annotating tensor shapes using Python types☆159Updated last year
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- Configure Python functions explicitly and safely☆126Updated 7 months ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- A utility library around PyTorch☆244Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 3 years ago
- Lightweight interface to AWS☆47Updated 5 years ago
- Functional tensors for probabilistic programming☆241Updated last year
- Equi-normalization of Neural Networks☆115Updated 6 years ago
- Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!☆377Updated 2 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 5 years ago
- Read Google Cloud Storage, Azure Blobs, and local paths with the same interface☆63Updated this week
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆570Updated 3 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆208Updated 2 years ago
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆391Updated last year
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆239Updated 2 years ago
- Dashboard for sacred. Monitor and access your past machine learning experiments.☆184Updated 6 years ago
- Differentiable bitonic sorting☆141Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆190Updated 2 years ago
- A demonstration of the attention mechanism with some toy experiments and explanations.☆108Updated 6 years ago
- ☆108Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- An ML framework to accelerate research and its path to production.☆265Updated 9 months ago
- Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.☆107Updated 2 years ago