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
- Code for the paper: Rotating Features for Object Discovery☆50Updated 8 months ago
- ☆156Updated 8 months ago
- Modern Fixed Point Systems using Pytorch☆89Updated last year
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆61Updated 2 years ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆73Updated 2 months ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆32Updated 7 months ago
- ☆49Updated last year
- ☆50Updated 4 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆140Updated 6 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆79Updated 9 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 5 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 release for "Stochastic Optimal Control Matching"☆33Updated 8 months 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
- ☆28Updated 3 weeks ago
- ☆37Updated last year
- ☆60Updated 10 months ago
- ☆62Updated 3 years ago
- ☆61Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆127Updated 2 months ago
- ☆51Updated 10 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆75Updated last month
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- Beyond Straight-Through☆94Updated last year
- ☆15Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆34Updated 11 months ago