NVlabs / tensorcom
☆109Updated 3 years ago
Alternatives and similar repositories for tensorcom:
Users that are interested in tensorcom are comparing it to the libraries listed below
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Train ImageNet in 18 minutes on AWS☆128Updated 10 months ago
- Efficient Data Loading Pipeline in Pure Python☆211Updated 4 years ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Make TFRecord Usable Again☆87Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- 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
- Example repository for custom C++/CUDA operators for TorchScript☆115Updated 2 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated last year
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- ☆25Updated 3 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆79Updated 3 years ago
- A GPU performance profiling tool for PyTorch models☆501Updated 3 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆64Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- ☆26Updated 5 years ago
- ☆47Updated 4 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆158Updated 3 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- ☆62Updated 4 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- On Network Design Spaces for Visual Recognition☆94Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago