Option Maps

Neural Network Components

ACT_MAP

Mapping of activation names to activation classes.

Key	Value
elu	ELU
gelu	GELU
leakyrelu	LeakyReLU
mish	Mish
none	Identity
prelu	PReLU
relu	ReLU
selu	SELU
sigmoid	Sigmoid
silu	SiLU
softplus	Softplus
swish	SiLU
tanh	Tanh

GCL_MAP

Mapping of graph convolutional layer names to classes.

Key	Value
cheb	ChebConv
gat	GATConv
gcn	GCNConv
gcnv	GraphConv
sage	SAGEConv

INIT_MAP_W

Mapping of weight initialization names to functions.

Key	Value
kaiming_normal	kaiming_normal_
kaiming_uniform	kaiming_uniform_
normal	normal_
orthogonal	orthogonal_
trunc_normal	trunc_normal_
uniform	uniform_
xavier_normal	xavier_normal_
xavier_uniform	xavier_uniform_

INIT_MAP_B

Mapping of bias initialization names to functions.

Key	Value
ones	ones_
zeros	zeros_

Neural Network Modules

NN_MAP

Options for preset neural network models.

Key	Value
gnn_1st	TakeFirstGraph
gnn_cat	IdenCatGNN
gnn_res	ResBlockGNN
gnn_smp	GNN
mlp_1st	TakeFirst
mlp_cat	IdenCatMLP
mlp_res	ResBlockMLP
mlp_smp	MLP
seq_rnn	VanillaRNN
seq_std	SimpleRNN

AE_MAP

Options for preset autoencoder types.

Key	Value
gnn_cat	(‘gnn_cat’, ‘gnn_1st’)
gnn_res	(‘gnn_res’, ‘gnn_res’)
gnn_seq_rnn	(‘seq_rnn’, ‘seq_gnn_smp’)
gnn_seq_smp	(‘seq_gnn_smp’, ‘seq_gnn_smp’)
gnn_seq_std	(‘seq_std’, ‘seq_gnn_smp’)
gnn_smp	(‘gnn_smp’, ‘gnn_smp’)
mlp_cat	(‘mlp_cat’, ‘mlp_1st’)
mlp_res	(‘mlp_res’, ‘mlp_res’)
mlp_seq_rnn	(‘seq_rnn’, ‘seq_mlp_smp’)
mlp_seq_smp	(‘seq_mlp_smp’, ‘seq_mlp_smp’)
mlp_seq_std	(‘seq_std’, ‘seq_mlp_smp’)
mlp_smp	(‘mlp_smp’, ‘mlp_smp’)

Model Components

ENC_MAP

Mapping of encoder names to encoder functions.

Key	Value
graph_auto	enc_graph_auto
graph_ctrl	enc_graph_ctrl
graph_iden	enc_graph_iden
iden	enc_iden
node_auto	enc_node_auto
node_ctrl	enc_node_ctrl
node_iden	enc_iden
node_raw_auto	enc_node_auto
node_raw_ctrl	enc_node_raw_ctrl
raw_auto	enc_smpl_auto
raw_ctrl	enc_raw_ctrl
smpl_auto	enc_smpl_auto
smpl_ctrl	enc_smpl_ctrl

DEC_MAP

Mapping of decoder names to decoder functions.

Key	Value
auto	dec_auto
graph	dec_graph
iden	dec_iden
node	dec_node

FZU_MAP

Mapping of feature concatenation names to functions.

Key	Value
blin_no_const	zu_blin_no_const
blin_with_const	zu_blin_with_const
cat	zu_cat_smpl
graph_blin_no_const	zu_blin_no_const_graph
graph_blin_with_const	zu_blin_with_const_graph
graph_cat	zu_cat_smpl_graph
none	zu_cat_none

DYN_MAP

Mapping of dynamics composer names to functions.

Key	Value
direct	dyn_direct
graph_direct	dyn_graph_direct
graph_skip	dyn_graph_skip
skip	dyn_skip

Training Utilities

LOSS_MAP

Mapping of loss names to loss classes.

Key	Value
mae	L1Loss
mse	MSELoss
vpt	VPTLoss
wmse	WMSELoss

LIN_MAP

Mapping of linear evaluation and features functions.

Key	Value
graph	linear_eval_graph, linear_features_graph
smpl	linear_eval_smpl, linear_features_smpl

SOL_MAP

Mapping of linear solver methods

Key	Value
ct_full	_ct_full_der
ct_full_log	_ct_full_log
ct_raw	_ct_raw
ct_sako_log	_ct_sako_log
ct_truncated	_ct_truncated_der
ct_truncated_log	_ct_truncated_log
dt_full	_dt_full
dt_raw	_dt_raw
dt_sako	_dt_sako
dt_truncated	_dt_truncated

Data Manipulation Utilities

CTRL_MAP

Mapping of control sampler names to functions.

Key	Value
chirp	chirp_control
gaussian	gaussian_control
sine	sine_control
sphere	sphere_control

X0_MAP

Mapping of initial condition sampler names to functions.

Key	Value
gaussian	gaussian_x0
grid	grid_x0
perturb	perturb_x0
uniform	uniform_x0