.coveragerc
.gitignore
.pre-commit-config.yaml
LICENSE
README.md
pyproject.toml
.github/FUNDING.yml
.github/disabled-workflows/test-slow.yaml
.github/workflows/publish.yaml
.github/workflows/test.yaml
doc/benchmarks.png
doc/impossible-triangle.jpeg
doc/inference-memory.png
doc/inference-throughput.png
doc/retention-dual-forms.jpeg
doc/retnet-scaling.jpeg
scripts/benchmark_inference.py
scripts/train_project_gutenberg.py
tests/__init__.py
tests/conftest.py
tests/test_retention.py
tests/test_retnet.py
yet_another_retnet/__init__.py
yet_another_retnet/retention.py
yet_another_retnet/retnet.py
yet_another_retnet.egg-info/PKG-INFO
yet_another_retnet.egg-info/SOURCES.txt
yet_another_retnet.egg-info/dependency_links.txt
yet_another_retnet.egg-info/requires.txt
yet_another_retnet.egg-info/top_level.txt
yet_another_retnet/utils/__init__.py
yet_another_retnet/utils/benchmark.py
yet_another_retnet/utils/gutenberg.py
yet_another_retnet/utils/profile.py