Merge of analysis tools feature branch to main#93

Merged

rxng8 merged 65 commits intomainfrom

Mar 13, 2025

Commits on Jul 26, 2024

Commits on Jul 27, 2024

fixed minor load bug in lif-cell
ago109
committed
fixed a blocking jit-partial call in lif update_theta method; when loading
Alexander Ororbia
committed

Commits on Jul 28, 2024

minor edit to dim-reduce
Alexander Ororbia
committed

Commits on Aug 7, 2024

Commits on Aug 8, 2024

Commits on Aug 9, 2024

Commits on Aug 11, 2024

minor tweak to dim-reduce in utils
Alexander Ororbia
committed

Commits on Nov 19, 2024

Additions for inhibition stuff
willgebhardt
committed

Commits on Mar 1, 2025

Commits on Mar 3, 2025

minor tweak to attentive prob code comments
Alexander Ororbia
committed
cleaned up probe parent fit routine
Alexander Ororbia
committed
cleaned up probe parent fit routine
Alexander Ororbia
committed
cleaned up probe parent fit routine
Alexander Ororbia
committed
cleaned up probe parent fit routine
Alexander Ororbia
committed

Commits on Mar 5, 2025

minor edits to attn probe
Alexander Ororbia
committed
update attentive probe with input layer norm
rxng8
committed

Commits on Mar 6, 2025

update input layer normalization
rxng8
committed
update code to fix nan bug
rxng8
committed
minor tweak to attn probe
Alexander Ororbia
committed
cleaned up probes
Alexander Ororbia
committed
cleaned up probes
Alexander Ororbia
committed
cleaned up probes
Alexander Ororbia
committed
cleaned up probes
Alexander Ororbia
committed
generalized dropout in terms of shape
Alexander Ororbia
committed
tweak to atten probe
Alexander Ororbia
committed
tweak to atten probe
Alexander Ororbia
committed
added silu/swish/elu to model_utils
Alexander Ororbia
committed
cleaned up model_utils
Alexander Ororbia
committed

Commits on Mar 7, 2025

fix bug in attention probe dropout, fix bug in None noise_key passed in the probing jit function, add the spliting of noise_keys to two dropout in two cross attention
rxng8
committed

Commits on Mar 10, 2025

hyperparameter tunning arguments added
rxng8
committed

Commits on Mar 11, 2025

Merge branch 'main' into analysis_tools
rxng8
committed

Commits on Mar 12, 2025

Commits on Mar 13, 2025