All modules for which code is available

• jraphx.nn.conv.edge_conv
• jraphx.nn.conv.gat_conv
• jraphx.nn.conv.gatv2_conv
• jraphx.nn.conv.gcn_conv
• jraphx.nn.conv.gin_conv
• jraphx.nn.conv.message_passing
• jraphx.nn.conv.sage_conv
• jraphx.nn.conv.transformer_conv
• jraphx.nn.models.basic_gnn
• jraphx.nn.models.gnn
• jraphx.nn.models.jumping_knowledge
• jraphx.nn.models.mlp
• jraphx.nn.norm.batch_norm
• jraphx.nn.norm.graph_norm
• jraphx.nn.norm.layer_norm
• jraphx.nn.pool.glob
• jraphx.nn.pool.topk_pool
• jraphx.utils.coalesce
• jraphx.utils.convert
• jraphx.utils.degree
• jraphx.utils.loop
• jraphx.utils.scatter