Index A | B | C | E | F | G | H | I | J | K | L | M | N | P | R | S | T | U | V A anti_targets (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory attribute) available_lens_artifacts() (in module tuned_lens.load_artifacts) B batch_axes (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) batch_shape (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) C clip() (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic method) cross_entropy() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) E entropy() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) F figure() (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic method) format() (tuned_lens.plotting.token_formatter.TokenFormatter method) forward() (tuned_lens.nn.lenses.Lens method) (tuned_lens.nn.lenses.LogitLens method) (tuned_lens.nn.lenses.TunedLens method) (tuned_lens.nn.unembed.Unembed method) forward_kl() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) from_dict() (tuned_lens.nn.lenses.TunedLensConfig class method) from_lens_and_cache() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory class method) from_lens_and_model() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory class method) from_model() (tuned_lens.nn.lenses.LogitLens class method) (tuned_lens.nn.lenses.TunedLens class method) from_model_and_pretrained() (tuned_lens.nn.lenses.TunedLens class method) from_unembed_and_pretrained() (tuned_lens.nn.lenses.TunedLens class method) G generate() (tuned_lens.nn.lenses.TunedLens method) H heatmap() (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic method) hover_over_entries (tuned_lens.plotting.trajectory_plotting.TrajectoryLabels attribute) I includes_output (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) input_ids (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory attribute) InversionOutput (class in tuned_lens.nn.unembed) invert() (tuned_lens.nn.unembed.Unembed method) J js_divergence() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) K kl_divergence() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) L label_strings (tuned_lens.plotting.trajectory_plotting.TrajectoryLabels attribute) Lens (class in tuned_lens.nn.lenses) load_lens_artifacts() (in module tuned_lens.load_artifacts) log_prob_diff() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) log_probs (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory attribute) LogitLens (class in tuned_lens.nn.lenses) M max (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) max_probability() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) min (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) model_log_probs (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) module tuned_lens.load_artifacts tuned_lens.nn.lenses tuned_lens.nn.unembed tuned_lens.plotting tuned_lens.plotting.prediction_trajectory tuned_lens.plotting.token_formatter tuned_lens.plotting.trajectory_plotting N n_batch_axis (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) name (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) num_layers (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) num_tokens (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) P pad_token_repr_to_max_len() (tuned_lens.plotting.token_formatter.TokenFormatter method) PredictionTrajectory (class in tuned_lens.plotting.prediction_trajectory) probs (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property) R rank() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) S save() (tuned_lens.nn.lenses.TunedLens method) sequence_labels (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) slice_sequence() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) stats (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) stride() (tuned_lens.plotting.trajectory_plotting.TrajectoryLabels method) (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic method) T targets (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory attribute) template_and_customdata() (tuned_lens.plotting.trajectory_plotting.TrajectoryLabels method) to_dict() (tuned_lens.nn.lenses.TunedLensConfig method) TokenFormatter (class in tuned_lens.plotting.token_formatter) total_variation() (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory method) trajectory_labels (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) TrajectoryLabels (class in tuned_lens.plotting.trajectory_plotting) TrajectoryStatistic (class in tuned_lens.plotting.trajectory_plotting) transform_hidden() (tuned_lens.nn.lenses.Lens method) (tuned_lens.nn.lenses.LogitLens method) (tuned_lens.nn.lenses.TunedLens method) trunc_string_left() (in module tuned_lens.plotting.trajectory_plotting) tuned_lens.load_artifacts module tuned_lens.nn.lenses module tuned_lens.nn.unembed module tuned_lens.plotting module tuned_lens.plotting.prediction_trajectory module tuned_lens.plotting.token_formatter module tuned_lens.plotting.trajectory_plotting module TunedLens (class in tuned_lens.nn.lenses) TunedLensConfig (class in tuned_lens.nn.lenses) U Unembed (class in tuned_lens.nn.unembed) unembedding_hash() (tuned_lens.nn.unembed.Unembed method) units (tuned_lens.plotting.trajectory_plotting.TrajectoryStatistic attribute) V vocab_size (tuned_lens.plotting.prediction_trajectory.PredictionTrajectory property)