physical-AI-interpretability documentation

Add your content using reStructuredText syntax. See the reStructuredText documentation for details.

Contents:

• src package
  • Subpackages
    • src.attention_maps package
      • Submodules
      • src.attention_maps.act_attention_mapper module
      • Module contents
    • src.ood package
      • Module contents
    • src.sae package
      • Submodules
      • src.sae.builder module
      • src.sae.config module
      • src.sae.sae module
      • src.sae.token_sampler module
      • src.sae.trainer module
      • Module contents
    • src.utils package
      • Submodules
      • src.utils.dataset_factory module
      • Module contents
  • Module contents
    • ACTPolicyWithAttention
      • ACTPolicyWithAttention.select_action()
      • ACTPolicyWithAttention.visualize_attention()
    • MultiModalSAE
      • MultiModalSAE.compute_loss()
      • MultiModalSAE.decode()
      • MultiModalSAE.encode()
      • MultiModalSAE.forward()
    • OODDetector
      • OODDetector.fit_ood_threshold_to_validation_dataset()
      • OODDetector.get_ood_stats()
      • OODDetector.get_reconstruction_error()
      • OODDetector.is_out_of_distribution()
      • OODDetector.needs_ood_fitting()
    • SAEBuilder
      • SAEBuilder.from_default_path()
      • SAEBuilder.load_from_experiment()
      • SAEBuilder.load_from_files()
      • SAEBuilder.load_from_hub()
      • SAEBuilder.load_with_auto_config()
    • SAETrainer
      • SAETrainer.collect_activations()
      • SAETrainer.create_model()
      • SAETrainer.create_optimizer_and_scheduler()
      • SAETrainer.generate_model_card()
      • SAETrainer.load_checkpoint()
      • SAETrainer.push_model_to_hub()
      • SAETrainer.save_checkpoint()
      • SAETrainer.save_complete_model()
      • SAETrainer.train()
      • SAETrainer.train_step()
    • SAETrainingConfig
      • SAETrainingConfig.activation_fn
      • SAETrainingConfig.batch_size
      • SAETrainingConfig.block_size
      • SAETrainingConfig.device
      • SAETrainingConfig.early_stopping_min_delta
      • SAETrainingConfig.early_stopping_patience
      • SAETrainingConfig.expansion_factor
      • SAETrainingConfig.feature_dim
      • SAETrainingConfig.fixed_tokens
      • SAETrainingConfig.gradient_clip_norm
      • SAETrainingConfig.infer_model_params_from_cache()
      • SAETrainingConfig.l1_penalty
      • SAETrainingConfig.learning_rate
      • SAETrainingConfig.log_every
      • SAETrainingConfig.lr_schedule
      • SAETrainingConfig.max_sampled_tokens
      • SAETrainingConfig.num_epochs
      • SAETrainingConfig.num_tokens
      • SAETrainingConfig.optimizer
      • SAETrainingConfig.sampling_strategy
      • SAETrainingConfig.sampling_stride
      • SAETrainingConfig.save_every
      • SAETrainingConfig.token_dim
      • SAETrainingConfig.use_token_sampling
      • SAETrainingConfig.validate_every
      • SAETrainingConfig.validation_split
      • SAETrainingConfig.warmup_epochs
      • SAETrainingConfig.weight_decay
    • TokenSampler
      • TokenSampler.get_sampling_info()
      • TokenSampler.sample_tokens()
    • TokenSamplerConfig
      • TokenSamplerConfig.block_size
      • TokenSamplerConfig.fixed_tokens
      • TokenSamplerConfig.max_sampled_tokens
      • TokenSamplerConfig.random_seed
      • TokenSamplerConfig.sampling_strategy
      • TokenSamplerConfig.sampling_stride
    • create_default_ood_params_path()
    • load_sae_model()
    • load_sae_model_simple()
    • make_dataset_without_config()