ofnote / tsalib
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆266Updated 4 years ago
Alternatives and similar repositories for tsalib
Users that are interested in tsalib are comparing it to the libraries listed below
Sorting:
- ☆153Updated 4 years ago
- Named Tensor implementation for Torch☆444Updated 2 years ago
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- Configure Python functions explicitly and safely☆126Updated 5 months ago
- Annotating tensor shapes using Python types☆159Updated last year
- A utility library around PyTorch☆244Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- Equi-normalization of Neural Networks☆115Updated 5 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- Differentiable bitonic sorting☆140Updated 4 years ago
- Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!☆377Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Lightweight interface to AWS☆47Updated 5 years ago
- Functional tensors for probabilistic programming☆240Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Automatic GPU+CPU memory profiling, re-use and memory leaks detection using jupyter/ipython experiment containers☆216Updated last year
- Concise deep learning for JAX☆183Updated 4 years ago
- Unit Testing for pytorch, based on mltest☆310Updated 4 years ago
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- An ML framework to accelerate research and its path to production.☆265Updated 8 months ago
- A user-centered Python package for differentiable probabilistic inference☆202Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 10 months ago
- Seamlessly integrate matplotlib figures as tensorflow summaries.☆120Updated 6 years ago
- Loss Patterns of Neural Networks☆84Updated 3 years ago
- ☆103Updated 4 years ago
- Training Transformer-XL on 128 GPUs☆140Updated 4 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 5 months ago
- ☆156Updated last year
- Tricks for Colab power users☆169Updated 5 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year