mkofinas / neural-graphs
Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
☆77Updated 7 months ago
Alternatives and similar repositories for neural-graphs:
Users that are interested in neural-graphs are comparing it to the libraries listed below
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- NF-Layers for constructing neural functionals.☆82Updated last year
- Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]☆86Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- ☆9Updated 2 years ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆29Updated 11 months ago
- Official Pytorch implementation of NeuralWalker☆28Updated 9 months ago
- ☆24Updated 5 months ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 6 months ago
- ☆24Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- Uncertainty Quantification over Graph with Conformalized Graph Neural Networks (NeurIPS 2023)☆75Updated last year
- Code repository for Trajectory Flow Matching☆56Updated 4 months ago
- How to Turn Your Knowledge Graph Embeddings into Generative Models☆50Updated 8 months ago
- ☆51Updated 9 months ago
- List of papers on NeurIPS2023☆89Updated last year
- The Energy Transformer block, in JAX☆56Updated last year
- Transformer-based Spectral Graph Neural Networks☆79Updated 5 months ago
- ☆32Updated 2 years ago
- Neural Diffusion Processes☆76Updated 7 months ago
- ☆49Updated last year
- NeurIPS23 "Flow Factorized Representation Learning"☆35Updated 4 months ago
- ☆24Updated 11 months ago
- A library for subgraph GNN based on pyg☆41Updated 3 months ago
- ☆43Updated 2 weeks ago
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learning☆14Updated 9 months ago
- ☆12Updated last year
- ☆116Updated last year
- ☆60Updated 3 years ago
- Neural Optimal Transport with Lagrangian Costs☆53Updated 7 months ago