TezRomacH / layer-to-layer-pytorch
PyTorch implementation of L2L execution algorithm
☆106Updated last year
Related projects ⓘ
Alternatives and complementary repositories for layer-to-layer-pytorch
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- Train ImageNet in 18 minutes on AWS☆126Updated 8 months ago
- Block-sparse primitives for PyTorch☆148Updated 3 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆116Updated 2 years ago
- PyProf2: PyTorch Profiling tool☆83Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Compression schema for gradients of activations in backward pass☆44Updated last year
- Learning to Initialize Neural Networks for Stable and Efficient Training☆136Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆249Updated 2 years ago
- Torch Distributed Experimental☆116Updated 3 months ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 3 years ago
- Research and development for optimizing transformers☆125Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- Implementation of Feedback Transformer in Pytorch☆104Updated 3 years ago
- A Re-implementation of Fixed-update Initialization☆151Updated 5 years ago
- Loss Patterns of Neural Networks☆82Updated 3 years ago
- Unofficial PyTorch Implementation of EvoNorm☆121Updated 3 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆296Updated 3 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆234Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆129Updated 2 years ago
- 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
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years 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
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆110Updated 4 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆42Updated 4 years ago
- ☆107Updated 3 years ago
- Programmable Neural Network Compression☆147Updated 2 years ago