MadryLab / trak
A fast, effective data attribution method for neural networks in PyTorch
☆202Updated 4 months ago
Alternatives and similar repositories for trak:
Users that are interested in trak are comparing it to the libraries listed below
- Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature☆139Updated 7 months ago
- Influence Analysis and Estimation - Survey, Papers, and Taxonomy☆72Updated last year
- AI Logging for Interpretability and Explainability🔬☆108Updated 9 months ago
- Data for "Datamodels: Predicting Predictions with Training Data"☆95Updated last year
- A simple PyTorch implementation of influence functions.☆85Updated 9 months ago
- ☆65Updated 3 years ago
- `dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.☆68Updated last month
- ☆60Updated 3 years ago
- DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Models (ICLR 2024)☆63Updated 5 months ago
- Source code of "Task arithmetic in the tangent space: Improved editing of pre-trained models".☆97Updated last year
- ☆71Updated this week
- ☆89Updated last month
- ☆88Updated last year
- ☆53Updated 2 years ago
- Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implici…☆104Updated last year
- Conformal Language Modeling☆28Updated last year
- ☆84Updated 8 months ago
- nanoGPT-like codebase for LLM training☆91Updated this week
- Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…☆149Updated 2 years ago
- ☆93Updated last year
- This is an official repository for "LAVA: Data Valuation without Pre-Specified Learning Algorithms" (ICLR2023).☆47Updated 9 months ago
- NeuroSurgeon is a package that enables researchers to uncover and manipulate subnetworks within models in Huggingface Transformers☆39Updated last month
- ☆61Updated 4 months ago
- A library for efficient patching and automatic circuit discovery.☆59Updated last month
- Efficient empirical NTKs in PyTorch☆18Updated 2 years ago
- ☆121Updated last year
- ☆23Updated 3 weeks ago
- ☆28Updated 8 months ago
- Official repository for CMU Machine Learning Department's 10732: Robustness and Adaptivity in Shifting Environments☆73Updated 2 years ago
- ☆46Updated last year