CEA-LIST / RPCDataloaderLinks
A variant of the PyTorch Dataloader using remote workers.
☆21Updated 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☆231Updated 2 years ago
- Demystify RAM Usage in Multi-Process Data Loaders☆205Updated 2 years ago
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆133Updated 2 years ago
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆76Updated 2 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- PyCUDA based PyTorch Extension Made Easy☆26Updated last year
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆250Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆78Updated 3 years ago
- A python library that provides common I/O interface across different storage backends.☆150Updated last week
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- gpu tester detects broken and slow gpus in a cluster☆72Updated 2 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- Share numpy arrays across processes efficiently ideal for large, read-only datasets☆48Updated 8 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆305Updated 3 years ago
- Pytorch Custom CUDA kernel for searchsorted☆137Updated 2 years ago
- Customized matrix multiplication kernels☆57Updated 3 years ago
- ☆165Updated 7 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆635Updated 6 months ago
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆30Updated 6 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆237Updated 2 years ago
- Scalable and Performant Data Loading☆364Updated this week
- ☆107Updated 4 years ago
- A Re-implementation of Fixed-update Initialization☆156Updated 6 years ago
- Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.☆26Updated 11 months ago
- ☆24Updated last year
- Batch computation of the linear assignment problem on GPU.☆105Updated 4 months ago
- ☆62Updated 5 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 4 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Updated 2 years ago