├── .gitignore ├── .vscode ├── extensions.json └── settings.json ├── LICENSE ├── README.md ├── app ├── .eslintrc.json ├── .gitignore ├── .prettierrc ├── app │ ├── components │ │ ├── ColorSpan.tsx │ │ ├── Controls.tsx │ │ ├── LatentExamples.tsx │ │ ├── LogitsTable.tsx │ │ └── prompt │ │ │ ├── PromptActivations.tsx │ │ │ ├── PromptLatentHeatmaps.tsx │ │ │ ├── PromptLayerHistograms.tsx │ │ │ ├── PromptLogitsInput.tsx │ │ │ ├── PromptLogitsRecon.tsx │ │ │ └── PromptLogitsSteer.tsx │ ├── favicon.ico │ ├── globals.css │ ├── layout.tsx │ ├── page.tsx │ ├── prompt │ │ └── [prompt] │ │ │ ├── page.client.tsx │ │ │ └── page.tsx │ └── use-select.tsx ├── components.json ├── components │ └── ui │ │ ├── button.tsx │ │ ├── card.tsx │ │ ├── chart.tsx │ │ ├── form.tsx │ │ ├── input.tsx │ │ ├── label.tsx │ │ ├── table.tsx │ │ ├── tabs.tsx │ │ └── textarea.tsx ├── lib │ ├── api.ts │ ├── format.ts │ ├── models.ts │ └── utils.ts ├── next.config.js ├── package-lock.json ├── package.json ├── postcss.config.js ├── tailwind.config.js └── tsconfig.json ├── citation.bib ├── figures.py ├── figures ├── __init__.py ├── embed_sim.py ├── entropy.py ├── heatmap.py ├── heatmap_aggregate.py ├── heatmap_prompt.py ├── layer_hist.py ├── layer_sim.py ├── layer_std.py ├── mmcs.py ├── num_layers.py ├── resid_sim.py ├── scatter_freq.py ├── test.py └── wdec_sim.py ├── layer_dists.py ├── layer_tests.py ├── mlsae ├── __init__.py ├── analysis │ ├── __init__.py │ ├── dists.py │ ├── examples.py │ └── variances.py ├── api │ ├── __init__.py │ ├── __main__.py │ ├── analyser.py │ └── models.py ├── metrics │ ├── __init__.py │ ├── auxiliary_loss.py │ ├── dead_latents.py │ ├── layerwise.py │ ├── layerwise_fvu.py │ ├── layerwise_l0_norm.py │ ├── layerwise_l1_norm.py │ ├── layerwise_logit_kl_div.py │ ├── layerwise_logit_mse.py │ ├── layerwise_loss_delta.py │ ├── layerwise_mse.py │ ├── mse_loss.py │ └── tests │ │ ├── __init__.py │ │ ├── test_dead_latents.py │ │ ├── test_layerwise_fvu.py │ │ ├── test_layerwise_l0_norm.py │ │ ├── test_layerwise_l1_norm.py │ │ ├── test_layerwise_mse.py │ │ └── test_loss_mse.py ├── model │ ├── __init__.py │ ├── autoencoders │ │ ├── __init__.py │ │ ├── standard.py │ │ ├── tests │ │ │ └── test_autoencoders.py │ │ ├── topk.py │ │ └── utils.py │ ├── data.py │ ├── decoder.py │ ├── geom_median.py │ ├── kernels.py │ ├── lightning.py │ ├── transformers │ │ ├── __init__.py │ │ ├── gemma2.py │ │ ├── gpt2.py │ │ ├── llama.py │ │ ├── models │ │ │ ├── gemma2 │ │ │ │ └── modeling_gemma2.py │ │ │ ├── gpt2 │ │ │ │ └── modeling_gpt2.py │ │ │ └── llama │ │ │ │ └── modeling_llama.py │ │ ├── pythia.py │ │ └── tests │ │ │ ├── test_gpt2.py │ │ │ └── test_llama.py │ └── types.py ├── model_card.py ├── trainer │ ├── __init__.py │ ├── config.py │ ├── test.py │ └── train.py └── utils.py ├── pyproject.toml ├── requirements.txt ├── test.py ├── tests.py ├── train.py ├── upload.py └── uv.lock /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/README.md -------------------------------------------------------------------------------- /app/.eslintrc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/.eslintrc.json -------------------------------------------------------------------------------- /app/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/.gitignore -------------------------------------------------------------------------------- /app/.prettierrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/.prettierrc -------------------------------------------------------------------------------- /app/app/components/ColorSpan.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/ColorSpan.tsx -------------------------------------------------------------------------------- /app/app/components/Controls.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/Controls.tsx -------------------------------------------------------------------------------- /app/app/components/LatentExamples.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/LatentExamples.tsx -------------------------------------------------------------------------------- /app/app/components/LogitsTable.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/LogitsTable.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptActivations.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptActivations.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptLatentHeatmaps.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptLatentHeatmaps.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptLayerHistograms.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptLayerHistograms.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptLogitsInput.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptLogitsInput.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptLogitsRecon.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptLogitsRecon.tsx -------------------------------------------------------------------------------- /app/app/components/prompt/PromptLogitsSteer.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/components/prompt/PromptLogitsSteer.tsx -------------------------------------------------------------------------------- /app/app/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/favicon.ico -------------------------------------------------------------------------------- /app/app/globals.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/globals.css -------------------------------------------------------------------------------- /app/app/layout.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/layout.tsx -------------------------------------------------------------------------------- /app/app/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/page.tsx -------------------------------------------------------------------------------- /app/app/prompt/[prompt]/page.client.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/prompt/[prompt]/page.client.tsx -------------------------------------------------------------------------------- /app/app/prompt/[prompt]/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/prompt/[prompt]/page.tsx -------------------------------------------------------------------------------- /app/app/use-select.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/app/use-select.tsx -------------------------------------------------------------------------------- /app/components.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components.json -------------------------------------------------------------------------------- /app/components/ui/button.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/button.tsx -------------------------------------------------------------------------------- /app/components/ui/card.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/card.tsx -------------------------------------------------------------------------------- /app/components/ui/chart.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/chart.tsx -------------------------------------------------------------------------------- /app/components/ui/form.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/form.tsx -------------------------------------------------------------------------------- /app/components/ui/input.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/input.tsx -------------------------------------------------------------------------------- /app/components/ui/label.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/label.tsx -------------------------------------------------------------------------------- /app/components/ui/table.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/table.tsx -------------------------------------------------------------------------------- /app/components/ui/tabs.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/tabs.tsx -------------------------------------------------------------------------------- /app/components/ui/textarea.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/components/ui/textarea.tsx -------------------------------------------------------------------------------- /app/lib/api.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/lib/api.ts -------------------------------------------------------------------------------- /app/lib/format.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/lib/format.ts -------------------------------------------------------------------------------- /app/lib/models.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/lib/models.ts -------------------------------------------------------------------------------- /app/lib/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/lib/utils.ts -------------------------------------------------------------------------------- /app/next.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/next.config.js -------------------------------------------------------------------------------- /app/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/package-lock.json -------------------------------------------------------------------------------- /app/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/package.json -------------------------------------------------------------------------------- /app/postcss.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/postcss.config.js -------------------------------------------------------------------------------- /app/tailwind.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/tailwind.config.js -------------------------------------------------------------------------------- /app/tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/app/tsconfig.json -------------------------------------------------------------------------------- /citation.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/citation.bib -------------------------------------------------------------------------------- /figures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures.py -------------------------------------------------------------------------------- /figures/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /figures/embed_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/embed_sim.py -------------------------------------------------------------------------------- /figures/entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/entropy.py -------------------------------------------------------------------------------- /figures/heatmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/heatmap.py -------------------------------------------------------------------------------- /figures/heatmap_aggregate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/heatmap_aggregate.py -------------------------------------------------------------------------------- /figures/heatmap_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/heatmap_prompt.py -------------------------------------------------------------------------------- /figures/layer_hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/layer_hist.py -------------------------------------------------------------------------------- /figures/layer_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/layer_sim.py -------------------------------------------------------------------------------- /figures/layer_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/layer_std.py -------------------------------------------------------------------------------- /figures/mmcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/mmcs.py -------------------------------------------------------------------------------- /figures/num_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/num_layers.py -------------------------------------------------------------------------------- /figures/resid_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/resid_sim.py -------------------------------------------------------------------------------- /figures/scatter_freq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/scatter_freq.py -------------------------------------------------------------------------------- /figures/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/test.py -------------------------------------------------------------------------------- /figures/wdec_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/figures/wdec_sim.py -------------------------------------------------------------------------------- /layer_dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/layer_dists.py -------------------------------------------------------------------------------- /layer_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/layer_tests.py -------------------------------------------------------------------------------- /mlsae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mlsae/analysis/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mlsae/analysis/dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/analysis/dists.py -------------------------------------------------------------------------------- /mlsae/analysis/examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/analysis/examples.py -------------------------------------------------------------------------------- /mlsae/analysis/variances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/analysis/variances.py -------------------------------------------------------------------------------- /mlsae/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mlsae/api/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/api/__main__.py -------------------------------------------------------------------------------- /mlsae/api/analyser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/api/analyser.py -------------------------------------------------------------------------------- /mlsae/api/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/api/models.py -------------------------------------------------------------------------------- /mlsae/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/__init__.py -------------------------------------------------------------------------------- /mlsae/metrics/auxiliary_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/auxiliary_loss.py -------------------------------------------------------------------------------- /mlsae/metrics/dead_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/dead_latents.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_fvu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_fvu.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_l0_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_l0_norm.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_l1_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_l1_norm.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_logit_kl_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_logit_kl_div.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_logit_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_logit_mse.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_loss_delta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_loss_delta.py -------------------------------------------------------------------------------- /mlsae/metrics/layerwise_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/layerwise_mse.py -------------------------------------------------------------------------------- /mlsae/metrics/mse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/mse_loss.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_dead_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_dead_latents.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_layerwise_fvu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_layerwise_fvu.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_layerwise_l0_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_layerwise_l0_norm.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_layerwise_l1_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_layerwise_l1_norm.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_layerwise_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_layerwise_mse.py -------------------------------------------------------------------------------- /mlsae/metrics/tests/test_loss_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/metrics/tests/test_loss_mse.py -------------------------------------------------------------------------------- /mlsae/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/__init__.py -------------------------------------------------------------------------------- /mlsae/model/autoencoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/autoencoders/__init__.py -------------------------------------------------------------------------------- /mlsae/model/autoencoders/standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/autoencoders/standard.py -------------------------------------------------------------------------------- /mlsae/model/autoencoders/tests/test_autoencoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/autoencoders/tests/test_autoencoders.py -------------------------------------------------------------------------------- /mlsae/model/autoencoders/topk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/autoencoders/topk.py -------------------------------------------------------------------------------- /mlsae/model/autoencoders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/autoencoders/utils.py -------------------------------------------------------------------------------- /mlsae/model/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/data.py -------------------------------------------------------------------------------- /mlsae/model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/decoder.py -------------------------------------------------------------------------------- /mlsae/model/geom_median.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/geom_median.py -------------------------------------------------------------------------------- /mlsae/model/kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/kernels.py -------------------------------------------------------------------------------- /mlsae/model/lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/lightning.py -------------------------------------------------------------------------------- /mlsae/model/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/__init__.py -------------------------------------------------------------------------------- /mlsae/model/transformers/gemma2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/gemma2.py -------------------------------------------------------------------------------- /mlsae/model/transformers/gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/gpt2.py -------------------------------------------------------------------------------- /mlsae/model/transformers/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/llama.py -------------------------------------------------------------------------------- /mlsae/model/transformers/models/gemma2/modeling_gemma2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/models/gemma2/modeling_gemma2.py -------------------------------------------------------------------------------- /mlsae/model/transformers/models/gpt2/modeling_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/models/gpt2/modeling_gpt2.py -------------------------------------------------------------------------------- /mlsae/model/transformers/models/llama/modeling_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/models/llama/modeling_llama.py -------------------------------------------------------------------------------- /mlsae/model/transformers/pythia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/pythia.py -------------------------------------------------------------------------------- /mlsae/model/transformers/tests/test_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/tests/test_gpt2.py -------------------------------------------------------------------------------- /mlsae/model/transformers/tests/test_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/transformers/tests/test_llama.py -------------------------------------------------------------------------------- /mlsae/model/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model/types.py -------------------------------------------------------------------------------- /mlsae/model_card.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/model_card.py -------------------------------------------------------------------------------- /mlsae/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/trainer/__init__.py -------------------------------------------------------------------------------- /mlsae/trainer/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/trainer/config.py -------------------------------------------------------------------------------- /mlsae/trainer/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/trainer/test.py -------------------------------------------------------------------------------- /mlsae/trainer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/trainer/train.py -------------------------------------------------------------------------------- /mlsae/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/mlsae/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/test.py -------------------------------------------------------------------------------- /tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/tests.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/train.py -------------------------------------------------------------------------------- /upload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/upload.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tim-lawson/mlsae/HEAD/uv.lock --------------------------------------------------------------------------------