[NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch
☆30Apr 13, 2024Updated 2 years ago
Alternatives and similar repositories for lps
Users that are interested in lps are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Learning energy decompositions for partial inference in GFlowNets☆16Jun 4, 2024Updated 2 years ago
- A Simple and Scalable Representation for Graph Generation (ICLR 2024)☆21Mar 19, 2024Updated 2 years ago
- Official Code Repository for the paper "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆16Jul 21, 2024Updated 2 years ago
- ☆26Apr 3, 2024Updated 2 years ago
- CEVAE with VampPrior☆11Jul 18, 2018Updated 8 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- [ICLR 2023] Equivariant Descriptor Fields: SE(3)-Equivariant Energy-Based Models for End-to-End Visual Robotic Manipulation Learning☆41Dec 5, 2023Updated 2 years ago
- CEVAE(Causal Effect Variational AutoEncoder) written with pytorch and pyro.☆10Feb 15, 2021Updated 5 years ago
- A library for programmatically generating equivariant layers through constraint solving☆287May 8, 2023Updated 3 years ago
- ☆18Mar 19, 2024Updated 2 years ago
- Official codebase for our NeurIPS paper, Symmetry-Informed Governing Equation Discovery.☆11Nov 13, 2024Updated last year
- ☆16Oct 3, 2022Updated 3 years ago
- Official JAX implementation of Riemannian Mean Flow for flow map learning on manifolds.☆22Jul 2, 2026Updated 3 weeks ago
- [NeurIPS'21] This repository is the implementation of "Learning 3D Dense Correspondence via Canonical Point Autoencoder"☆20Feb 16, 2022Updated 4 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
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Gaussain plane-wave neural operator (GPWNO) is a novel approach to predict the electron density of molecule, combining two types of the b…☆28Sep 19, 2024Updated last year
- [ICLR 2024 Oral] Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness.☆17Jan 19, 2024Updated 2 years ago
- ☆15Jul 30, 2025Updated 11 months ago
- ☆39May 23, 2025Updated last year
- Analysing ML conference data and plotting interesting statistics.☆11Aug 4, 2023Updated 2 years ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- [ECCV'22] Equivariant Hypergraph Neural Networks, in PyTorch☆32Apr 11, 2023Updated 3 years ago
- [CVPR 2026] Affostruction: 3D Affordance Grounding with Generative Reconstruction☆17May 14, 2026Updated 2 months ago
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Aug 1, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A Python package to learn the Koopman operator.☆82Jun 25, 2026Updated 3 weeks ago
- Data Science, Time Complexity and Inferential Uncertainty (TCIU)☆11Feb 27, 2023Updated 3 years ago
- [GSI 2023] Learning Lagrangian Fluid Mechanics with E(3)-Equivariant GNNs☆15Jun 3, 2024Updated 2 years ago
- [CVPR 2024 Highlight] Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation☆61Apr 5, 2024Updated 2 years ago
- ☆35Feb 12, 2025Updated last year
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆13Aug 5, 2022Updated 3 years ago
- Spatial Spectral Machine Learning☆14Oct 15, 2025Updated 9 months ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆76Apr 21, 2026Updated 3 months ago
- Stainless neural networks in JAX☆36Jun 13, 2026Updated last month
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆47Oct 17, 2025Updated 9 months ago
- [NeurIPS'21] Higher-order Transformers for sets, graphs, and hypergraphs, in PyTorch☆67Nov 28, 2022Updated 3 years ago
- Data and code required to reach the main conclusions of the fastsmcg paper☆11Sep 19, 2023Updated 2 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
- 🐼PANDA: Expanded Width-Aware Message Passing Beyond Rewiring, ICML 2024☆23Sep 14, 2024Updated last year
- PyTorch implementation for our paper "Improving GFlowNets for Text-to-Image Diffusion Alignment."☆31Sep 6, 2024Updated last year
- Official Pytorch Implementation of Relational Self-Attention, NeurIPS 2021☆49Dec 7, 2021Updated 4 years ago