etienne87 / pytorch-stream-dataloaderLinks
☆51Updated 2 years ago
Alternatives and similar repositories for pytorch-stream-dataloader
Users that are interested in pytorch-stream-dataloader are comparing it to the libraries listed below
Sorting:
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆129Updated 8 months ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- ☆36Updated 3 years ago
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆58Updated 4 years ago
- ☆109Updated 4 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 3 years ago
- A Python library for Continual Inference Networks in PyTorch☆52Updated 4 months ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆20Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆55Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 4 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Code for the paper PermuteFormer☆42Updated 3 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆17Updated 3 years ago
- A simple wrapper library for binding timm models as detectron2 backbones☆43Updated 2 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆48Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- Implementation of Hierarchical Transformer Memory (HTM) for Pytorch☆75Updated 3 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆214Updated 2 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 3 years ago
- ☆47Updated 4 years ago
- ☆33Updated last month
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆119Updated 4 years ago
- Implementation of Feedback Transformer in Pytorch☆107Updated 4 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- ☆25Updated 4 years ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆100Updated last year