Add constant revisit gate training option

Optimize DeMemWM memory retrieval and remove diagnostics

Optimize DeMemWM memory retrieval and remove diagnostics

fix wandb project

Clean DeMemWM deterministic memory slot handling

Update Berzelius training script overrides

Merge remote-tracking branch 'hf/main'

initial commit

Initial commit