gitextract_5qwi3ryd/

├── .github/
│   └── ci.yml
├── .gitignore
├── .pre-commit-config.yaml
├── Makefile
├── README.md
├── STORY.md
├── requirements.txt
├── setup.cfg
├── src/
│   ├── data.py
│   ├── memory/
│   │   ├── __init__.py
│   │   ├── associative.py
│   │   ├── batch_associative.py
│   │   └── multitoken_batch_associative.py
│   ├── models/
│   │   ├── __init__.py
│   │   ├── gpt2_associative.py
│   │   └── gpt2_working.py
│   ├── train.py
│   └── train_parallel.py
└── test/
    ├── __init__.py
    ├── memory/
    │   ├── __init__..py
    │   ├── test_associative.py
    │   ├── test_associative_batch.py
    │   ├── test_memory_features.py
    │   ├── test_memory_types.py
    │   └── test_multitoken.py
    ├── models/
    │   ├── __init__.py
    │   └── test_gpt2_associative.py
    └── test_training.py