Hugging Face's logo

Axym-Labs
/
AxoSim

PyTorch
neuroscience
neuronal-surrogate
state-space-model
time-series
Model card Files
xet
 Community
AxoSim
File size: 1,416 Bytes
4a5f6e9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
{
  "num_input": 1278,
  "num_output": 2,
  "num_branch": 118,
  "num_synapse_per_branch": 11,
  "input_to_synapse_routing": "neuronio_routing",
  "model_dim": 58,
  "num_layers": 2,
  "block_repeats": 1,
  "state_dim": 2,
  "conv_kernel": 4,
  "expansion": 2,
  "block_type": "mamba1",
  "mamba_projection_rank": 16,
  "dropout": 0.0,
  "residual_scale_init": 0.1,
  "synapse_gain_scale": null,
  "morphology_synapse_gain_scale": 1.0,
  "morphology_synapse_feature_dim": 9,
  "morphology_synapse_feature_hidden": 16,
  "morphology_synapse_feature_scale": 1.0,
  "branch_gain": false,
  "branch_nonlinearity": "tanh",
  "branch_nonlinearity_scale_init": 0.1,
  "branch_bilinear_rank": 0,
  "branch_bilinear_scale": null,
  "separate_heads": true,
  "soma_multiplicative_gate": false,
  "soma_multiplicative_scale_init": 0.1,
  "spike_voltage_coupling_scale": 4.0,
  "integrative_mixer": false,
  "integrative_mixer_short_kernel": 9,
  "integrative_mixer_long_kernel": 65,
  "integrative_mixer_position": "pre",
  "integrative_mixer_long_path_bias": 0.5,
  "local_tcn_scale": 1.0,
  "local_tcn_mode": "dilated",
  "local_tcn_position": "post",
  "share_local_tcn_pointwise": true,
  "local_tcn_pointwise_adapter_rank": 2,
  "morphology_ids": [
    "specimen_479770916",
    "specimen_487664663",
    "specimen_488679042",
    "specimen_488683425",
    "specimen_589427435"
  ],
  "morphology_embedding_scale": 0.15
}