TheStageAI / TorchIntegral
Integral Neural Networks in PyTorch
☆126Updated 2 months ago
Alternatives and similar repositories for TorchIntegral:
Users that are interested in TorchIntegral are comparing it to the libraries listed below
- Modern Fixed Point Systems using Pytorch☆86Updated last year
- ☆154Updated 6 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- A simple minimal implementation of Reversible Vision Transformers☆120Updated 11 months ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆200Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆352Updated this week
- NF-Layers for constructing neural functionals.☆82Updated last year
- Transformers as Meta-Learners for Implicit Neural Representations, in ECCV 2022☆149Updated 2 years ago
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆62Updated last year
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆337Updated 2 years ago
- Unofficial Implementation of Consistency Models in pytorch☆255Updated last year
- 94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds☆205Updated this week
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆129Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆557Updated 2 months ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆294Updated last month
- ☆306Updated 2 years ago
- A mini-library for training consistency models.☆234Updated last year
- Visualizing representations with diffusion based conditional generative model.☆89Updated last year
- Neighborhood Attention Extension. Bringing attention to a neighborhood near you!☆404Updated 2 weeks 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 the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Code for our NeurIPS 2022 paper☆366Updated 2 years ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆369Updated last year
- EDM2 and Autoguidance -- Official PyTorch implementation☆627Updated 2 months ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Batch computation of the linear assignment problem on GPU.☆63Updated 2 weeks ago
- Code for Fast Training of Diffusion Models with Masked Transformers☆387Updated 9 months ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆136Updated 3 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- [NeurIPS 2023] Official Implementation: "Ambient Diffusion: Learning Clean Distributions from Corrupted Data"☆77Updated last year