pseeth / autoclipLinks
Adaptive Gradient Clipping
☆133Updated 2 years ago
Alternatives and similar repositories for autoclip
Users that are interested in autoclip are comparing it to the libraries listed below
Sorting:
- ☆163Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆135Updated 2 months ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 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 fused cosine similarity attention in the same style as Flash Attention☆214Updated 2 years ago
- Implementation of Fast Transformer in Pytorch☆174Updated 3 years ago
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆121Updated 2 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆160Updated last year
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆204Updated last year
- Relative Positional Encoding for Transformers with Linear Complexity☆63Updated 3 years ago
- Traditional Machine Learning Models for Large-Scale Datasets in PyTorch.☆126Updated this week
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Online Normalization for Training Neural Networks (Companion Repository)☆82Updated 4 years ago
- Implementation of Linformer for Pytorch☆286Updated last year
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago
- Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.☆168Updated 2 months ago
- A PyTorch Implementation of the Sparsemax operator (https://arxiv.org/pdf/1803.09820.pdf)☆32Updated 2 years ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆94Updated last year
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated 2 years ago
- Implementation of Feedback Transformer in Pytorch☆107Updated 4 years ago
- Sequence Modeling with Structured State Spaces☆64Updated 2 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆217Updated 2 years ago
- Inspired by "Neural Networks Fail to Learn Periodic Functions and How to Fix It"☆66Updated last year
- An implementation of local windowed attention for language modeling☆450Updated 4 months ago
- ☆22Updated 7 months ago
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆264Updated 3 years ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆113Updated 4 years ago