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☆228Updated last year
- Demystify RAM Usage in Multi-Process Data Loaders☆204Updated 2 years ago
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- PyCUDA based PyTorch Extension Made Easy☆26Updated last year
- MONeT framework for reducing memory consumption of DNN training☆174Updated 4 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- ☆108Updated 4 years ago
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆132Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆217Updated 2 years ago
- A GPU performance profiling tool for PyTorch models☆508Updated 4 years ago
- Customized matrix multiplication kernels☆57Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆367Updated 7 years ago
- Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.☆26Updated 8 months ago
- Share numpy arrays across processes efficiently ideal for large, read-only datasets☆48Updated 8 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆222Updated 7 months ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated 2 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆629Updated 3 months ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆114Updated 5 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆306Updated 3 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆140Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 4 months ago
- ☆24Updated last year
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- Scalable and Performant Data Loading☆330Updated this week
- PyTorch layer-by-layer model profiler☆606Updated 4 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 4 years ago