ofnote / tsalib
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆264Updated 4 years ago
Alternatives and similar repositories for tsalib:
Users that are interested in tsalib are comparing it to the libraries listed below
- Named Tensor implementation for Torch☆444Updated 2 years ago
- ☆153Updated 4 years ago
- PyTorch functions and utilities to make your life easier☆195Updated 3 years ago
- Asynchronous Distributed Hyperparameter Optimization.☆290Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- Configure Python functions explicitly and safely☆126Updated 2 months ago
- Dashboard for sacred. Monitor and access your past machine learning experiments.☆183Updated 5 years ago
- A High Level API for Deep Learning in JAX☆472Updated 2 years ago
- Annotating tensor shapes using Python types☆158Updated last year
- A utility library around PyTorch☆244Updated 2 years ago
- Lightweight interface to AWS☆47Updated 5 years ago
- Equi-normalization of Neural Networks☆115Updated 5 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆570Updated 2 years ago
- Functional tensors for probabilistic programming☆237Updated last year
- Automatic GPU+CPU memory profiling, re-use and memory leaks detection using jupyter/ipython experiment containers☆211Updated last year
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆392Updated 8 months ago
- Concise deep learning for JAX☆184Updated 4 years ago
- Interactively retrieve data from sacred experiments.☆82Updated last month
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆252Updated 2 years ago
- Tricks for Colab power users☆171Updated 4 years ago
- A cleaner way to build neural networks for PyTorch.☆184Updated 5 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- A toolset for black-box hyperparameter optimisation.☆136Updated 5 years ago
- Differentiable bitonic sorting☆138Updated 4 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆210Updated 2 years ago
- Docs☆143Updated 2 months ago
- Decoupled Neural Interfaces using Synthetic Gradients for PyTorch☆236Updated 6 years ago