LICENSE.md
README.md
pyproject.toml
tamm/__init__.py
tamm/_compat.py
tamm/_helpers.py
tamm/_logger.py
tamm/_version.py
tamm/_version_guidance.py
tamm/_warnings.py
tamm/runtime_configuration.py
tamm/typing.py
tamm.egg-info/PKG-INFO
tamm.egg-info/SOURCES.txt
tamm.egg-info/dependency_links.txt
tamm.egg-info/entry_points.txt
tamm.egg-info/requires.txt
tamm.egg-info/top_level.txt
tamm/_adapters_v1/__init__.py
tamm/_adapters_v1/adapted_layer.py
tamm/_adapters_v1/adapter_api.py
tamm/_adapters_v1/layer_annotations.py
tamm/_adapters_v1/shared_context.py
tamm/_adapters_v1/utils.py
tamm/_adapters_v1/layer_adapters/__init__.py
tamm/_adapters_v1/layer_adapters/average.py
tamm/_adapters_v1/layer_adapters/bottleneck.py
tamm/_adapters_v1/layer_adapters/common.py
tamm/_adapters_v1/layer_adapters/lora.py
tamm/_adapters_v1/layer_adapters/prefix_tuning.py
tamm/_adapters_v1/layer_adapters/soft_mixing_lora.py
tamm/_adapters_v1/layer_adapters/stack.py
tamm/_adapters_v1/model_adapters/__init__.py
tamm/_adapters_v1/model_adapters/bottleneck.py
tamm/_adapters_v1/model_adapters/composition.py
tamm/_adapters_v1/model_adapters/lora.py
tamm/_adapters_v1/model_adapters/model_adapter.py
tamm/_cli/__init__.py
tamm/_cli/__main__.py
tamm/_cli/clean.py
tamm/_cli/common.py
tamm/_cli/from_tamm.py
tamm/_cli/ls.py
tamm/_cli/show.py
tamm/_cli/to_tamm.py
tamm/_cli/version.py
tamm/_cli/_listers/__init__.py
tamm/_cli/_listers/_base.py
tamm/_cli/_listers/_helpers.py
tamm/_cli/_listers/converters.py
tamm/_cli/_listers/models.py
tamm/_cli/_listers/preprocessors.py
tamm/_cli/_listers/tokenizers.py
tamm/_plugin/__init__.py
tamm/_plugin/cli.py
tamm/_plugin/core_import_callback.py
tamm/_plugin/core_transformers_import_callback.py
tamm/_plugin/utils.py
tamm/adapters/__init__.py
tamm/adapters/utils/__init__.py
tamm/ao/__init__.py
tamm/ao/utils.py
tamm/ao/arch_optimizers/__init__.py
tamm/ao/arch_optimizers/common.py
tamm/ao/arch_optimizers/kv_quant.py
tamm/ao/layers/__init__.py
tamm/ao/layers/fake_quantize.py
tamm/ao/layers/functional.py
tamm/ao/layers/kv_quantizer.py
tamm/ao/layers/quantizing_kv_cacher.py
tamm/context_vars/__init__.py
tamm/context_vars/_first_token_generation.py
tamm/context_vars/_freeze_params.py
tamm/context_vars/_model_build_device.py
tamm/context_vars/_plugin_init.py
tamm/context_vars/_pretrained.py
tamm/context_vars/_resolvers.py
tamm/converters/__init__.py
tamm/converters/common.py
tamm/converters/layers/__init__.py
tamm/converters/models/__init__.py
tamm/converters/models/common.py
tamm/converters/models/from_tamm.py
tamm/converters/models/registry.py
tamm/converters/models/to_tamm.py
tamm/generation/__init__.py
tamm/generation/token_generators/__init__.py
tamm/generation/token_generators/_vanilla.py
tamm/generation/token_generators/base.py
tamm/generation/token_generators/greedy.py
tamm/generation/token_generators/next_token_logic.py
tamm/generation/token_generators/sampling.py
tamm/generation/token_generators/speculative.py
tamm/generation/token_generators/utils.py
tamm/hf/__init__.py
tamm/hf/transformers/__init__.py
tamm/hf/transformers/configuration_utils.py
tamm/hf/transformers/modeling_utils.py
tamm/hf/transformers/models/__init__.py
tamm/hf/transformers/models/_mixins.py
tamm/hf/transformers/models/auto.py
tamm/hf/transformers/models/registry.py
tamm/hf/transformers/models/causal_lm/__init__.py
tamm/hf/transformers/models/causal_lm/configuration_causal_lm.py
tamm/hf/transformers/models/causal_lm/kv_cache.py
tamm/hf/transformers/models/causal_lm/modeling_causal_lm.py
tamm/hf/transformers/models/sentencepiece/__init__.py
tamm/hf/transformers/models/sentencepiece/configuration_sentencepiece.py
tamm/hf/transformers/models/sentencepiece/tokenization_sentencepiece.py
tamm/layers/__init__.py
tamm/layers/activation.py
tamm/layers/attention.py
tamm/layers/basic.py
tamm/layers/convolution.py
tamm/layers/decoding.py
tamm/layers/dropout.py
tamm/layers/embedding.py
tamm/layers/feed_forward.py
tamm/layers/flash_attention.py
tamm/layers/functional.py
tamm/layers/init.py
tamm/layers/lambda_layer.py
tamm/layers/linear.py
tamm/layers/loss.py
tamm/layers/norm.py
tamm/layers/pooler.py
tamm/layers/positional_encoding.py
tamm/layers/residual.py
tamm/layers/rope.py
tamm/layers/segmentation.py
tamm/layers/sequential.py
tamm/layers/side_outputs.py
tamm/layers/torch_nn.py
tamm/layers/common/__init__.py
tamm/layers/common/_deprecated.py
tamm/layers/common/_marker.py
tamm/layers/common/builder.py
tamm/layers/common/config.py
tamm/layers/common/mixins.py
tamm/layers/common/pretrained_loader.py
tamm/layers/common/context_hooks/__init__.py
tamm/layers/common/context_hooks/base.py
tamm/layers/common/context_hooks/default_device.py
tamm/layers/common/context_hooks/dtype.py
tamm/layers/common/context_hooks/freeze_params.py
tamm/layers/common/context_hooks/pretrained.py
tamm/layers/common/context_hooks/use_meta_init_trick.py
tamm/layers/common/post_hooks/__init__.py
tamm/layers/common/post_hooks/adapters.py
tamm/layers/common/post_hooks/arch_optimizer.py
tamm/layers/common/post_hooks/attach_config.py
tamm/layers/common/post_hooks/attach_metadata.py
tamm/layers/common/post_hooks/common.py
tamm/layers/common/post_hooks/freeze_params.py
tamm/layers/common/post_hooks/model_initializer.py
tamm/layers/mixture_of_experts/__init__.py
tamm/layers/mixture_of_experts/loss.py
tamm/layers/mixture_of_experts/mixture_of_experts.py
tamm/layers/mixture_of_experts/router.py
tamm/layers/mixture_of_experts/dispatcher/__init__.py
tamm/layers/mixture_of_experts/dispatcher/capacity_constrained.py
tamm/layers/mixture_of_experts/dispatcher/capacity_constraint.py
tamm/layers/mixture_of_experts/dispatcher/common.py
tamm/layers/mixture_of_experts/dispatcher/dropless.py
tamm/layers/mixture_of_experts/sampler/__init__.py
tamm/layers/mixture_of_experts/sampler/common.py
tamm/layers/mixture_of_experts/sampler/top_k.py
tamm/layers/transformer/__init__.py
tamm/layers/transformer/attention_mask.py
tamm/layers/transformer/layer.py
tamm/layers/transformer/positional_encoding.py
tamm/layers/transformer/stack.py
tamm/layers/transformer/text.py
tamm/layers/transformer/token_type_encoding.py
tamm/layers/transformer/vision.py
tamm/layers/transformer/kv_cache/__init__.py
tamm/layers/transformer/kv_cache/common.py
tamm/layers/transformer/kv_cache/speculative.py
tamm/layers/transformer/kv_cache/v0.py
tamm/layers/transformer/kv_cache/vanilla.py
tamm/layers/transformer/layer_sequence/__init__.py
tamm/layers/transformer/layer_sequence/common.py
tamm/layers/transformer/layer_sequence/kv_reuse.py
tamm/layers/transformer/layer_sequence/uniform.py
tamm/layers/transformer/layer_sequence/vanilla.py
tamm/layers/transformer/token_metadata/__init__.py
tamm/layers/transformer/token_metadata/common.py
tamm/layers/transformer/token_metadata/vanilla.py
tamm/model_repo/__init__.py
tamm/model_repo/_warning.py
tamm/model_repo/api.py
tamm/model_repo/exceptions.py
tamm/model_repo/file.py
tamm/model_repo/model_repo.py
tamm/model_repo/registry.py
tamm/model_repo/uri_handler.py
tamm/model_repo/utils.py
tamm/model_repo/publishing/__init__.py
tamm/model_repo/publishing/base.py
tamm/model_repo/publishing/config.py
tamm/models/__init__.py
tamm/models/configs.py
tamm/models/vision_transformer.py
tamm/models/afm_text/__init__.py
tamm/models/afm_text/afm_text_v7.py
tamm/models/checksum/__init__.py
tamm/models/checksum/_checksum.py
tamm/models/checksum/api.py
tamm/models/common/__init__.py
tamm/models/common/config.py
tamm/models/common/metadata.py
tamm/models/common/mixin.py
tamm/preprocessors/__init__.py
tamm/preprocessors/api.py
tamm/preprocessors/base.py
tamm/preprocessors/instruct_lm/__init__.py
tamm/preprocessors/instruct_lm/common.py
tamm/preprocessors/instruct_lm/jinja.py
tamm/preprocessors/instruct_lm/afm/__init__.py
tamm/preprocessors/instruct_lm/afm/afm_v6.py
tamm/preprocessors/instruct_lm/afm/utils.py
tamm/preprocessors/registry/__init__.py
tamm/preprocessors/registry/base.py
tamm/preprocessors/registry/uri_handler.py
tamm/tokenizers/__init__.py
tamm/tokenizers/afm.py
tamm/tokenizers/api.py
tamm/tokenizers/common.py
tamm/tokenizers/sentencepiece.py
tamm/tokenizers/text_processors.py
tamm/tokenizers/registry/__init__.py
tamm/tokenizers/registry/base.py
tamm/tokenizers/registry/placeholder.py
tamm/tokenizers/registry/uri_handler.py
tamm/utils/__init__.py
tamm/utils/_pretrained.py
tamm/utils/_torch_compatibility.py
tamm/utils/axlearn_utils.py
tamm/utils/optional_bool.py
tamm/utils/partial.py
tamm/utils/registry.py
tamm/utils/transformers_utils.py
tamm/utils/user_dir_utils.py
tamm/utils/json/__init__.py
tamm/utils/json/_hooks.py
tamm/utils/json/_normalization.py
tamm/utils/json/_utils.py
tamm/utils/json/api.py
tamm/utils/torch_utils/__init__.py
tamm/utils/torch_utils/exportable_dataclass.py
tamm/utils/torch_utils/itertools.py
tamm/utils/torch_utils/num_params.py
tamm/utils/torch_utils/tieable_params.py
tamm/utils/uri/__init__.py
tamm/utils/uri/_helpers.py
tamm/utils/uri/_protocol.py
tamm/utils/uri/uri_handler.py
tamm/utils/vision_utils/__init__.py
tamm/utils/vision_utils/_load.py