LICENSE
README.md
pyproject.toml
bergson/__init__.py
bergson/__main__.py
bergson/build.py
bergson/builder.py
bergson/collection.py
bergson/data.py
bergson/diagnose.py
bergson/distributed.py
bergson/format.py
bergson/gradients.py
bergson/huggingface.py
bergson/process_autocorrelation.py
bergson/process_grads.py
bergson/validate.py
bergson.egg-info/PKG-INFO
bergson.egg-info/SOURCES.txt
bergson.egg-info/dependency_links.txt
bergson.egg-info/entry_points.txt
bergson.egg-info/requires.txt
bergson.egg-info/top_level.txt
bergson/approx_unrolling/approx_unrolling_math.py
bergson/approx_unrolling/pipeline.py
bergson/approx_unrolling/precompute_checkpoints.py
bergson/approx_unrolling/segment_aggregation.py
bergson/cli/__init__.py
bergson/cli/commands.py
bergson/cli/trackstar.py
bergson/collector/__init__.py
bergson/collector/collector.py
bergson/collector/dist_autocorrelation_gradient_collector.py
bergson/collector/gradient_collectors.py
bergson/collector/in_memory_collector.py
bergson/config/__init__.py
bergson/config/config.py
bergson/config/config_io.py
bergson/hessians/apply_hessian.py
bergson/hessians/autocorrelation.py
bergson/hessians/eigenvectors.py
bergson/hessians/hessian_approximations.py
bergson/hessians/inversion.py
bergson/hessians/kfac.py
bergson/hessians/pipeline.py
bergson/hessians/preconditioner.py
bergson/hessians/shampoo.py
bergson/hessians/sharded_computation.py
bergson/hessians/tkfac.py
bergson/magic/__init__.py
bergson/magic/cli.py
bergson/magic/config.py
bergson/magic/data_stream.py
bergson/magic/dtensor_patch.py
bergson/magic/fsdp.py
bergson/magic/metasmoothness.py
bergson/magic/optim.py
bergson/magic/rtl_tqdm.py
bergson/magic/swap.py
bergson/magic/trainer.py
bergson/query/__init__.py
bergson/query/attributor.py
bergson/query/faiss_index.py
bergson/query/query_index.py
bergson/recall/__init__.py
bergson/recall/facts.py
bergson/recall/generate.py
bergson/recall/recall.py
bergson/score/__init__.py
bergson/score/score.py
bergson/score/score_writer.py
bergson/score/scorer.py
bergson/utils/__init__.py
bergson/utils/batch_size.py
bergson/utils/csv_writer.py
bergson/utils/gradcheck.py
bergson/utils/load_from_optimizer.py
bergson/utils/logger.py
bergson/utils/logging.py
bergson/utils/math.py
bergson/utils/peft.py
bergson/utils/utils.py
bergson/utils/worker_utils.py
tests/test_adam_state_loading.py
tests/test_advantages.py
tests/test_approx_unrolling_checkpoint_parse.py
tests/test_attention_head_normalizers.py
tests/test_attribute_tokens.py
tests/test_attributor.py
tests/test_banked_model_loading.py
tests/test_batch_size_invariance.py
tests/test_build.py
tests/test_builder.py
tests/test_cli.py
tests/test_compute_lambda.py
tests/test_config_runner.py
tests/test_data.py
tests/test_ddp.py
tests/test_diagnose.py
tests/test_distributed_batch_budget.py
tests/test_distributed_cap.py
tests/test_distributed_magic.py
tests/test_epoch_shuffle.py
tests/test_faiss_ann_cpu_load.py
tests/test_force_math_sdp.py
tests/test_format.py
tests/test_global_projection.py
tests/test_grad_clipping.py
tests/test_gradcheck.py
tests/test_gradients.py
tests/test_hessian.py
tests/test_log_fn.py
tests/test_magic.py
tests/test_metasmoothness.py
tests/test_multi_query_validate.py
tests/test_multinode.py
tests/test_muon.py
tests/test_normalizer.py
tests/test_per_token_lds.py
tests/test_projection_inner_products.py
tests/test_query_batching_invariance.py
tests/test_recall.py
tests/test_reduce.py
tests/test_score.py
tests/test_streaming_build.py
tests/test_tokenize.py
tests/test_trainer_callback.py
tests/test_truncation.py
tests/test_weighted_ce.py