transformer_lens.tools packageΒΆ
SubpackagesΒΆ
- transformer_lens.tools.analysis package
- Submodules
- Module contents
DirectLogitAttributionJacobianLensJacobianLens.jacobiansJacobianLens.n_promptsJacobianLens.d_modelJacobianLens.metadataJacobianLens.ablation_hooks()JacobianLens.clear_device_cache()JacobianLens.fit()JacobianLens.from_pretrained()JacobianLens.lens_vectors()JacobianLens.load()JacobianLens.merge()JacobianLens.readout()JacobianLens.save()JacobianLens.source_layersJacobianLens.steering_hooks()JacobianLens.swap_hooks()JacobianLens.transport()JacobianLens.validate_model()
JacobianLensReadoutJacobianLensReadout.lens_topk_valuesJacobianLensReadout.lens_topk_indicesJacobianLensReadout.model_topk_valuesJacobianLensReadout.model_topk_indicesJacobianLensReadout.lens_logitsJacobianLensReadout.model_logitsJacobianLensReadout.tokensJacobianLensReadout.positionsJacobianLensReadout.use_jacobianJacobianLensReadout.lens_logitsJacobianLensReadout.lens_topk_indicesJacobianLensReadout.lens_topk_valuesJacobianLensReadout.model_logitsJacobianLensReadout.model_topk_indicesJacobianLensReadout.model_topk_valuesJacobianLensReadout.positionsJacobianLensReadout.tokensJacobianLensReadout.top_tokens()JacobianLensReadout.use_jacobian
direct_logit_attribution()get_act_patch_direct_path()get_act_patch_direct_path_all_sources()
- transformer_lens.tools.model_registry package
- Submodules
- transformer_lens.tools.model_registry.alias_drift module
- transformer_lens.tools.model_registry.api module
- transformer_lens.tools.model_registry.discover_architectures module
- transformer_lens.tools.model_registry.exceptions module
- transformer_lens.tools.model_registry.generate_report module
- transformer_lens.tools.model_registry.hf_scraper module
- transformer_lens.tools.model_registry.registry_io module
- transformer_lens.tools.model_registry.relevancy module
- transformer_lens.tools.model_registry.schemas module
- transformer_lens.tools.model_registry.validate module
- transformer_lens.tools.model_registry.verification module
- transformer_lens.tools.model_registry.verify_models module
- Module contents
ArchitectureAnalysisArchitectureAnalysis.architecture_idArchitectureAnalysis.total_modelsArchitectureAnalysis.total_downloadsArchitectureAnalysis.priority_scoreArchitectureAnalysis.top_modelsArchitectureAnalysis.architecture_idArchitectureAnalysis.priority_scoreArchitectureAnalysis.to_dict()ArchitectureAnalysis.top_modelsArchitectureAnalysis.total_downloadsArchitectureAnalysis.total_models
ArchitectureGapArchitectureGap.architecture_idArchitectureGap.total_modelsArchitectureGap.sample_modelsArchitectureGap.total_downloadsArchitectureGap.min_param_countArchitectureGap.relevancy_scoreArchitectureGap.architecture_idArchitectureGap.from_dict()ArchitectureGap.min_param_countArchitectureGap.relevancy_scoreArchitectureGap.sample_modelsArchitectureGap.to_dict()ArchitectureGap.total_downloadsArchitectureGap.total_models
ArchitectureGapsReportArchitectureGapsReport.generated_atArchitectureGapsReport.scan_infoArchitectureGapsReport.total_unsupported_architecturesArchitectureGapsReport.total_unsupported_modelsArchitectureGapsReport.gapsArchitectureGapsReport.from_dict()ArchitectureGapsReport.gapsArchitectureGapsReport.generated_atArchitectureGapsReport.scan_infoArchitectureGapsReport.to_dict()ArchitectureGapsReport.total_unsupported_architecturesArchitectureGapsReport.total_unsupported_models
ArchitectureNotSupportedErrorArchitectureStatsArchitectureStats.architecture_idArchitectureStats.is_supportedArchitectureStats.model_countArchitectureStats.verified_countArchitectureStats.example_modelsArchitectureStats.architecture_idArchitectureStats.example_modelsArchitectureStats.is_supportedArchitectureStats.model_countArchitectureStats.to_dict()ArchitectureStats.verified_count
DataNotLoadedErrorDataValidationErrorModelEntryModelEntry.architecture_idModelEntry.model_idModelEntry.statusModelEntry.verified_dateModelEntry.metadataModelEntry.noteModelEntry.phase1_scoreModelEntry.phase2_scoreModelEntry.phase3_scoreModelEntry.phase4_scoreModelEntry.phase7_scoreModelEntry.phase8_scoreModelEntry.architecture_idModelEntry.from_dict()ModelEntry.metadataModelEntry.model_idModelEntry.noteModelEntry.phase1_scoreModelEntry.phase2_scoreModelEntry.phase3_scoreModelEntry.phase4_scoreModelEntry.phase7_scoreModelEntry.phase8_scoreModelEntry.statusModelEntry.to_dict()ModelEntry.verified_date
ModelMetadataModelNotFoundErrorModelRegistryErrorScanInfoSupportedModelsReportSupportedModelsReport.generated_atSupportedModelsReport.scan_infoSupportedModelsReport.total_architecturesSupportedModelsReport.total_modelsSupportedModelsReport.total_verifiedSupportedModelsReport.modelsSupportedModelsReport.from_dict()SupportedModelsReport.generated_atSupportedModelsReport.modelsSupportedModelsReport.scan_infoSupportedModelsReport.to_dict()SupportedModelsReport.total_architecturesSupportedModelsReport.total_modelsSupportedModelsReport.total_provisionalSupportedModelsReport.total_verified
VerificationHistoryVerificationHistory.recordsVerificationHistory.last_updatedVerificationHistory.add_record()VerificationHistory.from_dict()VerificationHistory.get_record()VerificationHistory.invalidate()VerificationHistory.is_verified()VerificationHistory.last_updatedVerificationHistory.recordsVerificationHistory.to_dict()
VerificationRecordVerificationRecord.model_idVerificationRecord.architecture_idVerificationRecord.verified_dateVerificationRecord.verified_byVerificationRecord.transformerlens_versionVerificationRecord.notesVerificationRecord.invalidatedVerificationRecord.invalidation_reasonVerificationRecord.architecture_idVerificationRecord.from_dict()VerificationRecord.invalidatedVerificationRecord.invalidation_reasonVerificationRecord.model_idVerificationRecord.notesVerificationRecord.to_dict()VerificationRecord.transformerlens_versionVerificationRecord.verified_byVerificationRecord.verified_date
- Submodules
Module contentsΒΆ
TransformerLens tools package.
This package contains utilities and tools for working with TransformerLens, including the model registry for discovering compatible HuggingFace models.
- Subpackages:
analysis: High-level interpretability analyses (e.g. Direct Logit Attribution)
model_registry: Tools for discovering and documenting supported models