LICENSE
README.md
pyproject.toml
bergson/__init__.py
bergson/__main__.py
bergson/build.py
bergson/collection.py
bergson/config.py
bergson/data.py
bergson/distributed.py
bergson/gradients.py
bergson/huggingface.py
bergson/process_preconditioners.py
bergson/reduce.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/collector/__init__.py
bergson/collector/collector.py
bergson/collector/dist_preconditioners_gradient_collector.py
bergson/collector/gradient_collectors.py
bergson/collector/in_memory_collector.py
bergson/hessians/apply_hessian.py
bergson/hessians/eigenvectors.py
bergson/hessians/hessian_approximations.py
bergson/hessians/kfac.py
bergson/hessians/shampoo.py
bergson/hessians/sharded_computation.py
bergson/hessians/tkfac.py
bergson/normalizer/__init__.py
bergson/normalizer/fit_normalizers.py
bergson/query/__init__.py
bergson/query/attributor.py
bergson/query/faiss_index.py
bergson/query/query_index.py
bergson/score/__init__.py
bergson/score/score.py
bergson/score/score_writer.py
bergson/score/scorer.py
bergson/utils/__init__.py
bergson/utils/auto_batch_size.py
bergson/utils/gradcheck.py
bergson/utils/logger.py
bergson/utils/math.py
bergson/utils/peft.py
bergson/utils/utils.py
bergson/utils/worker_utils.py
tests/test_attributor.py
tests/test_batch_size_invariance.py
tests/test_build.py
tests/test_data.py
tests/test_gradcheck.py
tests/test_gradients.py
tests/test_normalizer.py
tests/test_reduce.py
tests/test_score.py
tests/test_shrinkage.py
tests/test_trainer_callback.py