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/memory_utils.py
open_titans/generation/stopping_criteria.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/attention/linear_attention.py
open_titans/modules/attention/sliding_window.py
open_titans/modules/gates/__init__.py
open_titans/modules/gates/forget_gates.py
open_titans/modules/gates/miras_bias.py
open_titans/modules/gates/retention.py
open_titans/modules/memory/__init__.py
open_titans/modules/memory/functional.py
open_titans/modules/memory/memory_caching.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/test_time_mem.py
open_titans/modules/memory/update_rule.py
open_titans/optim/__init__.py
open_titans/optim/momentum_sgd.py
open_titans/optim/muon.py
open_titans/optim/nested_learning.py
open_titans/optim/test_time_optim.py
open_titans/trainer/__init__.py
open_titans/trainer/trainer_chronos.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_memory_update_rule.py
tests/test_miras.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