HesitantlyHuman / autoclipLinks
Implementation of adaptive gradient clipping for base pytorch
☆19Updated last year
Alternatives and similar repositories for autoclip
Users that are interested in autoclip are comparing it to the libraries listed below
Sorting:
- ☆24Updated last year
- 1D interpolation for pytorch☆175Updated last year
- Normalizing flows in PyTorch☆421Updated last month
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆513Updated 2 years ago
- ☆328Updated 3 years ago
- Code for MGDM algorithm, ICML 2025, https://arxiv.org/abs/2502.03332☆15Updated 6 months ago
- A parallel ODE solver for PyTorch☆271Updated last year
- ☆57Updated 5 months ago
- Adaptive Gradient Clipping☆151Updated 3 years ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆186Updated 4 years ago
- Universal Notation for Tensor Operations in Python.☆449Updated 7 months ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆624Updated 11 months ago
- VQ-VAE implementation in Pytorch☆28Updated 5 years ago
- Differentiable fast wavelet transforms in PyTorch with GPU support.☆395Updated last week
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆291Updated last year
- ☆189Updated last year
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆73Updated last year
- A tiny library for stochastic dataset caching in PyTorch.☆44Updated last year
- Gaussian-Bernoulli Restricted Boltzmann Machines☆105Updated 2 years ago
- Fast Multi-dimensional Sparse Attention☆662Updated last week
- PyTorch implementation of normalizing flow models☆909Updated last year
- Differentiable signal processing on the sphere for PyTorch☆601Updated last week
- Variational Autoencoder (VAE) with Normalizing Flows☆69Updated last year
- Implementation of the proposed minGRU in Pytorch☆310Updated 8 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆481Updated last year
- Code for the paper: Complex-Valued Autoencoders for Object Discovery☆56Updated 2 years ago
- ☆224Updated 11 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆281Updated this week
- Code repository for Particle Denoising Diffusion Sampler☆15Updated last year