/align: functions for alignment
/config: dataset and hyper-parameter configs
/model: transformer-based conditional feature extractor
/flow: functions for flow matching
/layers: functions for transformers
/experiment: functions for experiments
/utils: functions for attention mechnism
/preprocess & /xds: data preprocessing
test_neuronal_dynamics_flow_pretrain.py: main function during pre-training phases
test_neuronal_dynamics_flow_finetune.py: main function during fine-tuning phases