ofnote / tsalib
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆265Updated 4 years ago
Alternatives and similar repositories for tsalib:
Users that are interested in tsalib are comparing it to the libraries listed below
- ☆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
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- Annotating tensor shapes using Python types☆159Updated last year
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- A High Level API for Deep Learning in JAX☆475Updated 2 years ago
- Docs☆143Updated 5 months ago
- Functional tensors for probabilistic programming☆239Updated last year
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆571Updated 2 years ago
- Concise deep learning for JAX☆183Updated 4 years ago
- Automatic GPU+CPU memory profiling, re-use and memory leaks detection using jupyter/ipython experiment containers☆215Updated last year
- Unit Testing for pytorch, based on mltest☆310Updated 4 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 4 years ago
- A user-centered Python package for differentiable probabilistic inference☆202Updated 4 years ago
- A utility library around PyTorch☆244Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆697Updated 2 years ago
- Seamlessly integrate matplotlib figures as tensorflow summaries.☆120Updated 6 years ago
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- Equi-normalization of Neural Networks☆115Updated 5 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- An ML framework to accelerate research and its path to production.☆265Updated 7 months ago
- ☆155Updated last year
- Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!☆377Updated 2 years ago
- Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.☆110Updated 4 months ago
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,112Updated 3 years ago
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆391Updated 11 months ago