MadryLab / trakLinks
A fast, effective data attribution method for neural networks in PyTorch
☆218Updated 10 months ago
Alternatives and similar repositories for trak
Users that are interested in trak are comparing it to the libraries listed below
Sorting:
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆163Updated 3 months ago
- Influence Analysis and Estimation - Survey, Papers, and Taxonomy☆83Updated last year
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆86Updated 2 weeks ago
- A simple PyTorch implementation of influence functions.☆91Updated last year
- Data for "Datamodels: Predicting Predictions with Training Data"☆97Updated 2 years ago
- AI Logging for Interpretability and Explainability🔬☆128Updated last year
- ☆75Updated 3 years ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆77Updated last year
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".☆105Updated 2 years ago
- ☆240Updated last year
- ☆62Updated 3 years ago
- ☆106Updated 7 months ago
- ☆31Updated last year
- Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…☆108Updated last year
- A Mechanistic Understanding of Alignment Algorithms: A Case Study on DPO and Toxicity.☆80Updated 6 months ago
- ☆57Updated 2 years ago
- NeuroSurgeon is a package that enables researchers to uncover and manipulate subnetworks within models in Huggingface Transformers☆41Updated 7 months ago
- Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments☆74Updated 2 years ago
- Efficient empirical NTKs in PyTorch☆22Updated 3 years ago
- ☆23Updated last year
- ☆26Updated 7 months ago
- ☆37Updated 9 months ago
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆152Updated 2 years ago
- ☆50Updated last year
- Code for the paper "The Journey, Not the Destination: How Data Guides Diffusion Models"☆25Updated last year
- ☆99Updated last year
- Conformal Language Modeling☆32Updated last year
- Official Repository for ICML 2023 paper "Can Neural Network Memorization Be Localized?"☆19Updated last year
- Evaluate interpretability methods on localizing and disentangling concepts in LLMs.☆54Updated last year
- This is an official repository for "LAVA: Data Valuation without Pre-Specified Learning Algorithms" (ICLR2023).☆51Updated last year