Matteo He commited on
Commit
097677b
·
verified ·
1 Parent(s): b3ad680

Add λ-sweep iso-L0 family: 5 λ + 3-seed at λ=1.35 (Pythia-160M W_U d8192)

Browse files
pythia-160m/W_U/lambda-sweep/d8192/lam0p40_seed0.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=0.4), 32 snapshots, d_sae=8192, seed 0",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 0.4,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.7479858952216084,
74
+ "reconstruction_mse": 0.25193381249904634,
75
+ "mean_l0": 160.0768243408203,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 0,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam0p40_seed0.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=0.4), 32 snapshots, d_sae=8192, seed 0
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.7479858952216084
11
+ - Reconstruction MSE: 0.25193381249904634
12
+ - Mean L0: 160.0768243408203
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 0.4
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 0
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam0p40_seed0.safetensors")
38
+ config = json.load(open("lam0p40_seed0.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam0p40_seed0.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29c5a8e6cd17cb39a9c383371362636fc649c5bac693add96e692e621e4f9c4a
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p00_seed0.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.0), 32 snapshots, d_sae=8192, seed 0",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.0,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.6323049070911215,
74
+ "reconstruction_mse": 0.36757794439792635,
75
+ "mean_l0": 58.49865539550781,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 0,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p00_seed0.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.0), 32 snapshots, d_sae=8192, seed 0
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.6323049070911215
11
+ - Reconstruction MSE: 0.36757794439792635
12
+ - Mean L0: 58.49865539550781
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.0
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 0
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p00_seed0.safetensors")
38
+ config = json.load(open("lam1p00_seed0.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p00_seed0.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e60e0f2f23729140050503cfb60496d791808d0764b2907a0a7a3eb6559bd182
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p20_seed0.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.2), 32 snapshots, d_sae=8192, seed 0",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.2,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.6027639222075235,
74
+ "reconstruction_mse": 0.39710951745510104,
75
+ "mean_l0": 45.41338317871094,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 0,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p20_seed0.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.2), 32 snapshots, d_sae=8192, seed 0
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.6027639222075235
11
+ - Reconstruction MSE: 0.39710951745510104
12
+ - Mean L0: 45.41338317871094
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.2
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 0
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p20_seed0.safetensors")
38
+ config = json.load(open("lam1p20_seed0.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p20_seed0.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e03cc5543a16e184f7da8993aabaa67baad9e57ed19f36440b956f3791b2f45f
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed0.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.35), 32 snapshots, d_sae=8192, seed 0",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.35,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.5819729868726617,
74
+ "reconstruction_mse": 0.4178938287496567,
75
+ "mean_l0": 37.971342163085936,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 0,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed0.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.35), 32 snapshots, d_sae=8192, seed 0
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.5819729868726617
11
+ - Reconstruction MSE: 0.4178938287496567
12
+ - Mean L0: 37.971342163085936
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.35
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 0
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p35_seed0.safetensors")
38
+ config = json.load(open("lam1p35_seed0.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed0.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca937c8bc0270d32dbc4af90aba71b50131f4101efdfb546293c61e918f4d73f
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed1.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.35), 32 snapshots, d_sae=8192, seed 1",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.35,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.581652004193779,
74
+ "reconstruction_mse": 0.4182147091627121,
75
+ "mean_l0": 37.79685302734375,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 1,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed1.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.35), 32 snapshots, d_sae=8192, seed 1
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.581652004193779
11
+ - Reconstruction MSE: 0.4182147091627121
12
+ - Mean L0: 37.79685302734375
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.35
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 1
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p35_seed1.safetensors")
38
+ config = json.load(open("lam1p35_seed1.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed1.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:185d4f40c64f8a5956c9aec82ca7cd8b612b1b6d20468067463a08dd172c23fb
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed2.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.35), 32 snapshots, d_sae=8192, seed 2",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.35,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.582366028895355,
74
+ "reconstruction_mse": 0.4175009119510651,
75
+ "mean_l0": 37.980972900390626,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 2,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed2.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.35), 32 snapshots, d_sae=8192, seed 2
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.582366028895355
11
+ - Reconstruction MSE: 0.4175009119510651
12
+ - Mean L0: 37.980972900390626
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.35
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 2
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p35_seed2.safetensors")
38
+ config = json.load(open("lam1p35_seed2.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p35_seed2.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:727983bad97a97e20ff7089f967fc4b0071a45f1d9bf181bae2381f2f63d4e14
3
+ size 1611792824
pythia-160m/W_U/lambda-sweep/d8192/lam1p80_seed0.config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "public_label": "Pythia-160M W_U crosscoder, \u03bb-sweep iso-L0 (\u03bb=1.8), 32 snapshots, d_sae=8192, seed 0",
3
+ "config": {
4
+ "sae_type": "crosscoder",
5
+ "d_model": 768,
6
+ "expansion_factor": 10.6667,
7
+ "use_decoder_bias": true,
8
+ "act_fn": "jumprelu",
9
+ "norm_activation": "inference",
10
+ "sparsity_include_decoder_norm": true,
11
+ "top_k": 50,
12
+ "use_triton_kernel": false,
13
+ "sparsity_threshold_for_triton_spmm_kernel": 0.996,
14
+ "jumprelu_threshold_window": 4.0,
15
+ "hook_points": [
16
+ "wu_step_0",
17
+ "wu_step_1",
18
+ "wu_step_2",
19
+ "wu_step_3",
20
+ "wu_step_4",
21
+ "wu_step_5",
22
+ "wu_step_6",
23
+ "wu_step_7",
24
+ "wu_step_8",
25
+ "wu_step_9",
26
+ "wu_step_10",
27
+ "wu_step_11",
28
+ "wu_step_12",
29
+ "wu_step_13",
30
+ "wu_step_14",
31
+ "wu_step_15",
32
+ "wu_step_16",
33
+ "wu_step_17",
34
+ "wu_step_18",
35
+ "wu_step_19",
36
+ "wu_step_20",
37
+ "wu_step_21",
38
+ "wu_step_22",
39
+ "wu_step_23",
40
+ "wu_step_24",
41
+ "wu_step_25",
42
+ "wu_step_26",
43
+ "wu_step_27",
44
+ "wu_step_28",
45
+ "wu_step_29",
46
+ "wu_step_30",
47
+ "wu_step_31"
48
+ ]
49
+ },
50
+ "training": {
51
+ "lr": 5e-05,
52
+ "l1_coefficient": 1.8,
53
+ "frequency_scale": 0.01,
54
+ "tanh_stretch_coefficient": 1.0,
55
+ "jumprelu_lr_factor": 0.1,
56
+ "init_threshold": 0.1,
57
+ "decoder_transpose_init": 1.0,
58
+ "amp_dtype": "fp32",
59
+ "optimizer": "adam",
60
+ "n_epochs": 300,
61
+ "batch_size": 1024,
62
+ "expansion_factor": 10.6667,
63
+ "d_sae": 8192,
64
+ "input_preprocess": "center_scale",
65
+ "l1_warmup_fraction": 0.1,
66
+ "lr_warmup_fraction": 0.0,
67
+ "lr_decay_fraction": 0.0,
68
+ "auxk_coefficient": 0.03125,
69
+ "auxk_k": 512,
70
+ "dead_window_steps": 100
71
+ },
72
+ "quality": {
73
+ "explained_variance": 0.527703732469858,
74
+ "reconstruction_mse": 0.4721457928419113,
75
+ "mean_l0": 23.033436279296875,
76
+ "dead_rate": 0.0,
77
+ "d_sae": 8192,
78
+ "n_snapshots": 32,
79
+ "n_rows": 50304
80
+ },
81
+ "steps": [
82
+ 0,
83
+ 1,
84
+ 2,
85
+ 4,
86
+ 8,
87
+ 16,
88
+ 32,
89
+ 64,
90
+ 128,
91
+ 256,
92
+ 512,
93
+ 1000,
94
+ 2000,
95
+ 3000,
96
+ 4000,
97
+ 5000,
98
+ 6000,
99
+ 7000,
100
+ 8000,
101
+ 9000,
102
+ 14000,
103
+ 21000,
104
+ 27000,
105
+ 34000,
106
+ 47000,
107
+ 61000,
108
+ 75000,
109
+ 89000,
110
+ 102000,
111
+ 116000,
112
+ 130000,
113
+ 143000
114
+ ],
115
+ "model_name": "EleutherAI/pythia-160m",
116
+ "seed": 0,
117
+ "kind": "cross-snapshot",
118
+ "preprocess_mode": "center_scale"
119
+ }
pythia-160m/W_U/lambda-sweep/d8192/lam1p80_seed0.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Pythia-160M W_U crosscoder, λ-sweep iso-L0 (λ=1.8), 32 snapshots, d_sae=8192, seed 0
2
+
3
+ Companion artifact to *Parameter-trajectory crosscoders for vocabulary readout
4
+ evolution* (NeurIPS 2026). Member of the **λ-sweep iso-L0 family**: a 7-point
5
+ sparsity sweep at d_sae=8192, sharing the released training recipe except for
6
+ the L1 coefficient (and adding `auxk_coefficient=0.03125` to prevent dead-feature
7
+ collapse at high λ). Three replicate seeds at λ=1.35 give a variance estimate.
8
+
9
+ ## Quality
10
+ - Explained variance: 0.527703732469858
11
+ - Reconstruction MSE: 0.4721457928419113
12
+ - Mean L0: 23.033436279296875
13
+ - Dead-feature rate: 0.0
14
+ - d_sae: 8192
15
+ - n_snapshots: 32
16
+
17
+ ## Training recipe (key hparams)
18
+ - Activation: jumprelu
19
+ - L1 coefficient: 1.8
20
+ - AuxK coefficient: 0.03125
21
+ - AuxK k: 512
22
+ - Frequency penalty scale: 0.01
23
+ - Tanh-stretch coefficient: 1.0
24
+ - LR: 5e-05
25
+ - JumpReLU LR multiplier: 0.1
26
+ - Init threshold: 0.1
27
+ - Optimizer: adam
28
+ - Epochs: 300
29
+ - Batch size: 1024
30
+ - Preprocess: center_scale
31
+ - Seed: 0
32
+
33
+ ## Loading
34
+ ```python
35
+ from safetensors.torch import load_file
36
+ import json
37
+ weights = load_file("lam1p80_seed0.safetensors")
38
+ config = json.load(open("lam1p80_seed0.config.json"))
39
+ ```
pythia-160m/W_U/lambda-sweep/d8192/lam1p80_seed0.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0dba1e20536dad7105f47bf0afe97c6b908caba3bf8ba4a2ddd1fd49a4471e2c
3
+ size 1611792824