CEA-LIST / RPCDataloader
A variant of the PyTorch Dataloader using remote workers.
☆18Updated last year
Alternatives and similar repositories for RPCDataloader:
Users that are interested in RPCDataloader are comparing it to the libraries listed below
- Demystify RAM Usage in Multi-Process Data Loaders☆187Updated last year
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆217Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆75Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆22Updated 4 months ago
- Pytorch Custom CUDA kernel for searchsorted☆136Updated last year
- Collaborative documentation for and from Jean Zay users. Official Jean Zay documentation: http://www.idris.fr/eng/jean-zay/☆116Updated 6 months ago
- Website-based resource monitor for Slurm system☆35Updated last year
- ☆50Updated last year
- gpu tester detects broken and slow gpus in a cluster☆67Updated 2 years ago
- A LARS implementation in PyTorch☆341Updated 5 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- A tiny library for stochastic dataset caching in PyTorch.☆40Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆41Updated 3 years ago
- ☆22Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆183Updated 2 months ago
- Data-centric AI building blocks for computer vision applications☆46Updated this week
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated 2 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆70Updated 2 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- The PASS dataset: pretrained models and how to get the data☆263Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Customized matrix multiplication kernels☆53Updated 2 years ago
- ☆62Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆137Updated 3 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago