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:
- Demystify RAM Usage in Multi-Process Data Loaders☆204Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆229Updated last year
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆133Updated last year
- Customized matrix multiplication kernels☆57Updated 3 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 3 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆224Updated 3 years ago
- Batch computation of the linear assignment problem on GPU.☆96Updated 2 months ago
- ☆108Updated 4 years ago
- Auction algorithm for solving linear assignment problem (LAP)☆75Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- A python library that provides common I/O interface across different storage backends.☆147Updated 2 months ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- ☆62Updated 5 years ago
- gpu tester detects broken and slow gpus in a cluster☆72Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago
- PyCUDA based PyTorch Extension Made Easy☆26Updated last year
- Minimal effort CLIs derived from type hints and parse from command line, config files and environment variables☆405Updated this week
- Pytorch Custom CUDA kernel for searchsorted☆136Updated 2 years ago
- A Re-implementation of Fixed-update Initialization☆155Updated 6 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆218Updated 2 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 4 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆142Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 5 months ago
- Make TFRecord Usable Again☆90Updated 2 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆86Updated 4 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆306Updated 3 years ago