├── .gitignore ├── LICENSE ├── README.md ├── environment.yml ├── figures └── tangent.jpg ├── ruff.toml └── src ├── args.py ├── datasets ├── cars.py ├── cifar10.py ├── cifar100.py ├── common.py ├── dtd.py ├── eurosat.py ├── gtsrb.py ├── imagenet.py ├── mnist.py ├── registry.py ├── resisc45.py ├── stl10.py ├── sun397.py ├── svhn.py └── templates.py ├── distributed.py ├── eval.py ├── eval_single_task.py ├── eval_task_addition.py ├── eval_task_negation.py ├── finetune.py ├── heads.py ├── linearize.py ├── modeling.py ├── task_vectors.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/environment.yml -------------------------------------------------------------------------------- /figures/tangent.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/figures/tangent.jpg -------------------------------------------------------------------------------- /ruff.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/ruff.toml -------------------------------------------------------------------------------- /src/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/args.py -------------------------------------------------------------------------------- /src/datasets/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/cars.py -------------------------------------------------------------------------------- /src/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/cifar10.py -------------------------------------------------------------------------------- /src/datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/cifar100.py -------------------------------------------------------------------------------- /src/datasets/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/common.py -------------------------------------------------------------------------------- /src/datasets/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/dtd.py -------------------------------------------------------------------------------- /src/datasets/eurosat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/eurosat.py -------------------------------------------------------------------------------- /src/datasets/gtsrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/gtsrb.py -------------------------------------------------------------------------------- /src/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/imagenet.py -------------------------------------------------------------------------------- /src/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/mnist.py -------------------------------------------------------------------------------- /src/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/registry.py -------------------------------------------------------------------------------- /src/datasets/resisc45.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/resisc45.py -------------------------------------------------------------------------------- /src/datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/stl10.py -------------------------------------------------------------------------------- /src/datasets/sun397.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/sun397.py -------------------------------------------------------------------------------- /src/datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/svhn.py -------------------------------------------------------------------------------- /src/datasets/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/datasets/templates.py -------------------------------------------------------------------------------- /src/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/distributed.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/eval_single_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/eval_single_task.py -------------------------------------------------------------------------------- /src/eval_task_addition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/eval_task_addition.py -------------------------------------------------------------------------------- /src/eval_task_negation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/eval_task_negation.py -------------------------------------------------------------------------------- /src/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/finetune.py -------------------------------------------------------------------------------- /src/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/heads.py -------------------------------------------------------------------------------- /src/linearize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/linearize.py -------------------------------------------------------------------------------- /src/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/modeling.py -------------------------------------------------------------------------------- /src/task_vectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/task_vectors.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gortizji/tangent_task_arithmetic/HEAD/src/utils.py --------------------------------------------------------------------------------