huggingface / optimum-neuron
File Size

The distribution of size of files (measured in lines of code).

Intro
Learn more...
File Size Overall
22% | 17% | 36% | 13% | 9%
Legend:
1001+
501-1000
201-500
101-200
1-100


explore: grouped by folders | grouped by size | sunburst | 3D view
File Size per Extension
1001+
501-1000
201-500
101-200
1-100
py24% | 19% | 32% | 14% | 10%
ipynb0% | 0% | 93% | 6% | 0%
hcl0% | 0% | 0% | 0% | 100%
toml0% | 0% | 0% | 0% | 100%
in0% | 0% | 0% | 0% | 100%
File Size per Logical Decomposition
primary
1001+
501-1000
201-500
101-200
1-100
optimum25% | 20% | 31% | 13% | 8%
notebooks0% | 0% | 83% | 10% | 6%
tools0% | 0% | 69% | 20% | 10%
ROOT0% | 0% | 0% | 66% | 33%
benchmark0% | 0% | 0% | 0% | 100%
infrastructure0% | 0% | 0% | 0% | 100%
Longest Files (Top 50)
File# lines# units
trainers.py
in optimum/neuron
1607 43
modeling_utils.py
in optimum/neuron/models/training
1364 30
transformations_utils.py
in optimum/neuron/models/training
1293 58
modeling_diffusion.py
in optimum/neuron
1260 50
utils.py
in optimum/neuron/generation
1056 16
model_configs.py
in optimum/exporters/neuron
896 74
modeling_decoder.py
in optimum/neuron/models/inference/backend/modules/decoder
759 33
modeling_llama.py
in optimum/neuron/models/training/llama
691 32
__main__.py
in optimum/exporters/neuron
685 16
gqa.py
in optimum/neuron/models/inference/backend/modules/attention
651 24
model_wrappers.py
in optimum/exporters/neuron
567 26
modeling.py
in optimum/neuron
526 17
modeling_seq2seq.py
in optimum/neuron
501 23
runner.py
in optimum/neuron/utils
494 12
convert.py
in optimum/exporters/neuron
490 8
misc.py
in optimum/neuron/utils
475 14
utils.py
in optimum/exporters/neuron
468 11
llama2-13b-chatbot.ipynb
in notebooks/text-generation
461 -
modeling_traced.py
in optimum/neuron
450 19
deploy-mixtral-8x7b.ipynb
in notebooks/sagemaker
425 -
accelerator.py
in optimum/neuron/accelerate
422 18
layer.py
in optimum/neuron/peft/tuners/lora
406 19
modeling_llama.py
in optimum/neuron/models/inference/llama
396 15
deploy-llama-3-3-70b.ipynb
in notebooks/sagemaker
366 -
pipeline_controlnet_sd_xl.py
in optimum/neuron/pipelines/diffusers
357 1
qwen3-fine-tuning.ipynb
in notebooks/text-generation
350 -
peft_model.py
in optimum/neuron/peft
344 12
fine_tune_bert.ipynb
in notebooks/text-classification
324 -
attention_base.py
in optimum/neuron/models/inference/backend/modules/attention
323 8
base.py
in optimum/neuron/pipelines/transformers
319 3
309 6
CodeLlama-7B-Compilation.ipynb
in notebooks/text-generation
305 -
neuronx.py
in optimum/commands/export
303 4
generation_utils.py
in optimum/neuron/models/inference/backend/modules/generation
298 10
286 7
base.py
in optimum/exporters/neuron
276 20
pipeline_controlnet.py
in optimum/neuron/pipelines/diffusers
268 1
model.py
in optimum/neuron/models/bert
244 6
decoder_wrapper.py
in optimum/neuron/models/inference/backend/modules/decoder
242 16
modeling_granite.py
in optimum/neuron/models/training/granite
236 10
modeling_t5.py
in optimum/neuron/models/inference/t5
234 11
pretrained_model.py
in optimum/neuron/models/inference/backend
234 20
training_args.py
in optimum/neuron
229 9
argument_utils.py
in optimum/neuron/utils
228 10
sampling.py
in optimum/neuron/models/inference/backend/modules/generation
220 16
input_generators.py
in optimum/neuron/utils
207 11
configuration_utils.py
in optimum/neuron
198 10
cache_utils.py
in optimum/neuron/utils
195 14
cache.py
in optimum/commands/neuron
189 11
utils.py
in optimum/neuron/models/inference/backend/modules/attention
188 16
Files With Most Units (Top 50)
File# lines# units
model_configs.py
in optimum/exporters/neuron
896 74
transformations_utils.py
in optimum/neuron/models/training
1293 58
modeling_diffusion.py
in optimum/neuron
1260 50
trainers.py
in optimum/neuron
1607 43
modeling_decoder.py
in optimum/neuron/models/inference/backend/modules/decoder
759 33
modeling_llama.py
in optimum/neuron/models/training/llama
691 32
modeling_utils.py
in optimum/neuron/models/training
1364 30
model_wrappers.py
in optimum/exporters/neuron
567 26
gqa.py
in optimum/neuron/models/inference/backend/modules/attention
651 24
modeling_seq2seq.py
in optimum/neuron
501 23
base.py
in optimum/exporters/neuron
276 20
pretrained_model.py
in optimum/neuron/models/inference/backend
234 20
layer.py
in optimum/neuron/peft/tuners/lora
406 19
modeling_traced.py
in optimum/neuron
450 19
accelerator.py
in optimum/neuron/accelerate
422 18
modeling.py
in optimum/neuron
526 17
__main__.py
in optimum/exporters/neuron
685 16
utils.py
in optimum/neuron/generation
1056 16
decoder_wrapper.py
in optimum/neuron/models/inference/backend/modules/decoder
242 16
sampling.py
in optimum/neuron/models/inference/backend/modules/generation
220 16
utils.py
in optimum/neuron/models/inference/backend/modules/attention
188 16
modeling_llama.py
in optimum/neuron/models/inference/llama
396 15
cache_utils.py
in optimum/neuron/utils
195 14
patching.py
in optimum/neuron/utils
155 14
misc.py
in optimum/neuron/utils
475 14
pipeline_utils.py
in optimum/neuron/models/training
121 13
peft_model.py
in optimum/neuron/peft
344 12
runner.py
in optimum/neuron/utils
494 12
model.py
in optimum/neuron/models/whisper
173 12
cache.py
in optimum/commands/neuron
189 11
utils.py
in optimum/exporters/neuron
468 11
training_utils.py
in optimum/neuron/utils
131 11
input_generators.py
in optimum/neuron/utils
207 11
modeling_t5.py
in optimum/neuron/models/inference/t5
234 11
kv_cache_manager.py
in optimum/neuron/models/inference/backend/modules/kvcache
170 11
argument_utils.py
in optimum/neuron/utils
228 10
modeling_granite.py
in optimum/neuron/models/training/granite
236 10
checkpoint.py
in optimum/neuron/models/inference/backend/modules
99 10
generation_utils.py
in optimum/neuron/models/inference/backend/modules/generation
298 10
modeling_mixtral.py
in optimum/neuron/models/inference/mixtral
184 10
configuration_utils.py
in optimum/neuron
198 10
training_args.py
in optimum/neuron
229 9
173 9
convert.py
in optimum/exporters/neuron
490 8
import_utils.py
in optimum/neuron/utils
52 8
attention_base.py
in optimum/neuron/models/inference/backend/modules/attention
323 8
modeling_decoder.py
in optimum/neuron
170 8
version_utils.py
in optimum/neuron/utils
84 7
modeling_qwen3.py
in optimum/neuron/models/training/qwen3
181 7
modeling_granite.py
in optimum/neuron/models/inference/granite
118 7
Files With Long Lines (Top 45)

There are 45 files with lines longer than 120 characters. In total, there are 299 long lines.

File# lines# units# long lines
llama2-13b-chatbot.ipynb
in notebooks/text-generation
461 - 29
deploy-llama-3-3-70b.ipynb
in notebooks/sagemaker
366 - 26
deploy-mixtral-8x7b.ipynb
in notebooks/sagemaker
425 - 26
CodeLlama-7B-Compilation.ipynb
in notebooks/text-generation
305 - 21
fine_tune_bert.ipynb
in notebooks/text-classification
324 - 21
trainers.py
in optimum/neuron
1607 43 20
modeling_utils.py
in optimum/neuron/models/training
1364 30 20
qwen3-fine-tuning.ipynb
in notebooks/text-generation
350 - 18
neuronx.py
in optimum/commands/export
303 4 17
neuron.py
in optimum/commands/export
133 4 12
convert.py
in optimum/exporters/neuron
490 8 8
getting-started.ipynb
in notebooks/sentence-transformers
154 - 7
__main__.py
in optimum/exporters/neuron
685 16 5
utils.py
in optimum/exporters/neuron
468 11 5
transformations_utils.py
in optimum/neuron/models/training
1293 58 5
accelerator.py
in optimum/neuron/accelerate
422 18 5
modeling_diffusion.py
in optimum/neuron
1260 50 4
utils.py
in optimum/neuron/generation
1056 16 4
286 7 4
runner.py
in optimum/neuron/utils
494 12 3
version_utils.py
in optimum/neuron/utils
84 7 3
modeling_traced.py
in optimum/neuron
450 19 3
decoder_wrapper.py
in optimum/neuron/models/inference/backend/modules/decoder
242 16 3
model_wrappers.py
in optimum/exporters/neuron
567 26 2
base.py
in optimum/neuron/pipelines/transformers
319 3 2
token_selector.py
in optimum/neuron/generation
132 4 2
training_args.py
in optimum/neuron
229 9 2
sampling.py
in optimum/neuron/models/inference/backend/modules/generation
220 16 2
generation_utils.py
in optimum/neuron/models/inference/backend/modules/generation
298 10 2
modeling_llama.py
in optimum/neuron/models/inference/llama
396 15 2
misc.py
in optimum/neuron/accelerate/utils
121 7 2
base.py
in optimum/exporters/neuron
276 20 1
modeling_seq2seq.py
in optimum/neuron
501 23 1
pipeline_controlnet_sd_xl.py
in optimum/neuron/pipelines/diffusers
357 1 1
pipeline_utils.py
in optimum/neuron/pipelines/diffusers
49 3 1
cache_utils.py
in optimum/neuron/utils
195 14 1
misc.py
in optimum/neuron/utils
475 14 1
hf_argparser.py
in optimum/neuron
42 5 1
config.py
in optimum/neuron/models/inference/backend
138 6 1
gqa.py
in optimum/neuron/models/inference/backend/modules/attention
651 24 1
modeling_mixtral.py
in optimum/neuron/models/inference/mixtral
184 10 1
auto_model.py
in optimum/neuron/models
17 3 1
modeling_decoder.py
in optimum/neuron
170 8 1
configuration_utils.py
in optimum/neuron
198 10 1
173 9 1