LICENSE
MANIFEST.in
README.rst
VERSION
pyproject.toml
src/sagemaker/__init__.py
src/sagemaker/ai_registry/__init__.py
src/sagemaker/ai_registry/air_constants.py
src/sagemaker/ai_registry/air_hub.py
src/sagemaker/ai_registry/air_hub_entity.py
src/sagemaker/ai_registry/air_utils.py
src/sagemaker/ai_registry/dataset.py
src/sagemaker/ai_registry/dataset_format_detector.py
src/sagemaker/ai_registry/dataset_utils.py
src/sagemaker/ai_registry/dataset_validation.py
src/sagemaker/ai_registry/evaluator.py
src/sagemaker/ai_registry/utils.py
src/sagemaker/ai_registry/schemas/converse.json
src/sagemaker/ai_registry/schemas/dpo.json
src/sagemaker/ai_registry/schemas/genqa.json
src/sagemaker/ai_registry/schemas/hf_preference.json
src/sagemaker/ai_registry/schemas/hf_prompt_completion.json
src/sagemaker/ai_registry/schemas/openai_chat.json
src/sagemaker/ai_registry/schemas/verl.json
src/sagemaker/train/__init__.py
src/sagemaker/train/base_trainer.py
src/sagemaker/train/common.py
src/sagemaker/train/configs.py
src/sagemaker/train/constants.py
src/sagemaker/train/defaults.py
src/sagemaker/train/distributed.py
src/sagemaker/train/dpo_trainer.py
src/sagemaker/train/model_trainer.py
src/sagemaker/train/rlaif_trainer.py
src/sagemaker/train/rlvr_trainer.py
src/sagemaker/train/sft_trainer.py
src/sagemaker/train/templates.py
src/sagemaker/train/tuner.py
src/sagemaker/train/types.py
src/sagemaker/train/utils.py
src/sagemaker/train/aws_batch/__init__.py
src/sagemaker/train/aws_batch/batch_api_helper.py
src/sagemaker/train/aws_batch/boto_client.py
src/sagemaker/train/aws_batch/constants.py
src/sagemaker/train/aws_batch/exception.py
src/sagemaker/train/aws_batch/training_queue.py
src/sagemaker/train/aws_batch/training_queued_job.py
src/sagemaker/train/common_utils/constants.py
src/sagemaker/train/common_utils/finetune_utils.py
src/sagemaker/train/common_utils/get_mlflow_endpoint.py
src/sagemaker/train/common_utils/mlflow_metrics_util.py
src/sagemaker/train/common_utils/model_resolution.py
src/sagemaker/train/common_utils/recipe_utils.py
src/sagemaker/train/common_utils/show_results_utils.py
src/sagemaker/train/common_utils/trainer_wait.py
src/sagemaker/train/container_drivers/__init__.py
src/sagemaker/train/container_drivers/common/__init__.py
src/sagemaker/train/container_drivers/common/utils.py
src/sagemaker/train/container_drivers/distributed_drivers/__init__.py
src/sagemaker/train/container_drivers/distributed_drivers/basic_script_driver.py
src/sagemaker/train/container_drivers/distributed_drivers/mpi_driver.py
src/sagemaker/train/container_drivers/distributed_drivers/mpi_utils.py
src/sagemaker/train/container_drivers/distributed_drivers/torchrun_driver.py
src/sagemaker/train/container_drivers/scripts/__init__.py
src/sagemaker/train/container_drivers/scripts/environment.py
src/sagemaker/train/evaluate/__init__.py
src/sagemaker/train/evaluate/base_evaluator.py
src/sagemaker/train/evaluate/benchmark_evaluator.py
src/sagemaker/train/evaluate/constants.py
src/sagemaker/train/evaluate/custom_scorer_evaluator.py
src/sagemaker/train/evaluate/execution.py
src/sagemaker/train/evaluate/llm_as_judge_evaluator.py
src/sagemaker/train/evaluate/pipeline_templates.py
src/sagemaker/train/local/__init__.py
src/sagemaker/train/local/data.py
src/sagemaker/train/local/entities.py
src/sagemaker/train/local/local_container.py
src/sagemaker/train/remote_function/__init__.py
src/sagemaker/train/remote_function/checkpoint_location.py
src/sagemaker/train/remote_function/client.py
src/sagemaker/train/remote_function/custom_file_filter.py
src/sagemaker/train/remote_function/errors.py
src/sagemaker/train/remote_function/invoke_function.py
src/sagemaker/train/remote_function/job.py
src/sagemaker/train/remote_function/logging_config.py
src/sagemaker/train/remote_function/spark_config.py
src/sagemaker/train/remote_function/core/__init__.py
src/sagemaker/train/remote_function/core/_custom_dispatch_table.py
src/sagemaker/train/remote_function/core/pipeline_variables.py
src/sagemaker/train/remote_function/core/serialization.py
src/sagemaker/train/remote_function/core/stored_function.py
src/sagemaker/train/remote_function/runtime_environment/__init__.py
src/sagemaker/train/remote_function/runtime_environment/bootstrap_runtime_environment.py
src/sagemaker/train/remote_function/runtime_environment/mpi_utils_remote.py
src/sagemaker/train/remote_function/runtime_environment/runtime_environment_manager.py
src/sagemaker/train/remote_function/runtime_environment/spark_app.py
src/sagemaker/train/sm_recipes/__init__.py
src/sagemaker/train/sm_recipes/training_recipes.json
src/sagemaker/train/sm_recipes/utils.py
src/sagemaker_train.egg-info/PKG-INFO
src/sagemaker_train.egg-info/SOURCES.txt
src/sagemaker_train.egg-info/dependency_links.txt
src/sagemaker_train.egg-info/requires.txt
src/sagemaker_train.egg-info/top_level.txt