tmbdev-archive / webdataset-lightning
A small demonstration of using WebDataset with ImageNet and PyTorch Lightning
☆73Updated 9 months ago
Related projects: ⓘ
- ☆25Updated 3 years ago
- An open source implementation of CLIP.☆32Updated last year
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆120Updated 9 months ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆73Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆43Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆123Updated 2 years ago
- A criticism of a recent paper on buggy image downsampling methods in popular image processing and deep learning libraries.☆71Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆204Updated last year
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆39Updated 3 years ago
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆126Updated 8 months ago
- ☆107Updated 3 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Make TFRecord Usable Again☆87Updated last year
- ☆40Updated this week
- gpu tester detects broken and slow gpus in a cluster☆63Updated last year
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 2 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆78Updated 3 years ago
- Pytorch Lightning seed project with hydra☆18Updated 3 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- A convolution-free, transformer-only version of the CycleGAN framework☆32Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆123Updated 2 years ago