AllanYangZhou / nfn
NF-Layers for constructing neural functionals.
☆84Updated last year
Alternatives and similar repositories for nfn
Users that are interested in nfn are comparing it to the libraries listed below
Sorting:
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆51Updated 9 months ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- ☆156Updated 9 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆83Updated 2 months ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 6 months ago
- ☆50Updated 4 years ago
- This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …☆86Updated 2 years ago
- Modern Fixed Point Systems using Pytorch☆90Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆79Updated 9 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆78Updated 2 months ago
- Neural Diffusion Processes☆77Updated 9 months ago
- ☆51Updated 11 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated 11 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆141Updated 2 months ago
- ☆49Updated last year
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆32Updated 8 months ago
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆89Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- Code for ICLR 2023 Paper, "Stable Target Field for Reduced Variance Score Estimation in Diffusion Models”☆75Updated last year
- ☆64Updated 10 months ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆146Updated 3 years ago
- ☆60Updated 2 years ago
- ☆29Updated 9 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆38Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control☆25Updated last week