Path	Lines of Code
src/hyperpod_nemo_adapter/__init__.py	1
src/hyperpod_nemo_adapter/collections/__init__.py	1
src/hyperpod_nemo_adapter/collections/data/__init__.py	6
src/hyperpod_nemo_adapter/collections/data/base.py	122
src/hyperpod_nemo_adapter/collections/data/datasets/__init__.py	7
src/hyperpod_nemo_adapter/collections/data/datasets/dummy_dataset.py	15
src/hyperpod_nemo_adapter/collections/data/datasets/dummy_dpo_dataset.py	27
src/hyperpod_nemo_adapter/collections/data/datasets/hf_dataset.py	72
src/hyperpod_nemo_adapter/collections/data/datasets/hf_dpo_dataset.py	12
src/hyperpod_nemo_adapter/collections/data/dummy_data_module.py	25
src/hyperpod_nemo_adapter/collections/data/dummy_dpo_data_module.py	32
src/hyperpod_nemo_adapter/collections/data/hf_data_module.py	58
src/hyperpod_nemo_adapter/collections/data/hf_dpo_data_module.py	35
src/hyperpod_nemo_adapter/collections/data/hf_image_data_module.py	54
src/hyperpod_nemo_adapter/collections/data/vision_dataset.py	149
src/hyperpod_nemo_adapter/collections/model/__init__.py	1
src/hyperpod_nemo_adapter/collections/model/nlp/__init__.py	9
src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/__init__.py	6
src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/configuration_deepseek.py	99
src/hyperpod_nemo_adapter/collections/model/nlp/custom_models/modeling_deepseek.py	1163
src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_deepseek_model.py	82
src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_llama_model.py	75
src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_mistral_model.py	25
src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_mixtral_model.py	27
src/hyperpod_nemo_adapter/collections/model/nlp/sagemaker_qwen_model.py	25
src/hyperpod_nemo_adapter/collections/model/sagemaker_base_model.py	547
src/hyperpod_nemo_adapter/collections/parts/__init__.py	2
src/hyperpod_nemo_adapter/collections/parts/fsdp_strategy.py	417
src/hyperpod_nemo_adapter/collections/parts/sagemaker_trainer_builder.py	172
src/hyperpod_nemo_adapter/conf/__init__.py	1
src/hyperpod_nemo_adapter/conf/config_schemas.py	339
src/hyperpod_nemo_adapter/constants.py	53
src/hyperpod_nemo_adapter/patches/__init__.py	1
src/hyperpod_nemo_adapter/patches/patch_llama_flash_attn_cp.py	283
src/hyperpod_nemo_adapter/patches/patch_mllama_dtype.py	12
src/hyperpod_nemo_adapter/scripts/casting_utils/kernel.py	104
src/hyperpod_nemo_adapter/scripts/fp8_cast_bf16.py	63
src/hyperpod_nemo_adapter/scripts/merge_peft_checkpoint.py	66
src/hyperpod_nemo_adapter/utils/__init__.py	1
src/hyperpod_nemo_adapter/utils/app_state.py	39
src/hyperpod_nemo_adapter/utils/callbacks/__init__.py	1
src/hyperpod_nemo_adapter/utils/callbacks/base_ckpt_io.py	20
src/hyperpod_nemo_adapter/utils/callbacks/checkpoint.py	414
src/hyperpod_nemo_adapter/utils/callbacks/ckpt_io.py	94
src/hyperpod_nemo_adapter/utils/callbacks/full_ckpt_io.py	14
src/hyperpod_nemo_adapter/utils/callbacks/local_ckpt_io.py	216
src/hyperpod_nemo_adapter/utils/callbacks/peft_ckpt_io.py	150
src/hyperpod_nemo_adapter/utils/callbacks/sharded_ckpt_io.py	67
src/hyperpod_nemo_adapter/utils/config_utils.py	63
src/hyperpod_nemo_adapter/utils/dist_utils.py	23
src/hyperpod_nemo_adapter/utils/dpo_utils.py	220
src/hyperpod_nemo_adapter/utils/exp_manager.py	212
src/hyperpod_nemo_adapter/utils/fsdp_utils.py	127
src/hyperpod_nemo_adapter/utils/general_utils.py	13
src/hyperpod_nemo_adapter/utils/get_rank.py	35
src/hyperpod_nemo_adapter/utils/gpu_utils.py	15
src/hyperpod_nemo_adapter/utils/log_utils.py	23
src/hyperpod_nemo_adapter/utils/sm_utils.py	11
src/hyperpod_nemo_adapter/utils/tracer_utils.py	72
src/hyperpod_nemo_adapter/utils/train_utils.py	134
src/hyperpod_nemo_adapter/version/__version__.py	1