README.md
pyproject.toml
src/llm_dot_py.egg-info/PKG-INFO
src/llm_dot_py.egg-info/SOURCES.txt
src/llm_dot_py.egg-info/dependency_links.txt
src/llm_dot_py.egg-info/requires.txt
src/llm_dot_py.egg-info/top_level.txt
src/llm_py/__init__.py
src/llm_py/component.py
src/llm_py/configs.py
src/llm_py/model.py
src/llm_py/components/__init__.py
src/llm_py/components/activation/__init__.py
src/llm_py/components/activation/swiglu.py
src/llm_py/components/attention/__init__.py
src/llm_py/components/attention/cross_attention.py
src/llm_py/components/attention/gqa.py
src/llm_py/components/attention/mqa.py
src/llm_py/components/attention/self_attention.py
src/llm_py/components/common/residual.py
src/llm_py/components/embedding/__init__.py
src/llm_py/components/embedding/embedding.py
src/llm_py/components/ffn/__init__.py
src/llm_py/components/ffn/feed_forward.py
src/llm_py/components/heads/__init__.py
src/llm_py/components/heads/lm_head.py
src/llm_py/components/heads/moe_head.py
src/llm_py/components/norm/__init__.py
src/llm_py/components/norm/layer_norm.py
src/llm_py/components/norm/rms_norm.py
src/llm_py/components/pos/__init__.py
src/llm_py/components/pos/alibi.py
src/llm_py/components/pos/learned.py
src/llm_py/components/pos/relative.py
src/llm_py/components/pos/rotary.py
src/llm_py/components/pos/sinusoidal.py
tests/test_generation.py