NN# NN Layers Graph Convolutions Norm Layers General Layers Blocks Encoders Decoders Models BaseModel Spatiotemporal Models Temporal Models Base Dense GraphConv TemporalConv2d GatedTemporalConv2d AttentionEncoder MultiHeadAttention CausalLinearAttention Functional expand_then_cat() gated_tanh() reverse_tensor() sparse_softmax() sparse_multi_head_attention()