Navigator / src / navigator / neural / architectures / set_transformer