id: 1
unit: export default function TokenAblationmap()
file: sae-viewer/src/components/tokenAblationmap.tsx
start line: 27
end line: 110
size: 82 LOC
McCabe index: 20
number of parameters: 3

id: 2
unit: def forward()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 70 LOC
McCabe index: 5
number of parameters: 2

id: 3
unit: export default()
file: sae-viewer/src/components/featureInfo.tsx
start line: 10
end line: 105
size: 58 LOC
McCabe index: 6
number of parameters: 1

id: 4
unit: def main()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 56 LOC
McCabe index: 2
number of parameters: 0

id: 5
unit: export default function FeatureSelect()
file: sae-viewer/src/components/featureSelect.tsx
start line: 15
end line: 55
size: 37 LOC
McCabe index: 3
number of parameters: 4

id: 6
unit: def make_torch_comms()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 33 LOC
McCabe index: 6
number of parameters: 2

id: 7
unit: async function fetchData()
file: sae-viewer/src/components/featureInfo.tsx
start line: 20
end line: 49
size: 30 LOC
McCabe index: 5
number of parameters: 0

id: 8
unit: export default function TokenHeatmap()
file: sae-viewer/src/components/tokenHeatmap.tsx
start line: 21
end line: 49
size: 28 LOC
McCabe index: 4
number of parameters: 4

id: 9
unit: export default function Feed()
file: sae-viewer/src/feed.tsx
start line: 11
end line: 40
size: 28 LOC
McCabe index: 2
number of parameters: 0

id: 10
unit: def predict()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 24 LOC
McCabe index: 15
number of parameters: 2

id: 11
unit: def sharded_topk()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 22 LOC
McCabe index: 3
number of parameters: 4

id: 12
unit: def get_minimal_subsequence()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 1

id: 13
unit: def init_broadcast_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 4
number of parameters: 2

id: 14
unit: def all_broadcast()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 20 LOC
McCabe index: 3
number of parameters: 2

id: 15
unit: def triton_sum_dim0_in_fp32()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 1
number of parameters: 1

id: 16
unit: def sharded_grad_norm()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 18 LOC
McCabe index: 7
number of parameters: 3

id: 17
unit: def batch_parallelize()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 7
number of parameters: 3

id: 18
unit: def init_from_data_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 17 LOC
McCabe index: 1
number of parameters: 3

id: 19
unit: def triton_dense_dense_sparseout_matmul_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 16 LOC
McCabe index: 1
number of parameters: 0

id: 20
unit: def triton_add_mul_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 15 LOC
McCabe index: 1
number of parameters: 0

id: 21
unit: def forward()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 1
number of parameters: 3

id: 22
unit: def sh_allreduce_backward()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 14 LOC
McCabe index: 2
number of parameters: 2

id: 23
unit: def batched()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 5
number of parameters: 2

id: 24
unit: def triton_sparse_dense_matmul_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 1
number of parameters: 0

id: 25
unit: def sh_allreduce_forward()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 13 LOC
McCabe index: 2
number of parameters: 2

id: 26
unit: def triton_sparse_transpose_dense_matmul_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 0

id: 27
unit: def backward()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 12 LOC
McCabe index: 1
number of parameters: 2

id: 28
unit: export async function get_feature_info()
file: sae-viewer/src/interpAPI.ts
start line: 45
end line: 58
size: 11 LOC
McCabe index: 3
number of parameters: 2

id: 29
unit: def normalized_mse()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 3
number of parameters: 2

id: 30
unit: def _sh_comm_op()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 10 LOC
McCabe index: 4
number of parameters: 3

id: 31
unit: export function pathForFeature()
file: sae-viewer/src/autoencoder_registry.tsx
start line: 101
end line: 112
size: 9 LOC
McCabe index: 2
number of parameters: 1

id: 32
unit: def triton_mse_loss_fp16_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 1
number of parameters: 0

id: 33
unit: def dp_allreduce_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 9 LOC
McCabe index: 5
number of parameters: 2

id: 34
unit: function zip_sequence()
file: sae-viewer/src/components/tokenHeatmap.tsx
start line: 12
end line: 19
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 35
unit: function useOutsideClickAlerter()
file: sae-viewer/src/components/tooltip.tsx
start line: 9
end line: 24
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 36
unit: export function zip_sequence()
file: sae-viewer/src/types.ts
start line: 46
end line: 53
size: 8 LOC
McCabe index: 2
number of parameters: 1

id: 37
unit: export function interpolateColor()
file: sae-viewer/src/types.ts
start line: 105
end line: 112
size: 8 LOC
McCabe index: 1
number of parameters: 3

id: 38
unit: def triton_sum_dim0_in_fp32_kernel()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 39
unit: def __init__()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 0

id: 40
unit: def unit_norm_decoder_grad_adjustment_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 1

id: 41
unit: def temporary_weight_swap()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 3
number of parameters: 2

id: 42
unit: def forward()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 8 LOC
McCabe index: 1
number of parameters: 2

id: 43
unit: await fetch()
file: sae-viewer/src/interpAPI.ts
start line: 11
end line: 17
size: 7 LOC
McCabe index: 1
number of parameters: 5

id: 44
unit: await fetch()
file: sae-viewer/src/interpAPI.ts
start line: 24
end line: 30
size: 7 LOC
McCabe index: 1
number of parameters: 6

id: 45
unit: def __init__()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 3
number of parameters: 2

id: 46
unit: def step()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 47
unit: def use_ema_weights()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 7 LOC
McCabe index: 1
number of parameters: 1

id: 48
unit: normalized_activation: scaler()
file: sae-viewer/src/types.ts
start line: 100
end line: 106
size: 6 LOC
McCabe index: 1
number of parameters: 1

id: 49
unit: def triton_sparse_transpose_dense_matmul()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 50
unit: def triton_coo_sparse_dense_matmul()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 0

id: 51
unit: def sh_allreduce_scale()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 4
number of parameters: 2

id: 52
unit: def LN()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 53
unit: def encode_pre_act()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 3

id: 54
unit: def decode()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 55
unit: def state_dict()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 2
number of parameters: 4

id: 56
unit: def forward()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 6 LOC
McCabe index: 1
number of parameters: 2

id: 57
unit: def dumpf()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 58
unit: def apply_batched()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 59
unit: def a_star_c_fn()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 5
number of parameters: 1

id: 60
unit: def triton_add_mul_()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 61
unit: def backward()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 2

id: 62
unit: def triton_add_mul_()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 63
unit: def batch_tensors()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 64
unit: def __init__()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 3

id: 65
unit: def autoencoder_loss()
file: sparse_autoencoder/loss.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 1
number of parameters: 0

id: 66
unit: def preprocess()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 5 LOC
McCabe index: 2
number of parameters: 2

id: 67
unit: def todict()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 68
unit: def triton_sparse_dense_matmul()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 69
unit: def triton_dense_dense_sparseout_matmul()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 0

id: 70
unit: def auxk_mask_fn()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 1

id: 71
unit: def decode_sparse()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 72
unit: def logkv()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 3
number of parameters: 3

id: 73
unit: def dumpkvs()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 1

id: 74
unit: def v1()
file: sparse_autoencoder/paths.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 75
unit: def v4()
file: sparse_autoencoder/paths.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 2

id: 76
unit: def v5_32k()
file: sparse_autoencoder/paths.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 77
unit: def v5_128k()
file: sparse_autoencoder/paths.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 2
number of parameters: 2

id: 78
unit: def __init__()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 3

id: 79
unit: def state_dict()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 80
unit: def from_state_dict()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 4 LOC
McCabe index: 1
number of parameters: 4

id: 81
unit: function handleClickOutside()
file: sae-viewer/src/components/tooltip.tsx
start line: 14
end line: 16
size: 3 LOC
McCabe index: 3
number of parameters: 1

id: 82
unit: def loadf()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 83
unit: def load()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 84
unit: def create_n2g_explanation()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 0

id: 85
unit: def forward()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 86
unit: def mse()
file: sparse_autoencoder/kernels.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 87
unit: def print0()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 2
number of parameters: 2

id: 88
unit: def normalized_mean_squared_error()
file: sparse_autoencoder/loss.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 89
unit: def normalized_L1_loss()
file: sparse_autoencoder/loss.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 0

id: 90
unit: def __init__()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 4

id: 91
unit: def encode()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 92
unit: def __init__()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 93
unit: def forward()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 3 LOC
McCabe index: 1
number of parameters: 2

id: 94
unit: def predict()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 95
unit: def dump()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 96
unit: def load()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 97
unit: def __init__()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 98
unit: def predict_many()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 2

id: 99
unit: def first_position_fn()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 2
number of parameters: 1

id: 100
unit: def c_fn()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 4
number of parameters: 1

id: 101
unit: def zero_fn()
file: sparse_autoencoder/explanations.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 102
unit: def all_reduce()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 103
unit: def all_gather()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 104
unit: def broadcast()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 4

id: 105
unit: def barrier()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 106
unit: def size()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 107
unit: def sh_sum()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 2

id: 108
unit: def replica_shard_to_rank()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 3

id: 109
unit: def n_dirs()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 110
unit: def unit_norm_decoder_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 111
unit: def maybe_transpose()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 3
number of parameters: 1

id: 112
unit: def training_loop_()
file: sparse_autoencoder/train.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 113
unit: def from_state_dict()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 0

id: 114
unit: def weight()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1

id: 115
unit: def bias()
file: sparse_autoencoder/model.py
start line: 0
end line: 0
size: 2 LOC
McCabe index: 1
number of parameters: 1