Task representations replication

Replication of "Just-in-time and distributed task representations in language models" using Gemma 3 (270m - 4b).

Run

pip install -r requirements.txt
python task_vector_replication.py

Results are written to results/ as JSON (one file per model run).