LICENSE
README.md
pyproject.toml
src/fabrique/__init__.py
src/fabrique/_main.py
src/fabrique/export.py
src/fabrique/gcp_utils.py
src/fabrique/inspection.py
src/fabrique/loading.py
src/fabrique/lora.py
src/fabrique/sampling.py
src/fabrique/tokenizer_utils.py
src/fabrique/training.py
src/fabrique/utils.py
src/fabrique.egg-info/PKG-INFO
src/fabrique.egg-info/SOURCES.txt
src/fabrique.egg-info/dependency_links.txt
src/fabrique.egg-info/requires.txt
src/fabrique.egg-info/top_level.txt
src/fabrique/models/__init__.py
src/fabrique/models/gemma/__init__.py
src/fabrique/models/gemma/_main.py
src/fabrique/models/gemma/layers.py
src/fabrique/models/gemma/load_rules.py
src/fabrique/models/gemma/loader.py
src/fabrique/models/gemma/modeling.py
src/fabrique/models/gemma/modules.py
src/fabrique/models/gemma/vision.py
src/fabrique/models/gemma/vision_utils.py
src/gemma/__init__.py
src/gemma/gm/__init__.py
src/gemma/gm/ckpts/__init__.py
src/gemma/gm/ckpts/_checkpoint.py
src/gemma/gm/ckpts/_compat.py
src/gemma/gm/ckpts/_lora.py
src/gemma/gm/ckpts/_paths.py
src/gemma/gm/ckpts/_policy.py
src/gemma/gm/ckpts/_quantization.py
src/gemma/gm/data/__init__.py
src/gemma/gm/data/_functional.py
src/gemma/gm/data/_parquet.py
src/gemma/gm/data/_tasks.py
src/gemma/gm/data/_transforms.py
src/gemma/gm/evals/__init__.py
src/gemma/gm/evals/_sample.py
src/gemma/gm/losses/__init__.py
src/gemma/gm/losses/_dpo.py
src/gemma/gm/math/__init__.py
src/gemma/gm/math/_pos_utils.py
src/gemma/gm/math/_positional_embeddings.py
src/gemma/gm/nn/__init__.py
src/gemma/gm/nn/_config.py
src/gemma/gm/nn/_experimental.py
src/gemma/gm/nn/_gemma.py
src/gemma/gm/nn/_layers.py
src/gemma/gm/nn/_lora.py
src/gemma/gm/nn/_modules.py
src/gemma/gm/nn/_policy.py
src/gemma/gm/nn/_quantization.py
src/gemma/gm/nn/_transformer.py
src/gemma/gm/nn/config.py
src/gemma/gm/nn/gemma3n/__init__.py
src/gemma/gm/nn/gemma3n/_config.py
src/gemma/gm/nn/gemma3n/_gemma3n.py
src/gemma/gm/nn/gemma3n/_layers.py
src/gemma/gm/nn/gemma3n/_modules.py
src/gemma/gm/nn/gemma3n/_transformer.py
src/gemma/gm/sharding/__init__.py
src/gemma/gm/testing/__init__.py
src/gemma/gm/testing/_dummy_model.py
src/gemma/gm/testing/_dummy_tokenizer.py
src/gemma/gm/testing/_fixtures.py
src/gemma/gm/text/__init__.py
src/gemma/gm/text/_chat_sampler.py
src/gemma/gm/text/_prefill.py
src/gemma/gm/text/_sampler.py
src/gemma/gm/text/_sampler_loop.py
src/gemma/gm/text/_sampling.py
src/gemma/gm/text/_template.py
src/gemma/gm/text/_tokenizer.py
src/gemma/gm/text/_tool_sampler.py
src/gemma/gm/text/_turn_utils.py
src/gemma/gm/tools/__init__.py
src/gemma/gm/tools/_calculator.py
src/gemma/gm/tools/_file_explorer.py
src/gemma/gm/tools/_manager.py
src/gemma/gm/tools/_offline_tool_search.py
src/gemma/gm/tools/_tools.py
src/gemma/gm/typing/__init__.py
src/gemma/gm/typing/_common.py
src/gemma/gm/utils/__init__.py
src/gemma/gm/utils/_attention_mask.py
src/gemma/gm/utils/_cache_helper.py
src/gemma/gm/utils/_dtype_params.py
src/gemma/gm/utils/_jax_utils.py
src/gemma/gm/utils/_py.py
src/gemma/gm/utils/_types.py
src/gemma/gm/vision/_preprocess.py
src/gemma/gm/vision/_token_utils.py
src/gemma/multimodal/image.py
src/gemma/multimodal/vision.py
src/gemma/multimodal/vision_utils.py
src/gemma/peft/__init__.py
src/gemma/peft/_einsum_utils.py
src/gemma/peft/_interceptors.py
src/gemma/peft/_lora.py
src/gemma/peft/_quantization.py
src/gemma/peft/_quantization_utils.py
src/gemma/peft/_tree_utils.py
src/gemma/research/t5gemma/__init__.py
src/gemma/research/t5gemma/config.py
src/gemma/research/t5gemma/modules.py
src/gemma/research/t5gemma/sampling.py
src/gemma/research/t5gemma/t5gemma.py
tests/test_lora.py
tests/test_training.py
tests/test_utils.py