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
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- Customized matrix multiplication kernels☆54Updated 3 years ago
- Block-sparse primitives for PyTorch☆154Updated 4 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Research and development for optimizing transformers☆125Updated 4 years ago
- Compression schema for gradients of activations in backward pass☆44Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 3 years ago
- Torch Distributed Experimental☆115Updated 8 months ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆116Updated 3 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago
- Fast sparse deep learning on CPUs☆53Updated 2 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆137Updated 4 years ago
- Learning to Initialize Neural Networks for Stable and Efficient Training☆139Updated 2 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- Implementation of a Transformer, but completely in Triton☆263Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 4 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆156Updated 4 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆183Updated 2 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- ☆109Updated 4 years ago
- Loss Patterns of Neural Networks☆84Updated 3 years ago