| features.embeddings.linear | |
| features.edge_embedding.weight | |
| features.norm_edges | |
| W_e | |
| W_s.weight | |
| encoder_layers.0.norm1 | |
| encoder_layers.0.norm2 | |
| encoder_layers.0.norm3 | |
| encoder_layers.0.W1 | |
| encoder_layers.0.W2 | |
| encoder_layers.0.W3 | |
| encoder_layers.0.W11 | |
| encoder_layers.0.W12 | |
| encoder_layers.0.W13 | |
| encoder_layers.0.dense.W_in | |
| encoder_layers.0.dense.W_out | |
| encoder_layers.1.norm1 | |
| encoder_layers.1.norm2 | |
| encoder_layers.1.norm3 | |
| encoder_layers.1.W1 | |
| encoder_layers.1.W2 | |
| encoder_layers.1.W3 | |
| encoder_layers.1.W11 | |
| encoder_layers.1.W12 | |
| encoder_layers.1.W13 | |
| encoder_layers.1.dense.W_in | |
| encoder_layers.1.dense.W_out | |
| encoder_layers.2.norm1 | |
| encoder_layers.2.norm2 | |
| encoder_layers.2.norm3 | |
| encoder_layers.2.W1 | |
| encoder_layers.2.W2 | |
| encoder_layers.2.W3 | |
| encoder_layers.2.W11 | |
| encoder_layers.2.W12 | |
| encoder_layers.2.W13 | |
| encoder_layers.2.dense.W_in | |
| encoder_layers.2.dense.W_out | |
| decoder_layers.0.norm1 | |
| decoder_layers.0.norm2 | |
| decoder_layers.0.W1 | |
| decoder_layers.0.W2 | |
| decoder_layers.0.W3 | |
| decoder_layers.0.dense.W_in | |
| decoder_layers.0.dense.W_out | |
| decoder_layers.1.norm1 | |
| decoder_layers.1.norm2 | |
| decoder_layers.1.W1 | |
| decoder_layers.1.W2 | |
| decoder_layers.1.W3 | |
| decoder_layers.1.dense.W_in | |
| decoder_layers.1.dense.W_out | |
| decoder_layers.2.norm1 | |
| decoder_layers.2.norm2 | |
| decoder_layers.2.W1 | |
| decoder_layers.2.W2 | |
| decoder_layers.2.W3 | |
| decoder_layers.2.dense.W_in | |
| decoder_layers.2.dense.W_out | |
| W_out |