NVlabs / tensorcom
☆109Updated 4 years ago
Alternatives and similar repositories for tensorcom:
Users that are interested in tensorcom are comparing it to the libraries listed below
- ☆25Updated 4 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Efficient Data Loading Pipeline in Pure Python☆212Updated 4 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 5 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- Make TFRecord Usable Again☆87Updated last year
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- PyTorch DataLoader processed in multiple remote computation machines for heavy data processings☆67Updated 5 years ago
- ☆57Updated 2 years ago
- ☆62Updated 4 years ago
- ☆33Updated 9 months ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆197Updated 5 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Torch Distributed Experimental☆115Updated 9 months ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- MONeT framework for reducing memory consumption of DNN training☆173Updated 4 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆210Updated last year
- Useful PyTorch functions and modules that are not implemented in PyTorch by default☆187Updated last year
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆29Updated 5 years ago
- Swish Activation - PyTorch CUDA Implementation☆37Updated 5 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆257Updated 2 years ago
- Distributed, mixed-precision training with PyTorch☆90Updated 4 years ago
- Scailable ONNX python tools☆97Updated 6 months ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago