webdataset / tarp
Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.
☆123Updated 11 months ago
Related projects ⓘ
Alternatives and complementary repositories for tarp
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆73Updated 11 months ago
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆73Updated 2 years ago
- A python library that provides common I/O interface across different storage backends.☆134Updated 3 weeks ago
- gpu tester detects broken and slow gpus in a cluster☆67Updated last year
- CLOOB training (JAX) and inference (JAX and PyTorch)☆70Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆124Updated 2 years ago
- Simple python template☆40Updated 6 months ago
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆56Updated 2 years ago
- ☆27Updated 3 years ago
- Scalable and Performant Data Loading☆66Updated this week
- ☆107Updated 3 years ago
- Aggregating embeddings over time☆31Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆207Updated last year
- Optimized library for large-scale extraction of frames and audio from video.☆203Updated last year
- ☆64Updated 3 years ago
- ☆25Updated 3 years ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- ☆25Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆179Updated last year
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆82Updated last month
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆234Updated last year
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- Utilities for sequential processing of tar files.☆24Updated 2 years ago