LICENSE
README.md
pyproject.toml
src/zlynx/__init__.py
src/zlynx/utils.py
src/zlynx.egg-info/PKG-INFO
src/zlynx.egg-info/SOURCES.txt
src/zlynx.egg-info/dependency_links.txt
src/zlynx.egg-info/requires.txt
src/zlynx.egg-info/top_level.txt
src/zlynx/models/__init__.py
src/zlynx/models/base.py
src/zlynx/models/config.py
src/zlynx/models/infer.py
src/zlynx/models/outputs.py
src/zlynx/models/diffusion/dit/config.py
src/zlynx/models/diffusion/dit/model.py
src/zlynx/models/llama/__init__.py
src/zlynx/models/llama/config.py
src/zlynx/models/llama/model.py
src/zlynx/models/transformer/llama/__init__.py
src/zlynx/models/transformer/llama/config.py
src/zlynx/models/transformer/llama/model.py
src/zlynx/modules/__init__.py
src/zlynx/modules/attn.py
src/zlynx/modules/cache.py
src/zlynx/modules/embed.py
src/zlynx/modules/mlp.py
src/zlynx/modules/moe.py
src/zlynx/modules/norm.py
src/zlynx/modules/peft.py
src/zlynx/modules/rope.py
src/zlynx/trainer/__init__.py
src/zlynx/trainer/loss_fn.py
src/zlynx/trainer/optim.py
src/zlynx/trainer/trainer.py
src/zlynx/trainer/trainer_config.py
src/zlynx/trainer/utils.py
tests/test_dit.py
tests/test_dpo.py
tests/test_galore.py
tests/test_generate.py
tests/test_grpo.py
tests/test_load.py
tests/test_peft.py
tests/test_sft.py
tests/test_trainer.py