LICENSE
README.md
pyproject.toml
src/jraphx/__init__.py
src/jraphx/py.typed
src/jraphx.egg-info/PKG-INFO
src/jraphx.egg-info/SOURCES.txt
src/jraphx.egg-info/dependency_links.txt
src/jraphx.egg-info/requires.txt
src/jraphx.egg-info/top_level.txt
src/jraphx/data/__init__.py
src/jraphx/data/batch.py
src/jraphx/data/data.py
src/jraphx/nn/__init__.py
src/jraphx/nn/conv/__init__.py
src/jraphx/nn/conv/edge_conv.py
src/jraphx/nn/conv/gat_conv.py
src/jraphx/nn/conv/gatv2_conv.py
src/jraphx/nn/conv/gcn_conv.py
src/jraphx/nn/conv/gin_conv.py
src/jraphx/nn/conv/message_passing.py
src/jraphx/nn/conv/sage_conv.py
src/jraphx/nn/conv/transformer_conv.py
src/jraphx/nn/models/__init__.py
src/jraphx/nn/models/basic_gnn.py
src/jraphx/nn/models/gnn.py
src/jraphx/nn/models/jumping_knowledge.py
src/jraphx/nn/models/mlp.py
src/jraphx/nn/norm/__init__.py
src/jraphx/nn/norm/batch_norm.py
src/jraphx/nn/norm/graph_norm.py
src/jraphx/nn/norm/layer_norm.py
src/jraphx/nn/pool/__init__.py
src/jraphx/nn/pool/glob.py
src/jraphx/nn/pool/topk_pool.py
src/jraphx/utils/__init__.py
src/jraphx/utils/coalesce.py
src/jraphx/utils/convert.py
src/jraphx/utils/degree.py
src/jraphx/utils/loop.py
src/jraphx/utils/num_nodes.py
src/jraphx/utils/scatter.py
src/jraphx/utils/scatter_softmax.py