lucidrains / gotennet-pytorchView external linksLinks
Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
☆67Apr 7, 2025Updated 10 months ago
Alternatives and similar repositories for gotennet-pytorch
Users that are interested in gotennet-pytorch are comparing it to the libraries listed below
Sorting:
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Nov 9, 2024Updated last year
- Official implementation of HEGNN, a novel high-degree equivariant graph neural network proposed in the NeurIPS 2024 paper 'Are High-Degre…☆31Nov 8, 2024Updated last year
- Exploration into the Firefly algorithm in Pytorch☆41Feb 14, 2025Updated last year
- ☆11Oct 15, 2023Updated 2 years ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆60Jan 7, 2026Updated last month
- Axial Positional Embedding for Pytorch☆84Feb 25, 2025Updated 11 months ago
- [NeurIPS 2022] The implementation for the paper "Equivariant Graph Hierarchy-Based Neural Networks".☆32Oct 5, 2023Updated 2 years ago
- ☆34Jun 29, 2024Updated last year
- ☆34Feb 12, 2025Updated last year
- An implementation of the Equivariant Graph Neural Network (EGNN) layer type for DGL-PyTorch.☆15Dec 27, 2022Updated 3 years ago
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 4 months ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆67Dec 16, 2025Updated 2 months ago
- A language model suite for numbering antigen receptor sequences.☆37Feb 9, 2026Updated last week
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆18Nov 16, 2025Updated 3 months ago
- ROSA+: RWKV's ROSA implementation with fallback statistical predictor☆32Oct 13, 2025Updated 4 months ago
- ☆36Apr 25, 2022Updated 3 years ago
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆47Jul 16, 2023Updated 2 years ago
- Implementation of the proposed MaskBit from Bytedance AI☆83Nov 12, 2024Updated last year
- [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs☆269Feb 11, 2025Updated last year
- Repository containing the code used for running the experiments of the Poincare ResNet paper☆29Aug 25, 2023Updated 2 years ago
- Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch☆55Nov 25, 2024Updated last year
- MESS: Modern Electronic Structure Simulations☆20Sep 24, 2024Updated last year
- ☆23Dec 14, 2023Updated 2 years ago
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated 7 months ago
- Implementation of a framework for Genie2 in Pytorch☆156Jan 7, 2025Updated last year
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Jun 2, 2024Updated last year
- [ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations☆322Feb 11, 2025Updated last year
- This is the official implementation of the paper: Fractional Denoising for 3D Molecular Pre-training☆22Mar 30, 2024Updated last year
- Code, results and other artifacts from the paper introducing the WildChat-50m dataset and the Re-Wild model family.☆34Apr 1, 2025Updated 10 months ago
- OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.☆127Updated this week
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆101Aug 1, 2023Updated 2 years ago
- Implementation of FlowSite and HarmonicFlow from the paper "Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding …☆104Jul 30, 2024Updated last year
- A library for subgraph GNN based on pyg☆39Nov 28, 2024Updated last year
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆28Feb 24, 2025Updated 11 months ago
- Geometric Generative Model Tutorial at LoG 2024☆27Nov 27, 2024Updated last year
- Massively-Parallel Natural Extension of Reference Frame☆33Jan 18, 2023Updated 3 years ago
- Implementation for ICLR2024 Oral paper "Unified Generative Modeling of 3D Molecules with Bayesian Flow Networks"☆59Jun 4, 2024Updated last year
- List of Geometric GNNs for 3D atomic systems☆119Feb 29, 2024Updated last year
- Mirror Diffusion Models, NeurIPS 2023☆33Dec 1, 2023Updated 2 years ago