Index A | B | C | D | E | F | G | H | I | L | M | N | O | P | Q | R | S | T | U A aggregation (bergson.PreprocessConfig attribute) apply() (bergson.FiniteDiff method) attention (bergson.IndexConfig attribute) AttentionConfig (class in bergson) attribute_tokens (bergson.IndexConfig attribute) Attributor (class in bergson) auto_batch_size (bergson.IndexConfig attribute) B backward_hook() (bergson.GradientCollector method) (bergson.InMemoryCollector method) batch_size (bergson.FaissConfig attribute) (bergson.ScoreConfig attribute) bergson module Build (class in bergson.__main__) build_cfg (bergson.__main__.Build attribute) builder (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) Builder (class in bergson) C cfg (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) chunk_length (bergson.DataConfig attribute) clear() (bergson.FiniteDiff method) collect_gradients() (in module bergson) CollectorComputer (class in bergson) completion_column (bergson.DataConfig attribute) conversation_column (bergson.DataConfig attribute) D data (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) data_kwargs (bergson.DataConfig attribute) DataConfig (class in bergson) dataset (bergson.DataConfig attribute) decode_into_subclasses (bergson.__main__.Build attribute) (bergson.__main__.Hessian attribute) (bergson.__main__.Query attribute) (bergson.__main__.Reduce attribute) (bergson.__main__.Score attribute) (bergson.__main__.Trackstar attribute) (bergson.DataConfig attribute) (bergson.IndexConfig attribute) (bergson.PreprocessConfig attribute) (bergson.QueryConfig attribute) (bergson.ScoreConfig attribute) device_map_auto (bergson.QueryConfig attribute) E execute() (bergson.__main__.Build method) (bergson.__main__.Hessian method) (bergson.__main__.Query method) (bergson.__main__.Reduce method) (bergson.__main__.Score method) (bergson.__main__.Trackstar method) F faiss (bergson.QueryConfig attribute) FaissConfig (class in bergson) filter_modules (bergson.IndexConfig attribute) FiniteDiff (class in bergson) flush() (bergson.Builder method) force_math_sdp (bergson.IndexConfig attribute) format_template (bergson.DataConfig attribute) G grad_buffer (bergson.Builder attribute) GradientCollector (class in bergson) GradientProcessor (class in bergson) gradients (bergson.InMemoryCollector attribute) H head_dim (bergson.AttentionConfig attribute) head_size (bergson.AttentionConfig attribute) Hessian (class in bergson.__main__) hessian_cfg (bergson.__main__.Hessian attribute) hessian_path (bergson.PreprocessConfig attribute) hessians (bergson.GradientProcessor attribute) hessians_eigen (bergson.GradientProcessor attribute) higher_is_better (bergson.ScoreConfig attribute) I include_bias (bergson.GradientProcessor attribute) (bergson.IndexConfig attribute) index (bergson.QueryConfig attribute) index_cfg (bergson.__main__.Build attribute) (bergson.__main__.Hessian attribute) (bergson.__main__.Reduce attribute) (bergson.__main__.Score attribute) (bergson.__main__.Trackstar attribute) index_factory (bergson.FaissConfig attribute) IndexConfig (class in bergson) InMemoryCollector (class in bergson) L label_smoothing (bergson.IndexConfig attribute) load() (bergson.GradientProcessor class method) load_from_optimizer() (in module bergson) load_gradient_dataset() (in module bergson) load_gradients() (in module bergson) load_token_gradients() (in module bergson) loss_fn (bergson.IndexConfig attribute) loss_reduction (bergson.IndexConfig attribute) M max_batch_size (bergson.IndexConfig attribute) max_train_examples (bergson.FaissConfig attribute) mix_autocorrelation_matrices() (in module bergson) mmap_index (bergson.FaissConfig attribute) mod_grads (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) model (bergson.QueryConfig attribute) module bergson modules (bergson.IndexConfig attribute) (bergson.ScoreConfig attribute) N normalize_aggregated_grad (bergson.PreprocessConfig attribute) normalizers (bergson.GradientProcessor attribute) nprobe (bergson.FaissConfig attribute) num_heads (bergson.AttentionConfig attribute) num_shards (bergson.FaissConfig attribute) num_token_grads (bergson.TokenGradients property) O optimizer_state (bergson.IndexConfig attribute) P partial_run_path (bergson.IndexConfig property) precision (bergson.ScoreConfig attribute) precondition (bergson.Attributor attribute) preprocess_cfg (bergson.__main__.Build attribute) (bergson.__main__.Reduce attribute) (bergson.__main__.Score attribute) (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) PreprocessConfig (class in bergson) process_batch() (bergson.GradientCollector method) (bergson.InMemoryCollector method) profile (bergson.IndexConfig attribute) projection_dim (bergson.GradientProcessor attribute) (bergson.IndexConfig attribute) projection_target (bergson.GradientProcessor attribute) (bergson.IndexConfig attribute) projection_type (bergson.GradientProcessor attribute) (bergson.IndexConfig attribute) prompt_column (bergson.DataConfig attribute) Q Query (class in bergson.__main__) query_path (bergson.ScoreConfig attribute) QueryConfig (class in bergson) R record (bergson.QueryConfig attribute) Reduce (class in bergson.__main__) reshape_to_square (bergson.GradientProcessor attribute) (bergson.IndexConfig attribute) reward_column (bergson.DataConfig attribute) run_with_collector_hooks() (bergson.CollectorComputer method) S save() (bergson.GradientProcessor method) score (bergson.ScoreConfig attribute) Score (class in bergson.__main__) score() (bergson.Scorer method) score_cfg (bergson.__main__.Score attribute) ScoreConfig (class in bergson) scorer (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) Scorer (class in bergson) scores (bergson.InMemoryCollector attribute) search() (bergson.Attributor method) setup() (bergson.GradientCollector method) (bergson.InMemoryCollector method) skip_hessians (bergson.GradientCollector attribute) (bergson.InMemoryCollector attribute) skip_index (bergson.IndexConfig attribute) skip_nan_rewards (bergson.DataConfig attribute) split (bergson.DataConfig attribute) split_attention_modules (bergson.IndexConfig attribute) stats_sample_size (bergson.IndexConfig attribute) store() (bergson.FiniteDiff method) stream_shard_size (bergson.IndexConfig attribute) subset (bergson.DataConfig attribute) swap() (bergson.FiniteDiff method) T teardown() (bergson.Builder method) (bergson.GradientCollector method) (bergson.InMemoryCollector method) text_field (bergson.QueryConfig attribute) token_batch_size (bergson.IndexConfig attribute) TokenGradients (class in bergson) top_k (bergson.QueryConfig attribute) trace() (bergson.Attributor method) Trackstar (class in bergson.__main__) trackstar_cfg (bergson.__main__.Trackstar attribute) truncation (bergson.DataConfig attribute) U unit_norm (bergson.QueryConfig attribute) unit_normalize (bergson.PreprocessConfig attribute)