Code

Add GRIT model#777

Code

Open

pweigel wants to merge 16 commits intographnet-team:maingraphnet-team/graphnet:mainfrom pweigel:gritpweigel/graphnet:gritCopy head branch name to clipboard

+1,255-11

Commits

Commits on Dec 14, 2024

• Add GRIT layers and model

  pweigel

  committed
  b2a8b25

• Move linear encoders to embedding.py

  pweigel

  committed
  c7b8dba

• Add GRIT to __init__.py

  pweigel

  committed
  4d6ad04

• Fix imports, add RRWP utils and graph

  pweigel

  committed
  8a8c9fb

• Cleaning up grit model/layers

  pweigel

  committed
  3d52f70

• Cleaning up GRIT layers

  pweigel

  committed
  4fecd5b

• Remove duplicate pyg_softmax function

  pweigel

  committed
  02d4a0d

• Merge the attention calc into forward, reduce amount of saved data

  pweigel

  committed
  924d165

Commits on Dec 17, 2024

• Significant improvements to naming, docstrings

  pweigel

  committed
  7ccb6f0

• Remove TODO

  pweigel

  committed
  bb232e9

• Fix normalization layers

  pweigel

  committed
  1a2d80c

• Updating new graph/edge definitions

  pweigel

  committed
  5c4b0aa

• Added example training script, bug fixes

  pweigel

  committed
  c861d9d

• Improving SANGraphHead and simplify dims

  pweigel

  committed
  e540b0e

• Merge branch 'graphnet-team:main' into grit

  pweigel

  authored
  1e46b75

• Add newline to make flake8 happy

  pweigel

  committed
  48d5e85