etienne87 / pytorch-stream-dataloaderLinks
☆52Updated 3 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.☆132Updated last year
- A Python library for Continual Inference Networks in PyTorch☆54Updated 10 months ago
- ☆107Updated 4 years 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
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Updated 4 years ago
- ☆36Updated 3 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆86Updated 4 years ago
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 5 months ago
- An open source implementation of CLIP.☆33Updated 3 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Updated 4 years ago
- moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.☆56Updated 3 weeks ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Updated 2 years ago
- ☆47Updated 5 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆24Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆78Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆108Updated 4 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Updated 3 years ago
- [ICLR 2022] "As-ViT: Auto-scaling Vision Transformers without Training" by Wuyang Chen, Wei Huang, Xianzhi Du, Xiaodan Song, Zhangyang Wa…☆76Updated 3 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 5 years ago
- Code for the paper PermuteFormer☆42Updated 4 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…☆196Updated 4 years ago
- ☆25Updated 4 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆57Updated 4 years ago
- Automatic learning-rate scheduler☆46Updated 4 years ago