CEA-LIST / RPCDataloader
A variant of the PyTorch Dataloader using remote workers.
☆17Updated 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
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆209Updated 2 years ago
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆216Updated last year
- Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.☆25Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆224Updated this week
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- Data-centric AI building blocks for computer vision applications☆44Updated this week
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- A LARS implementation in PyTorch☆338Updated 4 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆210Updated last year
- Pytorch Custom CUDA kernel for searchsorted☆135Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆252Updated 2 years ago
- Pytorch implementation of LOST unsupervised object discovery method☆239Updated last year
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆136Updated 3 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 8 months ago
- A python library that provides common I/O interface across different storage backends.☆138Updated 2 months ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 3 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆125Updated 2 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆226Updated 2 years ago
- A GPU performance profiling tool for PyTorch models☆500Updated 3 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆432Updated 4 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆202Updated last year
- ☆109Updated 3 years ago
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated last year
- Estimate/count FLOPS for a given neural network using pytorch☆304Updated 2 years ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆341Updated last month