CEA-LIST / RPCDataloaderLinks
A variant of the PyTorch Dataloader using remote workers.
☆19Updated 2 years ago
Alternatives and similar repositories for RPCDataloader
Users that are interested in RPCDataloader are comparing it to the libraries listed below
Sorting:
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆225Updated last year
- Demystify RAM Usage in Multi-Process Data Loaders☆194Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆214Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.☆25Updated 4 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆233Updated 5 months ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 3 years ago
- Batch computation of the linear assignment problem on GPU.☆83Updated 5 months ago
- Examples for the WebDataset PyTorch Dataset Library☆50Updated 4 years ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆110Updated 2 months ago
- Data-centric AI building blocks for computer vision applications☆50Updated this week
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆72Updated 2 years ago
- MONeT framework for reducing memory consumption of DNN training☆173Updated 4 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆91Updated 3 years ago
- ☆109Updated 4 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆269Updated 3 years ago
- ☆12Updated last year
- The simplest but fast implementation of matrix multiplication in CUDA.☆36Updated 11 months ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆218Updated 2 years ago
- Scalable and Performant Data Loading☆278Updated this week
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Updated 3 years ago
- ☆22Updated 8 months ago
- Collaborative documentation for and from Jean Zay users. Official Jean Zay documentation: http://www.idris.fr/eng/jean-zay/☆123Updated 11 months ago
- PyTorch centric eager mode debugger☆47Updated 6 months ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆113Updated 4 years ago