LICENSE
README.md
pyproject.toml
dattri_llm/__init__.py
dattri_llm/options.py
dattri_llm/py.typed
dattri_llm/task.py
dattri_llm.egg-info/PKG-INFO
dattri_llm.egg-info/SOURCES.txt
dattri_llm.egg-info/dependency_links.txt
dattri_llm.egg-info/requires.txt
dattri_llm.egg-info/top_level.txt
dattri_llm/attribution/__init__.py
dattri_llm/attribution/arguments.py
dattri_llm/attribution/base.py
dattri_llm/attribution/score.py
dattri_llm/attribution/utils.py
dattri_llm/attribution/algorithm/__init__.py
dattri_llm/attribution/algorithm/adamw_influence.py
dattri_llm/attribution/algorithm/dvemb.py
dattri_llm/attribution/algorithm/kronecker.py
dattri_llm/attribution/algorithm/less.py
dattri_llm/attribution/algorithm/tracin.py
dattri_llm/attribution/algorithm/trajectory.py
dattri_llm/gradient/__init__.py
dattri_llm/gradient/async_writer.py
dattri_llm/gradient/datasets.py
dattri_llm/gradient/gradient.py
dattri_llm/gradient/optimizer_state.py
dattri_llm/gradient/snapshots.py
dattri_llm/gradient/storage_manager.py
dattri_llm/gradient/streaming.py
dattri_llm/gradient/callbacks/__init__.py
dattri_llm/gradient/callbacks/base.py
dattri_llm/gradient/callbacks/capture_callback.py
dattri_llm/gradient/callbacks/data_selection_callback.py
dattri_llm/gradient/callbacks/kronecker_covariance_callback.py
dattri_llm/gradient/callbacks/offload_callback.py
dattri_llm/gradient/callbacks/optimizer_state_callback.py
dattri_llm/gradient/callbacks/parameter_snapshot_callback.py
dattri_llm/gradient/hooks/__init__.py
dattri_llm/gradient/hooks/config.py
dattri_llm/gradient/hooks/hooks.py
dattri_llm/gradient/hooks/manager.py
dattri_llm/gradient/ops/__init__.py
dattri_llm/gradient/ops/dot.py
dattri_llm/gradient/ops/dtypes.py
dattri_llm/gradient/ops/kronecker.py
dattri_llm/gradient/ops/materialize.py
dattri_llm/gradient/ops/norm.py
dattri_llm/gradient/ops/optimizer.py
dattri_llm/gradient/ops/preprocess.py
dattri_llm/gradient/ops/projection.py
dattri_llm/gradient/ops/types.py
dattri_llm/utils/__init__.py
dattri_llm/utils/autograd.py
dattri_llm/utils/cache.py
dattri_llm/utils/distributed.py
dattri_llm/utils/hashing.py
dattri_llm/utils/module.py
tests/test_task.py