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:
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Updated 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
- ☆36Updated 3 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- ☆108Updated 4 years ago
- 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
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- ☆47Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Updated 2 years ago
- A Python library for Continual Inference Networks in PyTorch☆54Updated 9 months ago
- An open source implementation of CLIP.☆33Updated 3 years ago
- Code for the paper PermuteFormer☆42Updated 4 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 4 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆22Updated 3 years ago
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 3 months ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆78Updated 3 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated 2 years ago
- moai is a PyTorch-based AI Model Development Kit (MDK) created to improve data-driven model workflows, design and reproducibility.☆55Updated last month
- ☆25Updated 4 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆18Updated 3 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- Implementation of Feedback Transformer in Pytorch☆108Updated 4 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆219Updated 2 years ago
- High performance pytorch modules☆18Updated 2 years ago
- Code accompanying our NeurIPS 2020 traffic4cast challenge☆14Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 11 months ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆46Updated 5 years ago