MadryLab / trak
A fast, effective data attribution method for neural networks in PyTorch
☆187Updated last month
Alternatives and similar repositories for trak:
Users that are interested in trak are comparing it to the libraries listed below
- Influence Analysis and Estimation - Survey, Papers, and Taxonomy☆66Updated 10 months ago
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆120Updated 5 months ago
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆55Updated 2 weeks ago
- AI Logging for Interpretability and Explainability🔬☆97Updated 7 months ago
- ☆58Updated 3 years ago
- A simple PyTorch implementation of influence functions.☆84Updated 7 months ago
- ☆63Updated 2 years ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆57Updated 3 months ago
- Data for "Datamodels: Predicting Predictions with Training Data"☆94Updated last year
- Landing Page for TOFU☆107Updated 3 weeks ago
- Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments☆73Updated 2 years ago
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".☆92Updated last year
- ☆210Updated 8 months ago
- ☆82Updated 11 months ago
- Conformal Language Modeling☆27Updated last year
- ☆51Updated last year
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆147Updated 2 years ago
- ☆46Updated 11 months ago
- ☆41Updated this week
- ☆28Updated last year
- This is an official repository for "LAVA: Data Valuation without Pre-Specified Learning Algorithms" (ICLR2023).☆45Updated 7 months ago
- ☆86Updated last year
- Distributionally robust neural networks for group shifts☆253Updated 2 years ago
- This is the starter kit for the Trojan Detection Challenge 2023 (LLM Edition), a NeurIPS 2023 competition.☆82Updated 7 months ago
- ☆27Updated 6 months ago
- Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…☆100Updated last year
- Efficient empirical NTKs in PyTorch☆18Updated 2 years ago
- NeuroSurgeon is a package that enables researchers to uncover and manipulate subnetworks within models in Huggingface Transformers☆39Updated 5 months ago
- A resource repository for representation engineering in large language models☆90Updated 2 months ago
- Official Repository for ICML 2023 paper "Can Neural Network Memorization Be Localized?"☆17Updated last year