TheStageAI / TorchIntegral
Integral Neural Networks in PyTorch
☆122Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for TorchIntegral
- Modern Fixed Point Systems using Pytorch☆82Updated last year
- ☆149Updated 3 months ago
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆59Updated last year
- wavelet implicit neural representations☆137Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Transformers as Meta-Learners for Implicit Neural Representations, in ECCV 2022☆147Updated 2 years ago
- A simple minimal implementation of Reversible Vision Transformers☆116Updated 7 months ago
- NF-Layers for constructing neural functionals.☆75Updated 10 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆326Updated 3 weeks ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)☆257Updated 2 months ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆194Updated 8 months ago
- Use Jax functions in Pytorch☆227Updated last year
- Code for our CVPR'23 paper: "Polynomial Implicit Neural Representations For Large Diverse Datasets"☆36Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆124Updated last year
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆337Updated 2 years ago
- A mini-library for training consistency models.☆213Updated 10 months ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆26Updated 9 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆238Updated 10 months ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- GENIE: Higher-Order Denoising Diffusion Solvers☆88Updated last year
- ☆212Updated last year
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆359Updated last year
- ☆256Updated last year
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆47Updated 6 months ago
- Code for our NeurIPS 2022 paper☆361Updated last year
- Flow-matching algorithms in JAX☆73Updated 2 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆77Updated 9 months ago
- Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)☆521Updated 5 months ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆359Updated last year