TezRomacH / layer-to-layer-pytorch
PyTorch implementation of L2L execution algorithm
☆107Updated 2 years ago
Alternatives and similar repositories for layer-to-layer-pytorch:
Users that are interested in layer-to-layer-pytorch are comparing it to the libraries listed below
- Train ImageNet in 18 minutes on AWS☆127Updated 10 months ago
- MONeT framework for reducing memory consumption of DNN training☆174Updated 3 years ago
- Block-sparse primitives for PyTorch☆154Updated 3 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆116Updated 3 years ago
- Torch Distributed Experimental☆115Updated 5 months ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆63Updated 2 years ago
- Learning to Initialize Neural Networks for Stable and Efficient Training☆138Updated 2 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- Compression schema for gradients of activations in backward pass☆44Updated last year
- ☆109Updated 3 years ago
- Implementation of a Transformer, but completely in Triton☆253Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 2 years ago
- Research and development for optimizing transformers☆125Updated 3 years ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆111Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆193Updated 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
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated last year
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- This repository contains example code to build models on TPUs☆30Updated last year
- Efficient, check-pointed data loading for deep learning with massive data sets.☆205Updated last year
- Knowledge Distillation Toolkit☆90Updated 4 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 3 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago