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
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- ☆154Updated 7 months ago
- [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts☆59Updated 2 years ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆138Updated 4 months ago
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 4 months ago
- Artificial Kuramoto Oscillatory Neurons☆66Updated 3 weeks ago
- Modern Fixed Point Systems using Pytorch☆88Updated last year
- ☆49Updated 3 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆64Updated 2 weeks ago
- ☆23Updated 3 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- ☆57Updated 8 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆77Updated 7 months ago
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆63Updated last year
- ☆49Updated last year
- ☆23Updated 7 months ago
- Neural Diffusion Processes☆76Updated 7 months ago
- ☆61Updated 2 years ago
- Flow-matching algorithms in JAX☆85Updated 7 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆67Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 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
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 6 months ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- Efficient Video Prediction via Sparsely Conditioned Flow Matching. In ICCV, 2023.☆13Updated 9 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆23Updated last year