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☆197Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆226Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆75Updated last year
- Pytorch Custom CUDA kernel for searchsorted☆137Updated last year
- Examples for the WebDataset PyTorch Dataset Library☆51Updated 4 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 6 years ago
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆131Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 3 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 4 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 5 years ago
- gpu tester detects broken and slow gpus in a cluster☆70Updated 2 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆236Updated 2 years ago
- ☆165Updated 6 years ago
- Profiling and inspecting memory in pytorch☆1,066Updated last year
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆129Updated 3 years ago
- A GPU performance profiling tool for PyTorch models☆504Updated 4 years ago
- A Re-implementation of Fixed-update Initialization☆155Updated 6 years ago
- Customized matrix multiplication kernels☆56Updated 3 years ago
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆30Updated 5 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆236Updated last week
- A python library that provides common I/O interface across different storage backends.☆147Updated last month
- ☆62Updated 5 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆83Updated 7 years ago
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆42Updated 4 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆215Updated 2 years ago
- PyTorch layer-by-layer model profiler☆606Updated 4 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago