Caching
Language model activations
encoding.utils.ActivationCache
persists multi-layer activations to a single.pkl
per configencoding.utils.LazyLayerCache
loads metadata fast and layers on demandCache keys include story, model name, context type, lookback, dataset_type, etc.
Speech activations
encoding.utils.SpeechActivationCache
andSpeechLazyLayerCache
mirror the API for speech features with optional time arrays
Model artifacts
encoding.utils.ModelSaver
records hyperparameters and metrics per run directory (hash + timestamp)Use
list_runs()
to enumerate previous runs