etienne87 / pytorch-stream-dataloaderLinks
☆52Updated 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.☆131Updated last year
- ☆36Updated 3 years ago
- A Python library for Continual Inference Networks in PyTorch☆53Updated 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
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Updated 4 years ago
- ☆108Updated 4 years ago
- An open source implementation of CLIP.☆33Updated 3 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- 🎲 Iterable dataset resampling in PyTorch☆92Updated 3 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 3 months ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆86Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆218Updated 2 years ago
- ☆47Updated 4 years ago
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Updated 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
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Implementing DropPath/StochasticDepth in PyTorch☆17Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- High performance pytorch modules☆18Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆56Updated 2 years ago
- Official Pytorch Implementation for "Continual Transformers: Redundancy-Free Attention for Online Inference" [ICLR 2023]☆28Updated 2 years ago
- ☆45Updated 5 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Updated 4 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
- Datasets, transforms and samplers for video in PyTorch☆88Updated 2 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- ☆25Updated 4 years ago