Path	Lines of Code
MANIFEST.in	15
optimum/tpu/__init__.py	3
optimum/tpu/cli.py	76
optimum/tpu/distributed_model.py	79
optimum/tpu/fsdp_v2.py	43
optimum/tpu/generation/__init__.py	2
optimum/tpu/generation/logits_process.py	34
optimum/tpu/generation/token_selector.py	116
optimum/tpu/jetstream_pt_support.py	11
optimum/tpu/model.py	43
optimum/tpu/modeling.py	47
optimum/tpu/modeling_gemma.py	877
optimum/tpu/modeling_llama.py	1124
optimum/tpu/modeling_mistral.py	1106
optimum/tpu/static_cache_xla.py	23
optimum/tpu/version.py	3
optimum/tpu/xla_logger.py	14
optimum/tpu/xla_model_parallel.py	548
optimum/tpu/xla_mp_comm.py	39
pyproject.toml	98
setup.cfg	15
text-generation-inference/Cargo.toml	42
text-generation-inference/server/pyproject.toml	32
text-generation-inference/server/text_generation_server/auto_generator.py	26
text-generation-inference/server/text_generation_server/cli.py	78
text-generation-inference/server/text_generation_server/generator.py	681
text-generation-inference/server/text_generation_server/generator_base.py	25
text-generation-inference/server/text_generation_server/interceptor.py	23
text-generation-inference/server/text_generation_server/jetstream_pt_support/__init__.py	1
text-generation-inference/server/text_generation_server/jetstream_pt_support/compatibility.py	24
text-generation-inference/server/text_generation_server/jetstream_pt_support/engine_loader.py	182
text-generation-inference/server/text_generation_server/jetstream_pt_support/generator.py	439
text-generation-inference/server/text_generation_server/jetstream_pt_support/logits_process.py	33
text-generation-inference/server/text_generation_server/jetstream_pt_support/models/__init__.py	3
text-generation-inference/server/text_generation_server/jetstream_pt_support/models/gemma_model_hf.py	23
text-generation-inference/server/text_generation_server/jetstream_pt_support/models/llama_model_exportable_hf.py	102
text-generation-inference/server/text_generation_server/jetstream_pt_support/models/mixtral_model_hf.py	41
text-generation-inference/server/text_generation_server/jetstream_pt_support/token_selector.py	135
text-generation-inference/server/text_generation_server/server.py	89
text-generation-inference/server/text_generation_server/version.py	3