LICENSE
README.md
pyproject.toml
setup.py
open_titans/__init__.py
open_titans.egg-info/PKG-INFO
open_titans.egg-info/SOURCES.txt
open_titans.egg-info/dependency_links.txt
open_titans.egg-info/requires.txt
open_titans.egg-info/top_level.txt
open_titans/configs/__init__.py
open_titans/configs/configuration_atlas.py
open_titans/configs/configuration_titans.py
open_titans/configs/configuration_utils.py
open_titans/generation/__init__.py
open_titans/generation/generation_mixin.py
open_titans/generation/titans_cache.py
open_titans/models/__init__.py
open_titans/models/modeling_utils.py
open_titans/models/atlas/configuration_atlas.py
open_titans/models/atlas/modeling_atlas.py
open_titans/models/atlas/registry.py
open_titans/models/miras/__init__.py
open_titans/models/miras/configuration_miras.py
open_titans/models/miras/modeling_miras.py
open_titans/models/miras/registry.py
open_titans/models/titans_mac/__init__.py
open_titans/models/titans_mac/configuration_mac.py
open_titans/models/titans_mac/modeling_mac.py
open_titans/models/titans_mag/__init__.py
open_titans/models/titans_mag/configuration_mag.py
open_titans/models/titans_mag/modeling_mag.py
open_titans/models/titans_mal/__init__.py
open_titans/models/titans_mal/configuration_mal.py
open_titans/models/titans_mal/modeling_mal.py
open_titans/modules/__init__.py
open_titans/modules/attention/__init__.py
open_titans/modules/attention/attentional_bias.py
open_titans/modules/gates/__init__.py
open_titans/modules/gates/retention.py
open_titans/modules/memory/__init__.py
open_titans/modules/memory/functional.py
open_titans/modules/memory/memory_model.py
open_titans/modules/memory/neural_memory.py
open_titans/modules/memory/retrospective.py
open_titans/modules/memory/update_rule.py
open_titans/optim/__init__.py
open_titans/optim/muon.py
open_titans/trainer/__init__.py
open_titans/trainer/nested_learning.py
open_titans/trainer/training_args.py
open_titans/utils/__init__.py
open_titans/utils/hardware_utils.py
open_titans/utils/memory_tracker.py
tests/test_atlas.py
tests/test_attentional_bias.py
tests/test_generation.py
tests/test_memory_update_rule.py
tests/test_miras.py
tests/test_nested_learning.py
tests/test_neural_memory.py
tests/test_retention_regularization.py
tests/test_titans_mac.py
tests/test_titans_mag.py
tests/test_titans_mal.py