equivariant-neural-networks

There are 6 repositories under equivariant-neural-networks topic.

  • atomicarchitects/equiformer_v2

    [ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations

    Language:Python23652033

  • atomicarchitects/equiformer

    [ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs

    Language:Python22461842
  • MorphoSymm

    Danfoa/MorphoSymm

    Tools for exploiting Morphological Symmetries in robotics

    Language:Python532114

  • arnab39/equiadapt

    Library to make any existing neural network architecture equivariant

    Language:Jupyter Notebook51534

  • smsharma/eqnn-jax

    Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.

    Language:Python24533

  • VITA-Group/TAPE

    [Preprint] "Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding" by Jiajun Zhu, Peihao Wang, Ruisi Cai, Jason D. Lee, Pan Li, Zhangyang Wang

    Language:Python80