pseeth / autoclip
Adaptive Gradient Clipping
☆130Updated 2 years ago
Alternatives and similar repositories for autoclip:
Users that are interested in autoclip are comparing it to the libraries listed below
- ☆164Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆131Updated 3 weeks ago
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆119Updated 2 years ago
- Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.☆125Updated 3 weeks ago
- Implementation of Fast Transformer in Pytorch☆173Updated 3 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆203Updated last year
- Implementation of Linformer for Pytorch☆278Updated last year
- ☆22Updated 5 months ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆214Updated 4 years ago
- Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.☆166Updated 3 weeks ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆212Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Relative Positional Encoding for Transformers with Linear Complexity☆62Updated 3 years ago
- Ranger deep learning optimizer rewrite to use newest components☆329Updated last year
- Sequence modeling with Mega.☆295Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆81Updated last year
- Inspired by "Neural Networks Fail to Learn Periodic Functions and How to Fix It"☆65Updated 11 months ago
- Online Normalization for Training Neural Networks (Companion Repository)☆81Updated 3 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆230Updated 2 months ago
- Code to accompany the paper "Hierarchical Quantized Autoencoders"☆37Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 11 months ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆260Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- A PyTorch Implementation of the Sparsemax operator (https://arxiv.org/pdf/1803.09820.pdf)☆32Updated 2 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 2 years ago