Blocks (caldera.blocks)

Network building blocks for creating graph neural networks.

Generic Blocks

Flex(module_type)	Flex.
MLP(*latent_sizes[, layer_norm, dropout, …])	A multilayer perceptron.

Encoder/Decoder Blocks

NodeBlock(mlp)	param mlp
EdgeBlock(mlp)	param mlp
GlobalBlock(mlp)

Message Passing Blocks

AggregatingNodeBlock(mlp, edge_aggregator)	param mlp
AggregatingEdgeBlock(mlp)	param mlp
AggregatingGlobalBlock(mlp[, …])
Aggregator(aggregator[, dim, dim_size])	Aggregation layer.
MultiAggregator(input_size, aggregators[, …])	A differentiable and trainable way to select the aggregation function.

“””