{ "device": "cuda:0", "seed": 42, "dtype": "torch.bfloat16", "hook_point_in": "blocks.26.ln2.hook_normalized", "hook_point_out": "blocks.26.hook_mlp_out", "use_decoder_bias": true, "apply_decoder_bias_to_pre_encoder": false, "expansion_factor": 32, "d_model": 4096, "d_sae": 131072, "bias_init_method": "all_zero", "act_fn": "jumprelu", "jump_relu_threshold": 0.2099609375, "norm_activation": "dataset-wise", "dataset_average_activation_norm": { "in": 64.0, "out": 8.4375 }, "decoder_exactly_fixed_norm": false, "sparsity_include_decoder_norm": true, "use_glu_encoder": false, "init_decoder_norm": 0.5, "init_encoder_norm": 0.5, "init_encoder_with_decoder_transpose": false, "lp": 1, "l1_coefficient": 8e-05, "l1_coefficient_warmup_steps": 78125, "top_k": 50, "k_warmup_steps": 78125, "use_batch_norm_mse": true, "use_ghost_grads": false, "tp_size": 1, "ddp_size": 1 }