ofnote / tsalib
Tensor Shape Annotation Library (numpy, tensorflow, pytorch, ...)
☆262Updated 4 years ago
Related projects ⓘ
Alternatives and complementary repositories for tsalib
- Named Tensor implementation for Torch☆443Updated 2 years ago
- ☆155Updated 4 years ago
- Annotating tensor shapes using Python types☆158Updated last year
- PyTorch functions and utilities to make your life easier☆195Updated 3 years ago
- Asynchronous Distributed Hyperparameter Optimization.☆286Updated 11 months ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- A utility library around PyTorch☆244Updated last year
- Configure Python functions explicitly and safely☆126Updated this week
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Interactively retrieve data from sacred experiments.☆81Updated 3 weeks ago
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Functional tensors for probabilistic programming☆236Updated last year
- A user-centered Python package for differentiable probabilistic inference☆203Updated 4 years ago
- An ML framework to accelerate research and its path to production.☆262Updated 2 months ago
- Seamlessly integrate matplotlib figures as tensorflow summaries.☆119Updated 5 years ago
- Equi-normalization of Neural Networks☆115Updated 5 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆391Updated 6 months ago
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆569Updated 2 years ago
- Dashboard for sacred. Monitor and access your past machine learning experiments.☆184Updated 5 years ago
- Unit Testing for pytorch, based on mltest☆311Updated 4 years ago
- A High Level API for Deep Learning in JAX☆470Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- Differentiable bitonic sorting☆138Updated 4 years ago
- Basic experiment framework for tensorflow.☆92Updated 3 years ago
- Python library to easily log experiments and parallelize hyperparameter search for neural networks☆736Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- Documentation:☆120Updated last year
- Concise deep learning for JAX☆184Updated 4 years ago