[NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch
☆30Apr 13, 2024Updated last year
Alternatives and similar repositories for lps
Users that are interested in lps are comparing it to the libraries listed below
Sorting:
- Official Code Repository for the paper "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆15Jul 21, 2024Updated last year
- Learning energy decompositions for partial inference in GFlowNets☆16Jun 4, 2024Updated last year
- A Simple and Scalable Representation for Graph Generation (ICLR 2024)☆22Mar 19, 2024Updated last year
- ☆25Apr 3, 2024Updated last year
- ☆11Aug 30, 2024Updated last year
- Long Range Graph Benchmark, NeurIPS 2022 Track on D&B☆164Nov 28, 2023Updated 2 years ago
- Official codebase for our NeurIPS paper, Symmetry-Informed Governing Equation Discovery.☆11Nov 13, 2024Updated last year
- [ECCV'22] Equivariant Hypergraph Neural Networks, in PyTorch☆31Apr 11, 2023Updated 2 years ago
- Extract features and bounding boxes using the original Bottom-up Attention Faster-RCNN in a few lines of Python code☆11Sep 18, 2022Updated 3 years ago
- Codes for the paper "CausalCite: A Causal Formulation of Paper Citations" (2023)☆16Jan 11, 2024Updated 2 years ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- [ICLR'25 Spotlight] Revisiting Random Walks for Learning on Graphs (RWNN), in PyTorch☆17Mar 4, 2025Updated last year
- ☆36Jun 14, 2025Updated 8 months ago
- Graph generation with K2-trees (ICLR 2024)☆12Mar 19, 2024Updated last year
- Official code repository for the papers "Anti-Symmetric DGN: a stable architecture for Deep Graph Networks" accepted at ICLR 2023; "Non-D…☆15Jan 2, 2025Updated last year
- [ICLR 2023] Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learning☆38Dec 5, 2023Updated 2 years ago
- ☆38May 23, 2025Updated 9 months ago
- Implementation of papers in 101 lines of code.☆18Nov 12, 2023Updated 2 years ago
- Code for the paper- Multi-conditioned Graph Diffusion for Neural Architecture Search (https://openreview.net/forum?id=5VotySkajV)☆16Sep 7, 2024Updated last year
- Neural SPH☆41Nov 23, 2025Updated 3 months ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆41Mar 1, 2025Updated last year
- [ICML 2024] How Interpretable Are Interpretable Graph Neural Networks?☆15Jun 27, 2024Updated last year
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Aug 1, 2025Updated 7 months ago
- ☆45Oct 17, 2025Updated 4 months ago
- Spatial Spectral Machine Learning☆14Oct 15, 2025Updated 4 months ago
- ☆17Mar 19, 2024Updated last year
- [ICLR 2024 Oral] Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness.☆17Jan 19, 2024Updated 2 years ago
- A library for programmatically generating equivariant layers through constraint solving☆281May 8, 2023Updated 2 years ago
- [NeurIPS'21] Higher-order Transformers for sets, graphs, and hypergraphs, in PyTorch☆67Nov 28, 2022Updated 3 years ago
- [EMNLP 2024] Official implementation of "Hierarchical Deconstruction of LLM Reasoning: A Graph-Based Framework for Analyzing Knowledge Ut…☆23Dec 4, 2024Updated last year
- PyTorch implementation of Pseudo-Riemannian Graph Convolutional Networks (NeurIPS'22))☆17Jul 5, 2024Updated last year
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆68Jan 1, 2025Updated last year
- [ICLR 2024 Spotlight] Official Implementation of "Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Product…☆69Nov 2, 2024Updated last year
- FleVRS: Towards Flexible Visual Relationship Segmentation, NeurIPS 2024☆22Dec 9, 2024Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆82Jul 23, 2024Updated last year
- This is an official implementation for "GRIT: Graph Inductive Biases in Transformers without Message Passing".☆127Aug 20, 2025Updated 6 months ago
- ☆47Jan 16, 2024Updated 2 years ago
- 🐼PANDA: Expanded Width-Aware Message Passing Beyond Rewiring, ICML 2024☆22Sep 14, 2024Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆111Oct 11, 2025Updated 4 months ago