TheStageAI / TorchIntegral
Integral Neural Networks in PyTorch
☆125Updated 5 months ago
Alternatives and similar repositories for TorchIntegral
Users that are interested in TorchIntegral are comparing it to the libraries listed below
Sorting:
- ☆156Updated 9 months ago
- Modern Fixed Point Systems using Pytorch☆91Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆204Updated last year
- [NeurIPS 2023] Official Implementation: "Ambient Diffusion: Learning Clean Distributions from Corrupted Data"☆79Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆69Updated 2 years ago
- Transformers as Meta-Learners for Implicit Neural Representations, in ECCV 2022☆152Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆213Updated 2 years ago
- Batch computation of the linear assignment problem on GPU.☆78Updated 3 months ago
- wavelet implicit neural representations☆158Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆369Updated last month
- Code for our NeurIPS 2022 paper☆368Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆89Updated 3 years ago
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆317Updated 4 months ago
- A simple minimal implementation of Reversible Vision Transformers☆125Updated last year
- My take on Flow Matching☆55Updated 4 months ago
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆341Updated 2 years ago
- ☆39Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆138Updated last year
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆142Updated 7 months ago
- GENIE: Higher-Order Denoising Diffusion Solvers☆95Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Updated last year
- ☆181Updated 5 months ago
- Flow-matching algorithms in JAX☆90Updated 9 months ago
- Named tensors with first-class dimensions for PyTorch☆319Updated last year
- Train VAE like a boss☆278Updated 6 months ago
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆47Updated 4 years ago
- Easy Hypernetworks in Pytorch and Jax☆100Updated 2 years ago
- [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator☆157Updated 2 years ago
- [ECCV 2024] Official PyTorch implementation of RoPE-ViT "Rotary Position Embedding for Vision Transformer"☆321Updated 4 months ago