├── .github └── workflows │ ├── bump-version-release.yml │ ├── post-merge-validate-and-draft-release.yml │ └── release-publish.yml ├── .gitignore ├── .python-version ├── .rules ├── .vscode ├── launch.json └── settings.json ├── LICENSE ├── README.md ├── lalamo ├── __init__.py ├── common.py ├── data │ ├── __init__.py │ ├── huggingface_message.py │ ├── lalamo_completions.py │ └── utils.py ├── main.py ├── message_processor.py ├── model_import │ ├── __init__.py │ ├── common.py │ ├── decoder_configs │ │ ├── __init__.py │ │ ├── common.py │ │ ├── executorch.py │ │ └── huggingface │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── gemma2.py │ │ │ ├── gemma3.py │ │ │ ├── gpt_oss.py │ │ │ ├── llama.py │ │ │ ├── llamba.py │ │ │ ├── mistral.py │ │ │ ├── modern_bert.py │ │ │ ├── qwen2.py │ │ │ └── qwen3.py │ ├── huggingface_generation_config.py │ ├── huggingface_tokenizer_config.py │ ├── loaders │ │ ├── __init__.py │ │ ├── common.py │ │ ├── executorch.py │ │ ├── huggingface.py │ │ └── utils.py │ └── model_specs │ │ ├── __init__.py │ │ ├── common.py │ │ ├── deepseek.py │ │ ├── gemma.py │ │ ├── gpt_oss.py │ │ ├── huggingface.py │ │ ├── llama.py │ │ ├── llamba.py │ │ ├── mirai.py │ │ ├── mistral.py │ │ ├── pleias.py │ │ ├── polaris.py │ │ ├── qwen.py │ │ └── reka.py ├── models │ ├── __init__.py │ ├── common.py │ ├── language_model.py │ └── router.py ├── modules │ ├── __init__.py │ ├── activations.py │ ├── classifier.py │ ├── common.py │ ├── decoder.py │ ├── embedding.py │ ├── linear.py │ ├── mlp.py │ ├── mlx_interop.py │ ├── normalization.py │ ├── rope.py │ ├── token_mixers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── common.py │ │ ├── mamba.py │ │ └── state │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── kv_cache.py │ │ │ └── mamba_state.py │ ├── torch_interop.py │ ├── transformer.py │ ├── transformer_layer.py │ └── utils.py ├── quantization.py ├── registry_abc.py ├── sampling.py ├── speculator │ ├── __init__.py │ ├── common.py │ ├── estimator.py │ ├── inference.py │ ├── ngram.py │ └── utils.py └── utils.py ├── pyproject.toml ├── tests ├── __init__.py ├── cartesia_mlx_tracer.py ├── common.py ├── executorch_llama │ ├── LICENSE │ ├── __init__.py │ ├── rope.py │ ├── source_transformation │ │ ├── __init__.py │ │ ├── lora.py │ │ ├── pre_quantization.py │ │ └── quantize.py │ └── transformer.py ├── huggingface_tracer.py ├── mlx_tracer.py ├── test_cartesia_mlx_models.py ├── test_generation.py ├── test_huggingface_model_conversion.py ├── test_huggingface_models.py ├── test_mlx_models.py ├── test_model_spec.py ├── test_models.py ├── test_moe.py ├── test_parameter_tree.py └── test_registry_abc.py └── uv.lock /.github/workflows/bump-version-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.github/workflows/bump-version-release.yml -------------------------------------------------------------------------------- /.github/workflows/post-merge-validate-and-draft-release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.github/workflows/post-merge-validate-and-draft-release.yml -------------------------------------------------------------------------------- /.github/workflows/release-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.github/workflows/release-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.gitignore -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.12 2 | -------------------------------------------------------------------------------- /.rules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.rules -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/README.md -------------------------------------------------------------------------------- /lalamo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/__init__.py -------------------------------------------------------------------------------- /lalamo/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/common.py -------------------------------------------------------------------------------- /lalamo/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/data/__init__.py -------------------------------------------------------------------------------- /lalamo/data/huggingface_message.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/data/huggingface_message.py -------------------------------------------------------------------------------- /lalamo/data/lalamo_completions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/data/lalamo_completions.py -------------------------------------------------------------------------------- /lalamo/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/data/utils.py -------------------------------------------------------------------------------- /lalamo/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/main.py -------------------------------------------------------------------------------- /lalamo/message_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/message_processor.py -------------------------------------------------------------------------------- /lalamo/model_import/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/__init__.py -------------------------------------------------------------------------------- /lalamo/model_import/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/common.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/__init__.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/common.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/executorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/executorch.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/__init__.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/common.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/gemma2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/gemma2.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/gemma3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/gemma3.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/gpt_oss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/gpt_oss.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/llama.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/llamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/llamba.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/mistral.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/modern_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/modern_bert.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/qwen2.py -------------------------------------------------------------------------------- /lalamo/model_import/decoder_configs/huggingface/qwen3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/decoder_configs/huggingface/qwen3.py -------------------------------------------------------------------------------- /lalamo/model_import/huggingface_generation_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/huggingface_generation_config.py -------------------------------------------------------------------------------- /lalamo/model_import/huggingface_tokenizer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/huggingface_tokenizer_config.py -------------------------------------------------------------------------------- /lalamo/model_import/loaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/loaders/__init__.py -------------------------------------------------------------------------------- /lalamo/model_import/loaders/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/loaders/common.py -------------------------------------------------------------------------------- /lalamo/model_import/loaders/executorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/loaders/executorch.py -------------------------------------------------------------------------------- /lalamo/model_import/loaders/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/loaders/huggingface.py -------------------------------------------------------------------------------- /lalamo/model_import/loaders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/loaders/utils.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/__init__.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/common.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/deepseek.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/deepseek.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/gemma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/gemma.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/gpt_oss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/gpt_oss.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/huggingface.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/llama.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/llamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/llamba.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/mirai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/mirai.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/mistral.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/pleias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/pleias.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/polaris.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/polaris.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/qwen.py -------------------------------------------------------------------------------- /lalamo/model_import/model_specs/reka.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/model_import/model_specs/reka.py -------------------------------------------------------------------------------- /lalamo/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/models/__init__.py -------------------------------------------------------------------------------- /lalamo/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/models/common.py -------------------------------------------------------------------------------- /lalamo/models/language_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/models/language_model.py -------------------------------------------------------------------------------- /lalamo/models/router.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/models/router.py -------------------------------------------------------------------------------- /lalamo/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/__init__.py -------------------------------------------------------------------------------- /lalamo/modules/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/activations.py -------------------------------------------------------------------------------- /lalamo/modules/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/classifier.py -------------------------------------------------------------------------------- /lalamo/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/common.py -------------------------------------------------------------------------------- /lalamo/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/decoder.py -------------------------------------------------------------------------------- /lalamo/modules/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/embedding.py -------------------------------------------------------------------------------- /lalamo/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/linear.py -------------------------------------------------------------------------------- /lalamo/modules/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/mlp.py -------------------------------------------------------------------------------- /lalamo/modules/mlx_interop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/mlx_interop.py -------------------------------------------------------------------------------- /lalamo/modules/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/normalization.py -------------------------------------------------------------------------------- /lalamo/modules/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/rope.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/__init__.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/attention.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/common.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/mamba.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/state/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/state/__init__.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/state/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/state/common.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/state/kv_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/state/kv_cache.py -------------------------------------------------------------------------------- /lalamo/modules/token_mixers/state/mamba_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/token_mixers/state/mamba_state.py -------------------------------------------------------------------------------- /lalamo/modules/torch_interop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/torch_interop.py -------------------------------------------------------------------------------- /lalamo/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/transformer.py -------------------------------------------------------------------------------- /lalamo/modules/transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/transformer_layer.py -------------------------------------------------------------------------------- /lalamo/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/modules/utils.py -------------------------------------------------------------------------------- /lalamo/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/quantization.py -------------------------------------------------------------------------------- /lalamo/registry_abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/registry_abc.py -------------------------------------------------------------------------------- /lalamo/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/sampling.py -------------------------------------------------------------------------------- /lalamo/speculator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/__init__.py -------------------------------------------------------------------------------- /lalamo/speculator/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/common.py -------------------------------------------------------------------------------- /lalamo/speculator/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/estimator.py -------------------------------------------------------------------------------- /lalamo/speculator/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/inference.py -------------------------------------------------------------------------------- /lalamo/speculator/ngram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/ngram.py -------------------------------------------------------------------------------- /lalamo/speculator/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/speculator/utils.py -------------------------------------------------------------------------------- /lalamo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/lalamo/utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/cartesia_mlx_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/cartesia_mlx_tracer.py -------------------------------------------------------------------------------- /tests/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/common.py -------------------------------------------------------------------------------- /tests/executorch_llama/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/LICENSE -------------------------------------------------------------------------------- /tests/executorch_llama/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/executorch_llama/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/rope.py -------------------------------------------------------------------------------- /tests/executorch_llama/source_transformation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/executorch_llama/source_transformation/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/source_transformation/lora.py -------------------------------------------------------------------------------- /tests/executorch_llama/source_transformation/pre_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/source_transformation/pre_quantization.py -------------------------------------------------------------------------------- /tests/executorch_llama/source_transformation/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/source_transformation/quantize.py -------------------------------------------------------------------------------- /tests/executorch_llama/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/executorch_llama/transformer.py -------------------------------------------------------------------------------- /tests/huggingface_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/huggingface_tracer.py -------------------------------------------------------------------------------- /tests/mlx_tracer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/mlx_tracer.py -------------------------------------------------------------------------------- /tests/test_cartesia_mlx_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_cartesia_mlx_models.py -------------------------------------------------------------------------------- /tests/test_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_generation.py -------------------------------------------------------------------------------- /tests/test_huggingface_model_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_huggingface_model_conversion.py -------------------------------------------------------------------------------- /tests/test_huggingface_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_huggingface_models.py -------------------------------------------------------------------------------- /tests/test_mlx_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_mlx_models.py -------------------------------------------------------------------------------- /tests/test_model_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_model_spec.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_moe.py -------------------------------------------------------------------------------- /tests/test_parameter_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_parameter_tree.py -------------------------------------------------------------------------------- /tests/test_registry_abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/tests/test_registry_abc.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trymirai/lalamo/HEAD/uv.lock --------------------------------------------------------------------------------