AllanYangZhou / nfn
NF-Layers for constructing neural functionals.
☆82Updated 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 7 months ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆59Updated 2 years ago
- Modern Fixed Point Systems using Pytorch☆88Updated last year
- ☆50Updated 4 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- ☆155Updated 7 months ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 6 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆67Updated last month
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 4 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- Neural Diffusion Processes☆76Updated 7 months ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆138Updated 5 months ago
- ☆23Updated 3 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆77Updated 8 months ago
- ☆24Updated 7 months ago
- ☆58Updated 9 months ago
- ☆61Updated 2 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 …☆84Updated 2 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- [ICML 2024] LLM and Simulation as Bilevel Optimizers: A New Paradigm to Advance Physical Scientific Discovery☆65Updated 9 months ago
- Score-based generative models for compact manifolds☆108Updated last year
- Code release for "Stochastic Optimal Control Matching"☆30Updated 7 months ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆112Updated last month
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆140Updated 3 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- ☆24Updated last week
- code for "Riemannian Flow Matching on General Geometries".☆206Updated last year