Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | Z _ __call__() (oumi.core.collators.text_collator_with_padding.TextCollatorWithPadding method) (oumi.core.collators.text_completions_collator_with_padding.TextCompletionsCollatorWithPadding method) (oumi.core.collators.vision_language_collator_with_padding.VisionLanguageCollatorWithPadding method) (oumi.core.collators.vision_language_sft_collator.VisionLanguageSftCollator method) (oumi.core.processors.base_image_processor.BaseImageProcessor method) (oumi.core.processors.base_processor.BaseProcessor method) (oumi.core.processors.default_image_processor.DefaultImageProcessor method) (oumi.core.processors.default_processor.DefaultProcessor method) __enter__() (oumi.launcher.clients.slurm_client.SlurmLogStream method) (oumi.performance.telemetry.CudaTimerContext method) (oumi.performance.telemetry.TimerContext method) __eq__() (oumi.core.configs.BaseConfig method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) __exit__() (oumi.launcher.clients.slurm_client.SlurmLogStream method) (oumi.performance.telemetry.CudaTimerContext method) (oumi.performance.telemetry.TimerContext method) __finalize_and_validate__() (oumi.core.configs.BaseConfig method) (oumi.core.configs.DataParams method) (oumi.core.configs.JobConfig method) (oumi.core.configs.ModelParams method) (oumi.core.configs.params.base_params.BaseParams method) (oumi.core.configs.params.data_params.DataParams method) (oumi.core.configs.params.model_params.ModelParams method) (oumi.launcher.JobConfig method) __getitem__() (oumi.core.datasets.BaseMapDataset method) (oumi.core.datasets.PackedSftDataset method) (oumi.core.registry.Registry method) (oumi.core.types.Conversation method) (oumi.datasets.DebugClassificationDataset method) __iter__() (oumi.core.configs.BaseConfig method) (oumi.core.configs.params.base_params.BaseParams method) (oumi.core.datasets.BaseIterableDataset method) (oumi.core.datasets.BasePretrainingDataset method) (oumi.core.datasets.PretrainingAsyncTextDataset method) __len__() (oumi.core.datasets.BaseMapDataset method) (oumi.datasets.DebugClassificationDataset method) __missing__() (oumi.utils.placeholders.SafeDict method) __post_init__() (oumi.core.configs.AlpacaEvalTaskParams method) (oumi.core.configs.AnalyzeConfig method) (oumi.core.configs.AsyncEvaluationConfig method) (oumi.core.configs.AttributeCombination method) (oumi.core.configs.DatasetParams method) (oumi.core.configs.DatasetSplitParams method) (oumi.core.configs.DocumentSegmentationParams method) (oumi.core.configs.DocumentSource method) (oumi.core.configs.EvaluationConfig method) (oumi.core.configs.EvaluationTaskParams method) (oumi.core.configs.ExampleSource method) (oumi.core.configs.GeneralSynthesisParams method) (oumi.core.configs.GeneratedAttribute method) (oumi.core.configs.GeneratedAttributePostprocessingParams method) (oumi.core.configs.GenerationParams method) (oumi.core.configs.GrpoParams method) (oumi.core.configs.GuidedDecodingParams method) (oumi.core.configs.LMHarnessTaskParams method) (oumi.core.configs.ModelParams method) (oumi.core.configs.params.data_params.DatasetParams method) (oumi.core.configs.params.data_params.DatasetSplitParams method) (oumi.core.configs.params.deepspeed_params.DeepSpeedParams method) (oumi.core.configs.params.evaluation_params.AlpacaEvalTaskParams method) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams method) (oumi.core.configs.params.evaluation_params.LMHarnessTaskParams method) (oumi.core.configs.params.generation_params.GenerationParams method) (oumi.core.configs.params.gkd_params.GkdParams method) (oumi.core.configs.params.grpo_params.GrpoParams method) (oumi.core.configs.params.guided_decoding_params.GuidedDecodingParams method) (oumi.core.configs.params.judge_params.JudgeParams method) (oumi.core.configs.params.model_params.ModelParams method) (oumi.core.configs.params.profiler_params.ProfilerScheduleParams method) (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams method) (oumi.core.configs.params.remote_params.RemoteParams method) (oumi.core.configs.params.synthesis_params.AttributeCombination method) (oumi.core.configs.params.synthesis_params.DatasetSource method) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams method) (oumi.core.configs.params.synthesis_params.DocumentSource method) (oumi.core.configs.params.synthesis_params.ExampleSource method) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams method) (oumi.core.configs.params.synthesis_params.GeneratedAttribute method) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams method) (oumi.core.configs.params.synthesis_params.SampledAttribute method) (oumi.core.configs.params.synthesis_params.SampledAttributeValue method) (oumi.core.configs.params.synthesis_params.TransformationStrategy method) (oumi.core.configs.params.synthesis_params.TransformedAttribute method) (oumi.core.configs.params.training_params.TrainingParams method) (oumi.core.configs.params.tuning_params.TuningParams method) (oumi.core.configs.QuantizationConfig method) (oumi.core.configs.RemoteParams method) (oumi.core.configs.SampledAttribute method) (oumi.core.configs.SampledAttributeValue method) (oumi.core.configs.SynthesisConfig method) (oumi.core.configs.TrainingConfig method) (oumi.core.configs.TrainingParams method) (oumi.core.configs.TransformationStrategy method) (oumi.core.configs.TransformedAttribute method) (oumi.core.configs.TuningParams method) (oumi.utils.torch_utils.ModelParameterCount method) __repr__() (oumi.core.registry.Registry method) (oumi.core.types.ContentItem method) (oumi.core.types.Conversation method) (oumi.core.types.Message method) __str__() (oumi.core.types.Role method) (oumi.core.types.Type method) A accelerators (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) activation_checkpointing (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) active (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) adam_beta1 (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) adam_beta2 (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) adam_epsilon (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) ADAPTER_AND_BASE_MODEL (oumi.core.configs.params.peft_params.PeftSaveMode attribute) (oumi.core.configs.PeftSaveMode attribute) adapter_model (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) ADAPTER_ONLY (oumi.core.configs.params.peft_params.PeftSaveMode attribute) (oumi.core.configs.PeftSaveMode attribute) AdaptiveConcurrencyParams (class in oumi.core.configs.params.remote_params) added_prefix (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) added_suffix (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) additional_info (oumi.quantize.QuantizationResult attribute) ALL_EXHAUSTED (oumi.core.configs.MixtureStrategy attribute) (oumi.core.configs.params.data_params.MixtureStrategy attribute) all_params (oumi.utils.torch_utils.ModelParameterCount attribute) allgather_bucket_size (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) allgather_partitions (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) allow_feature_reshape (oumi.core.feature_generators.FeatureGeneratorOptions attribute) ALPACA_EVAL (oumi.core.configs.EvaluationBackend attribute) (oumi.core.configs.params.evaluation_params.EvaluationBackend attribute) AlpacaDataset (class in oumi.datasets) (class in oumi.datasets.sft) AlpacaEvalDataset (class in oumi.datasets) (class in oumi.datasets.evaluation) AlpacaEvalTaskParams (class in oumi.core.configs) (class in oumi.core.configs.params.evaluation_params) analysis_df (oumi.core.analyze.DatasetAnalyzer property) analysis_results (oumi.core.analyze.DatasetAnalyzer property) analysis_summary (oumi.core.analyze.DatasetAnalyzer property) analyze_dataset() (oumi.core.analyze.DatasetAnalyzer method) analyze_sample() (oumi.core.analyze.LengthAnalyzer method) (oumi.core.analyze.SampleAnalyzer method) AnalyzeConfig (class in oumi.core.configs) analyzers (oumi.core.configs.AnalyzeConfig attribute) ANTHROPIC (oumi.core.configs.InferenceEngineType attribute) anthropic_version (oumi.inference.AnthropicInferenceEngine attribute) AnthropicInferenceEngine (class in oumi.inference) api_key (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) api_key_env_varname (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) (oumi.inference.AnthropicInferenceEngine property) (oumi.inference.BedrockInferenceEngine property) (oumi.inference.DeepSeekInferenceEngine attribute) (oumi.inference.GoogleGeminiInferenceEngine attribute) (oumi.inference.LambdaInferenceEngine property) (oumi.inference.OpenAIInferenceEngine property) (oumi.inference.ParasailInferenceEngine property) (oumi.inference.RemoteInferenceEngine attribute) (oumi.inference.RemoteVLLMInferenceEngine property) (oumi.inference.SambanovaInferenceEngine property) (oumi.inference.TogetherInferenceEngine property) api_url (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) append_id_to_string() (oumi.core.types.Conversation method) apply_chat_template() (oumi.core.inference.BaseInferenceEngine method) (oumi.core.processors.base_processor.BaseProcessor method) (oumi.core.processors.default_processor.DefaultProcessor method) ArgillaDollyDataset (class in oumi.datasets) (class in oumi.datasets.sft) ArgillaMagpieUltraDataset (class in oumi.datasets) (class in oumi.datasets.sft) as_generator() (oumi.core.datasets.BaseMapDataset method) ASSISTANT (oumi.core.types.Role attribute) assistant_only (oumi.core.datasets.BaseSftDataset property) AsyncEvaluationConfig (class in oumi.core.configs) attn_implementation (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) attribute_map (oumi.core.configs.params.synthesis_params.DatasetSource attribute) AttributeCombination (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) AttributeFormatter (class in oumi.core.synthesis.attribute_formatter) AttributeSynthesizer (class in oumi.core.synthesis.attribute_synthesizer) AttributeTransformer (class in oumi.core.synthesis.attribute_transformation) auto_wrap_policy (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) AutoWrapPolicy (class in oumi.core.configs) (class in oumi.core.configs.params.fsdp_params) awq_group_size (oumi.core.configs.QuantizationConfig attribute) awq_version (oumi.core.configs.QuantizationConfig attribute) awq_zero_point (oumi.core.configs.QuantizationConfig attribute) AwqQuantization (class in oumi.quantize) AWS (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) AyaDataset (class in oumi.datasets) (class in oumi.datasets.sft) AZURE (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) B backend (oumi.utils.verl_model_merger.ModelMergerConfig attribute) backoff_factor (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) BACKWARD_POST (oumi.core.configs.BackwardPrefetch attribute) (oumi.core.configs.params.fsdp_params.BackwardPrefetch attribute) BACKWARD_PRE (oumi.core.configs.BackwardPrefetch attribute) (oumi.core.configs.params.fsdp_params.BackwardPrefetch attribute) backward_prefetch (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) BackwardPrefetch (class in oumi.core.configs) (class in oumi.core.configs.params.fsdp_params) base64encode_content_item_image_bytes() (in module oumi.utils.conversation_utils) base_url (oumi.inference.AnthropicInferenceEngine property) (oumi.inference.BedrockInferenceEngine property) (oumi.inference.DeepSeekInferenceEngine attribute) (oumi.inference.GoogleGeminiInferenceEngine attribute) (oumi.inference.LambdaInferenceEngine property) (oumi.inference.OpenAIInferenceEngine property) (oumi.inference.ParasailInferenceEngine property) (oumi.inference.RemoteInferenceEngine attribute) (oumi.inference.RemoteVLLMInferenceEngine property) (oumi.inference.SambanovaInferenceEngine property) (oumi.inference.TogetherInferenceEngine property) BaseCloud (class in oumi.core.launcher) BaseCluster (class in oumi.core.launcher) BaseConfig (class in oumi.core.configs) BaseConversationFeatureGenerator (class in oumi.core.feature_generators) BaseDpoDataset (class in oumi.core.datasets) BaseExperimentalDpoDataset (class in oumi.core.datasets) BaseExperimentalGrpoDataset (class in oumi.core.datasets) BaseExperimentalKtoDataset (class in oumi.core.datasets) BaseImageProcessor (class in oumi.core.processors.base_image_processor) BaseInferenceEngine (class in oumi.core.inference) BaseIterableDataset (class in oumi.core.datasets) BaseJudge (class in oumi.judges) BaseMapDataset (class in oumi.core.datasets) BaseModel (class in oumi.core.models) BaseModelMerger (class in oumi.utils.verl_model_merger) BaseParams (class in oumi.core.configs.params.base_params) BasePretrainingDataset (class in oumi.core.datasets) BaseProcessor (class in oumi.core.processors.base_processor) BaseQuantization (class in oumi.quantize) BaseSftDataset (class in oumi.core.datasets) BaseTokenizer (in module oumi.core.tokenizers) BaseTrainer (class in oumi.core.trainers) BaseTrainerCallback (in module oumi.core.callbacks) BaseTuner (class in oumi.core.tuners) BATCH (oumi.launcher.clusters.frontier_cluster.FrontierCluster.SupportedQueues attribute) batch() (in module oumi.utils.batching) batch_completion_window (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) batch_size (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) (oumi.core.configs.QuantizationConfig attribute) BEDROCK (oumi.core.configs.InferenceEngineType attribute) BedrockInferenceEngine (class in oumi.inference) berry_bench() (in module oumi.evaluation.registry) BerryBenchGrpoDataset (class in oumi.datasets.grpo) beta (oumi.core.configs.params.gkd_params.GkdParams attribute) BF16 (oumi.core.configs.MixedPrecisionDtype attribute) (oumi.core.configs.params.deepspeed_params.DeepSpeedPrecision attribute) (oumi.core.configs.params.training_params.MixedPrecisionDtype attribute) binary (oumi.core.types.ContentItem attribute) BitNetCallback (class in oumi.core.callbacks) BitsAndBytesQuantization (class in oumi.quantize) block_size (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) bnb_4bit_compute_dtype (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) bnb_4bit_quant_storage (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) bnb_4bit_quant_type (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) BOOL (oumi.core.configs.JudgeOutputType attribute) (oumi.core.configs.params.judge_params.JudgeOutputType attribute) buffer_count (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) buffer_count_nvme (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) build_chat_template() (in module oumi.builders) build_collator_from_config() (in module oumi.builders) build_data_collator() (in module oumi.builders) build_dataset() (in module oumi.builders) build_dataset_mixture() (in module oumi.builders) build_metrics_function() (in module oumi.builders) build_model() (in module oumi.builders) build_optimizer() (in module oumi.builders) build_peft_model() (in module oumi.builders) build_processor() (in module oumi.builders) build_quantizer() (in module oumi.builders) build_reward_functions() (in module oumi.builders) build_rollout_function() (in module oumi.builders) build_tokenizer() (in module oumi.builders) build_trainer() (in module oumi.builders) build_training_callbacks() (in module oumi.builders) build_tuner() (in module oumi.builders) C C4Dataset (class in oumi.datasets) (class in oumi.datasets.pretraining) CachedItem (class in oumi.utils.hf_cache_utils) calculate_mfu() (in module oumi.performance.mfu) calculate_mfu_from_model_flops_per_second() (in module oumi.performance.mfu) calibration_samples (oumi.core.configs.QuantizationConfig attribute) cancel() (in module oumi.launcher) (oumi.launcher.clients.local_client.LocalClient method) (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.sky_client.SkyClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) (oumi.launcher.Launcher method) cancel_job() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) CANCELLED (oumi.core.launcher.JobState attribute) CATEGORICAL (oumi.core.configs.params.tuning_params.ParamType attribute) CHAT (oumi.core.configs.params.synthesis_params.TransformationType attribute) (oumi.core.configs.TransformationType attribute) chat_template (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) chat_template_kwargs (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) chat_transform (oumi.core.configs.params.synthesis_params.TransformationStrategy attribute) (oumi.core.configs.TransformationStrategy attribute) ChatqaDataset (class in oumi.datasets) (class in oumi.datasets.sft) ChatqaTatqaDataset (class in oumi.datasets) (class in oumi.datasets.sft) ChatRAGBenchDataset (class in oumi.datasets) (class in oumi.datasets.sft) check_package_prerequisites() (in module oumi.utils.packaging) check_prerequisites() (in module oumi.core.evaluation.utils.platform_prerequisites) checkpoints_dir (oumi.core.configs.AsyncEvaluationConfig attribute) choice (oumi.core.configs.GuidedDecodingParams attribute) (oumi.core.configs.params.guided_decoding_params.GuidedDecodingParams attribute) cleanup_temp (oumi.core.configs.QuantizationConfig attribute) clear() (oumi.core.registry.Registry method) clock_speed_graphics (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) clock_speed_memory (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) clock_speed_sm (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) close() (oumi.launcher.clients.slurm_client.SlurmLogStream method) cloud (oumi.core.configs.JobResources attribute) CLOUD (oumi.core.registry.RegistryType attribute) cloud (oumi.launcher.JobResources attribute) cluster (oumi.core.launcher.JobStatus attribute) CNNClassifier (class in oumi.models) CoALMDataset (class in oumi.datasets) (class in oumi.datasets.sft) COCOCaptionsDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) coerce_model_to_dtype() (in module oumi.utils.torch_utils) collate_images() (oumi.core.collators.vision_language_collator_with_padding.VisionLanguageCollatorWithPadding method) collator_kwargs (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) collator_name (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) collect_telemetry_for_all_ranks (oumi.core.configs.params.telemetry_params.TelemetryParams attribute) (oumi.core.configs.TelemetryParams attribute) column_names (oumi.core.datasets.PretrainingAsyncTextDataset property) combination (oumi.core.configs.AttributeCombination attribute) (oumi.core.configs.params.synthesis_params.AttributeCombination attribute) combination_sampling (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) compile (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) compute_cross_rank_summaries() (oumi.performance.telemetry.TelemetryTracker method) compute_flattened_text_content() (oumi.core.types.Message method) compute_letter_count_reward() (in module oumi.datasets.grpo.rewards) compute_sharp_target_token_length_reward() (in module oumi.datasets.grpo.rewards) compute_soft_target_token_length_reward() (in module oumi.datasets.grpo.rewards) compute_statistics() (in module oumi.utils.analysis_utils) compute_utf8_len() (in module oumi.utils.str_utils) concurrency_step (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) CONFIG (oumi.core.configs.DatasetSource attribute) configure_dependency_warnings() (in module oumi.utils.logging) configure_logger() (in module oumi.utils.logging) connection_timeout (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) CONSTANT (oumi.core.configs.params.training_params.SchedulerType attribute) (oumi.core.configs.SchedulerType attribute) contains() (oumi.core.registry.Registry method) contains_image_content_items_only() (oumi.core.types.Message method) contains_images() (oumi.core.types.Message method) contains_single_image_content_item_only() (oumi.core.types.Message method) contains_single_text_content_item_only() (oumi.core.types.Message method) contains_text() (oumi.core.types.Message method) contains_text_content_items_only() (oumi.core.types.Message method) content (oumi.core.configs.params.synthesis_params.TextMessage attribute) (oumi.core.configs.TextMessage attribute) (oumi.core.types.ContentItem attribute) (oumi.core.types.Message attribute) (oumi.core.types.TemplatedMessage property) content_items (oumi.core.types.Message property) ContentItem (class in oumi.core.types) ContentItemCounts (class in oumi.core.types) contiguous_gradients (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) Conversation (class in oumi.core.types) conversation() (oumi.core.datasets.BaseExperimentalGrpoDataset method) (oumi.core.datasets.BaseSftDataset method) conversation_df (oumi.core.analyze.DatasetAnalyzer property) conversation_id (oumi.core.configs.params.synthesis_params.TextConversation attribute) (oumi.core.configs.TextConversation attribute) (oumi.core.types.Conversation attribute) conversation_to_dataframes() (in module oumi.utils.analysis_utils) conversations() (oumi.core.datasets.BaseExperimentalGrpoDataset method) (oumi.core.datasets.BaseSftDataset method) convert_all_keys_to_serializable_types() (in module oumi.utils.serialization_utils) convert_content_items_to_json_list() (in module oumi.utils.conversation_utils) convert_dataset_to_dataframes() (in module oumi.utils.analysis_utils) convert_message_content_item_to_json_dict() (in module oumi.utils.conversation_utils) convert_message_to_json_content() (in module oumi.utils.conversation_utils) convert_message_to_json_content_list() (in module oumi.utils.conversation_utils) convert_pil_image_mode() (in module oumi.utils.image_utils) convert_to_list_of_tensors() (in module oumi.utils.torch_utils) COSINE (oumi.core.configs.params.training_params.SchedulerType attribute) (oumi.core.configs.SchedulerType attribute) COSINE_WITH_MIN_LR (oumi.core.configs.params.training_params.SchedulerType attribute) (oumi.core.configs.SchedulerType attribute) COSINE_WITH_RESTARTS (oumi.core.configs.params.training_params.SchedulerType attribute) (oumi.core.configs.SchedulerType attribute) count_content_items() (oumi.core.types.Message method) count_letters() (in module oumi.evaluation.registry) count_model_parameters() (in module oumi.utils.torch_utils) countdown_reward() (in module oumi.datasets.grpo.rewards) CountdownGrpoDataset (class in oumi.datasets.grpo) CPU (oumi.core.configs.params.deepspeed_params.DeepSpeedOffloadDevice attribute) cpu_offload (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) cpus (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) create_list_of_message_json_dicts() (in module oumi.utils.conversation_utils) create_model_summary() (in module oumi.utils.torch_utils) create_ones_like() (in module oumi.utils.torch_utils) create_png_bytes_from_image() (in module oumi.utils.image_utils) create_png_bytes_from_image_bytes() (in module oumi.utils.image_utils) create_png_bytes_from_image_list() (in module oumi.utils.image_utils) create_study() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) criterion (oumi.core.models.BaseModel property) (oumi.models.CNNClassifier property) (oumi.models.MLPEncoder property) cuda_measurements (oumi.performance.telemetry.TelemetryState attribute) cuda_timer() (oumi.performance.telemetry.TelemetryTracker method) CudaTimerContext (class in oumi.performance.telemetry) CUSTOM (oumi.core.configs.EvaluationBackend attribute) (oumi.core.configs.params.evaluation_params.EvaluationBackend attribute) custom_eval_metrics (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) cut_prefix (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) cut_suffix (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) D data (oumi.core.configs.TrainingConfig attribute) (oumi.core.configs.TuningConfig attribute) (oumi.core.datasets.BaseIterableDataset property) (oumi.core.datasets.BaseMapDataset property) data_seed (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) DataCollatorForCompletionOnlyLM (class in oumi.core.collators.trl_data_collator_for_completion_only_lm) dataloader_main_process_only (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) dataloader_num_workers (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) dataloader_persistent_workers (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) dataloader_prefetch_factor (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) DataParams (class in oumi.core.configs) (class in oumi.core.configs.params.data_params) DATASET (oumi.core.registry.RegistryType attribute) dataset_format (oumi.core.configs.AnalyzeConfig attribute) dataset_kwargs (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) dataset_name (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) (oumi.core.datasets.BaseDpoDataset attribute) (oumi.core.datasets.BaseExperimentalDpoDataset attribute) (oumi.core.datasets.BaseExperimentalGrpoDataset attribute) (oumi.core.datasets.BaseExperimentalKtoDataset attribute) (oumi.core.datasets.BaseIterableDataset attribute) (oumi.core.datasets.BaseMapDataset attribute) (oumi.core.datasets.BasePretrainingDataset attribute) (oumi.core.datasets.BaseSftDataset attribute) (oumi.core.datasets.PackedSftDataset attribute) (oumi.core.datasets.VisionLanguageDpoDataset attribute) (oumi.core.datasets.VisionLanguageSftDataset attribute) (oumi.datasets.AlpacaDataset attribute) (oumi.datasets.AlpacaEvalDataset attribute) (oumi.datasets.ArgillaDollyDataset attribute) (oumi.datasets.ArgillaMagpieUltraDataset attribute) (oumi.datasets.AyaDataset attribute) (oumi.datasets.C4Dataset attribute) (oumi.datasets.ChatqaDataset attribute) (oumi.datasets.ChatqaTatqaDataset attribute) (oumi.datasets.CoALMDataset attribute) (oumi.datasets.COCOCaptionsDataset attribute) (oumi.datasets.DebugPretrainingDataset attribute) (oumi.datasets.DebugSftDataset attribute) (oumi.datasets.DocmatixDataset attribute) (oumi.datasets.DolmaDataset attribute) (oumi.datasets.evaluation.AlpacaEvalDataset attribute) (oumi.datasets.FalconRefinedWebDataset attribute) (oumi.datasets.FineWebEduDataset attribute) (oumi.datasets.Flickr30kDataset attribute) (oumi.datasets.Geometry3kDataset attribute) (oumi.datasets.grpo.BerryBenchGrpoDataset attribute) (oumi.datasets.grpo.CountdownGrpoDataset attribute) (oumi.datasets.grpo.Gsm8kGrpoDataset attribute) (oumi.datasets.grpo.LetterCountGrpoDataset attribute) (oumi.datasets.grpo.TldrGrpoDataset attribute) (oumi.datasets.HuggingFaceDataset attribute) (oumi.datasets.HuggingFaceVisionDataset attribute) (oumi.datasets.LetterCountGrpoDataset attribute) (oumi.datasets.LlavaInstructMixVsftDataset attribute) (oumi.datasets.LmmsLabMultimodalOpenR1Dataset attribute) (oumi.datasets.MagpieProDataset attribute) (oumi.datasets.MnistSftDataset attribute) (oumi.datasets.OpenbmbRlaifVDataset attribute) (oumi.datasets.OpenO1SFTDataset attribute) (oumi.datasets.OrpoDpoMix40kDataset attribute) (oumi.datasets.PileV1Dataset attribute) (oumi.datasets.PixmoAskModelAnythingDataset attribute) (oumi.datasets.PixmoCapDataset attribute) (oumi.datasets.PixmoCapQADataset attribute) (oumi.datasets.preference_tuning.KtoMix14kDataset attribute) (oumi.datasets.preference_tuning.OrpoDpoMix40kDataset attribute) (oumi.datasets.pretraining.C4Dataset attribute) (oumi.datasets.pretraining.DolmaDataset attribute) (oumi.datasets.pretraining.FalconRefinedWebDataset attribute) (oumi.datasets.pretraining.FineWebEduDataset attribute) (oumi.datasets.pretraining.PileV1Dataset attribute) (oumi.datasets.pretraining.RedPajamaDataV1Dataset attribute) (oumi.datasets.pretraining.RedPajamaDataV2Dataset attribute) (oumi.datasets.pretraining.SlimPajamaDataset attribute) (oumi.datasets.pretraining.StarCoderDataset attribute) (oumi.datasets.pretraining.TheStackDataset attribute) (oumi.datasets.pretraining.TinyStoriesDataset attribute) (oumi.datasets.pretraining.TinyTextbooksDataset attribute) (oumi.datasets.pretraining.WikipediaDataset attribute) (oumi.datasets.pretraining.WikiTextDataset attribute) (oumi.datasets.pretraining.YouTubeCommonsDataset attribute) (oumi.datasets.PromptResponseDataset attribute) (oumi.datasets.RedPajamaDataV1Dataset attribute) (oumi.datasets.RedPajamaDataV2Dataset attribute) (oumi.datasets.sft.AlpacaDataset attribute) (oumi.datasets.sft.ArgillaDollyDataset attribute) (oumi.datasets.sft.ArgillaMagpieUltraDataset attribute) (oumi.datasets.sft.AyaDataset attribute) (oumi.datasets.sft.ChatqaDataset attribute) (oumi.datasets.sft.ChatqaTatqaDataset attribute) (oumi.datasets.sft.ChatRAGBenchDataset attribute) (oumi.datasets.sft.CoALMDataset attribute) (oumi.datasets.sft.HuggingFaceDataset attribute) (oumi.datasets.sft.MagpieProDataset attribute) (oumi.datasets.sft.OpenO1SFTDataset attribute) (oumi.datasets.sft.PromptResponseDataset attribute) (oumi.datasets.sft.TextSftJsonLinesDataset attribute) (oumi.datasets.sft.Tulu3MixtureDataset attribute) (oumi.datasets.sft.UltrachatH4Dataset attribute) (oumi.datasets.sft.WildChatDataset attribute) (oumi.datasets.SlimPajamaDataset attribute) (oumi.datasets.StarCoderDataset attribute) (oumi.datasets.TextSftJsonLinesDataset attribute) (oumi.datasets.TheCauldronDataset attribute) (oumi.datasets.TheStackDataset attribute) (oumi.datasets.TinyStoriesDataset attribute) (oumi.datasets.TinyTextbooksDataset attribute) (oumi.datasets.TldrGrpoDataset attribute) (oumi.datasets.Tulu3MixtureDataset attribute) (oumi.datasets.UltrachatH4Dataset attribute) (oumi.datasets.vision_language.COCOCaptionsDataset attribute) (oumi.datasets.vision_language.DocmatixDataset attribute) (oumi.datasets.vision_language.Flickr30kDataset attribute) (oumi.datasets.vision_language.Geometry3kDataset attribute) (oumi.datasets.vision_language.HuggingFaceVisionDataset attribute) (oumi.datasets.vision_language.LlavaInstructMixVsftDataset attribute) (oumi.datasets.vision_language.LmmsLabMultimodalOpenR1Dataset attribute) (oumi.datasets.vision_language.MnistSftDataset attribute) (oumi.datasets.vision_language.OpenbmbRlaifVDataset attribute) (oumi.datasets.vision_language.PixmoAskModelAnythingDataset attribute) (oumi.datasets.vision_language.PixmoCapDataset attribute) (oumi.datasets.vision_language.PixmoCapQADataset attribute) (oumi.datasets.vision_language.TheCauldronDataset attribute) (oumi.datasets.vision_language.VisionDpoJsonlinesDataset attribute) (oumi.datasets.vision_language.VLJsonlinesDataset attribute) (oumi.datasets.vision_language.Vqav2SmallDataset attribute) (oumi.datasets.VisionDpoJsonlinesDataset attribute) (oumi.datasets.VLJsonlinesDataset attribute) (oumi.datasets.Vqav2SmallDataset attribute) (oumi.datasets.WikipediaDataset attribute) (oumi.datasets.WikiTextDataset attribute) (oumi.datasets.WildChatDataset attribute) (oumi.datasets.YouTubeCommonsDataset attribute) dataset_path (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) (oumi.core.datasets.BaseIterableDataset attribute) (oumi.core.datasets.BaseMapDataset attribute) dataset_source (oumi.core.configs.AnalyzeConfig attribute) DatasetAnalyzer (class in oumi.core.analyze) DatasetParams (class in oumi.core.configs) (class in oumi.core.configs.params.data_params) DatasetPath (class in oumi.core.synthesis.dataset_ingestion) DatasetPlanner (class in oumi.core.synthesis.dataset_planner) DatasetReader (class in oumi.core.synthesis.dataset_ingestion) datasets (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) DatasetSource (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) DatasetSourceParam (in module oumi.core.configs) DatasetSplit (class in oumi.core.configs) (class in oumi.core.configs.params.data_params) DatasetSplitParams (class in oumi.core.configs) (class in oumi.core.configs.params.data_params) DatasetStorageType (class in oumi.core.synthesis.dataset_ingestion) DataSynthesizer (class in oumi.core.synthesis.data_synthesizer) ddp_find_unused_parameters (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) DEBUG (oumi.launcher.clients.polaris_client.PolarisClient.SupportedQueues attribute) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) DEBUG_PREEMPT (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) DEBUG_SCALING (oumi.launcher.clients.polaris_client.PolarisClient.SupportedQueues attribute) DebugClassificationDataset (class in oumi.datasets) DebugPretrainingDataset (class in oumi.datasets) DebugSftDataset (class in oumi.datasets) DEEPSEEK (oumi.core.configs.InferenceEngineType attribute) DeepSeekInferenceEngine (class in oumi.inference) deepspeed (oumi.core.configs.TrainingConfig attribute) deepspeed_config_path (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) DeepSpeedOffloadDevice (class in oumi.core.configs.params.deepspeed_params) DeepSpeedParams (class in oumi.core.configs.params.deepspeed_params) DeepSpeedPrecision (class in oumi.core.configs.params.deepspeed_params) DEFAULT (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) default() (oumi.utils.serialization_utils.TorchJsonEncoder method) default_dataset (oumi.core.datasets.BaseIterableDataset attribute) (oumi.core.datasets.BaseMapDataset attribute) (oumi.core.datasets.BaseSftDataset attribute) (oumi.datasets.AlpacaDataset attribute) (oumi.datasets.AlpacaEvalDataset attribute) (oumi.datasets.ArgillaDollyDataset attribute) (oumi.datasets.ArgillaMagpieUltraDataset attribute) (oumi.datasets.AyaDataset attribute) (oumi.datasets.C4Dataset attribute) (oumi.datasets.ChatqaDataset attribute) (oumi.datasets.ChatRAGBenchDataset attribute) (oumi.datasets.CoALMDataset attribute) (oumi.datasets.COCOCaptionsDataset attribute) (oumi.datasets.DebugPretrainingDataset attribute) (oumi.datasets.DebugSftDataset attribute) (oumi.datasets.DocmatixDataset attribute) (oumi.datasets.DolmaDataset attribute) (oumi.datasets.evaluation.AlpacaEvalDataset attribute) (oumi.datasets.FalconRefinedWebDataset attribute) (oumi.datasets.FineWebEduDataset attribute) (oumi.datasets.Flickr30kDataset attribute) (oumi.datasets.Geometry3kDataset attribute) (oumi.datasets.grpo.BerryBenchGrpoDataset attribute) (oumi.datasets.grpo.CountdownGrpoDataset attribute) (oumi.datasets.grpo.Gsm8kGrpoDataset attribute) (oumi.datasets.grpo.LetterCountGrpoDataset attribute) (oumi.datasets.grpo.TldrGrpoDataset attribute) (oumi.datasets.LetterCountGrpoDataset attribute) (oumi.datasets.LlavaInstructMixVsftDataset attribute) (oumi.datasets.MagpieProDataset attribute) (oumi.datasets.MnistSftDataset attribute) (oumi.datasets.OpenbmbRlaifVDataset attribute) (oumi.datasets.OpenO1SFTDataset attribute) (oumi.datasets.OrpoDpoMix40kDataset attribute) (oumi.datasets.PileV1Dataset attribute) (oumi.datasets.PixmoAskModelAnythingDataset attribute) (oumi.datasets.PixmoCapDataset attribute) (oumi.datasets.PixmoCapQADataset attribute) (oumi.datasets.preference_tuning.KtoMix14kDataset attribute) (oumi.datasets.preference_tuning.OrpoDpoMix40kDataset attribute) (oumi.datasets.pretraining.C4Dataset attribute) (oumi.datasets.pretraining.DolmaDataset attribute) (oumi.datasets.pretraining.FalconRefinedWebDataset attribute) (oumi.datasets.pretraining.FineWebEduDataset attribute) (oumi.datasets.pretraining.PileV1Dataset attribute) (oumi.datasets.pretraining.RedPajamaDataV1Dataset attribute) (oumi.datasets.pretraining.RedPajamaDataV2Dataset attribute) (oumi.datasets.pretraining.SlimPajamaDataset attribute) (oumi.datasets.pretraining.StarCoderDataset attribute) (oumi.datasets.pretraining.TheStackDataset attribute) (oumi.datasets.pretraining.TinyStoriesDataset attribute) (oumi.datasets.pretraining.TinyTextbooksDataset attribute) (oumi.datasets.pretraining.WikipediaDataset attribute) (oumi.datasets.pretraining.WikiTextDataset attribute) (oumi.datasets.pretraining.YouTubeCommonsDataset attribute) (oumi.datasets.PromptResponseDataset attribute) (oumi.datasets.RedPajamaDataV1Dataset attribute) (oumi.datasets.RedPajamaDataV2Dataset attribute) (oumi.datasets.sft.AlpacaDataset attribute) (oumi.datasets.sft.ArgillaDollyDataset attribute) (oumi.datasets.sft.ArgillaMagpieUltraDataset attribute) (oumi.datasets.sft.AyaDataset attribute) (oumi.datasets.sft.ChatqaDataset attribute) (oumi.datasets.sft.ChatRAGBenchDataset attribute) (oumi.datasets.sft.CoALMDataset attribute) (oumi.datasets.sft.MagpieProDataset attribute) (oumi.datasets.sft.OpenO1SFTDataset attribute) (oumi.datasets.sft.PromptResponseDataset attribute) (oumi.datasets.sft.TextSftJsonLinesDataset attribute) (oumi.datasets.sft.Tulu3MixtureDataset attribute) (oumi.datasets.sft.UltrachatH4Dataset attribute) (oumi.datasets.sft.WildChatDataset attribute) (oumi.datasets.SlimPajamaDataset attribute) (oumi.datasets.StarCoderDataset attribute) (oumi.datasets.TextSftJsonLinesDataset attribute) (oumi.datasets.TheCauldronDataset attribute) (oumi.datasets.TheStackDataset attribute) (oumi.datasets.TinyStoriesDataset attribute) (oumi.datasets.TinyTextbooksDataset attribute) (oumi.datasets.TldrGrpoDataset attribute) (oumi.datasets.Tulu3MixtureDataset attribute) (oumi.datasets.UltrachatH4Dataset attribute) (oumi.datasets.vision_language.COCOCaptionsDataset attribute) (oumi.datasets.vision_language.DocmatixDataset attribute) (oumi.datasets.vision_language.Flickr30kDataset attribute) (oumi.datasets.vision_language.Geometry3kDataset attribute) (oumi.datasets.vision_language.LlavaInstructMixVsftDataset attribute) (oumi.datasets.vision_language.MnistSftDataset attribute) (oumi.datasets.vision_language.OpenbmbRlaifVDataset attribute) (oumi.datasets.vision_language.PixmoAskModelAnythingDataset attribute) (oumi.datasets.vision_language.PixmoCapDataset attribute) (oumi.datasets.vision_language.PixmoCapQADataset attribute) (oumi.datasets.vision_language.TheCauldronDataset attribute) (oumi.datasets.vision_language.VisionDpoJsonlinesDataset attribute) (oumi.datasets.vision_language.VLJsonlinesDataset attribute) (oumi.datasets.vision_language.Vqav2SmallDataset attribute) (oumi.datasets.VisionDpoJsonlinesDataset attribute) (oumi.datasets.VLJsonlinesDataset attribute) (oumi.datasets.Vqav2SmallDataset attribute) (oumi.datasets.WikipediaDataset attribute) (oumi.datasets.WikiTextDataset attribute) (oumi.datasets.WildChatDataset attribute) (oumi.datasets.YouTubeCommonsDataset attribute) default_prompt (oumi.datasets.COCOCaptionsDataset attribute) (oumi.datasets.vision_language.COCOCaptionsDataset attribute) default_subset (oumi.core.datasets.BaseIterableDataset attribute) (oumi.core.datasets.BaseMapDataset attribute) (oumi.datasets.ChatqaDataset attribute) (oumi.datasets.ChatqaTatqaDataset attribute) (oumi.datasets.ChatRAGBenchDataset attribute) (oumi.datasets.sft.ChatqaDataset attribute) (oumi.datasets.sft.ChatqaTatqaDataset attribute) (oumi.datasets.sft.ChatRAGBenchDataset attribute) default_system_message (oumi.datasets.ChatRAGBenchDataset attribute) (oumi.datasets.sft.ChatRAGBenchDataset attribute) DefaultImageProcessor (class in oumi.core.processors.default_image_processor) DefaultProcessor (class in oumi.core.processors.default_processor) delete_file() (oumi.inference.RemoteInferenceEngine method) DEMAND (oumi.launcher.clients.polaris_client.PolarisClient.SupportedQueues attribute) dep_log_level (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) description (oumi.core.configs.params.synthesis_params.SampledAttribute attribute) (oumi.core.configs.params.synthesis_params.SampledAttributeValue attribute) (oumi.core.configs.SampledAttribute attribute) (oumi.core.configs.SampledAttributeValue attribute) device (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) device_cleanup() (in module oumi.utils.torch_utils) device_count (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) device_index (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) device_map (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) DICT (oumi.core.configs.params.synthesis_params.TransformationType attribute) (oumi.core.configs.TransformationType attribute) dict_cache() (in module oumi.utils.cache_utils) dict_lru_cache() (in module oumi.utils.cache_utils) dict_transform (oumi.core.configs.params.synthesis_params.TransformationStrategy attribute) (oumi.core.configs.TransformationStrategy attribute) DIRECT (oumi.core.configs.DatasetSource attribute) disable_dropout (oumi.core.configs.params.gkd_params.GkdParams attribute) disable_dropout() (in module oumi.utils.torch_naming_heuristics) disk_size (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) disk_tier (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) DocmatixDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) DocumentReader (class in oumi.core.synthesis.document_ingestion) DocumentSegmentationParams (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) DocumentSegmenter (class in oumi.core.synthesis.document_ingestion) DocumentSource (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) DolmaDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) done (oumi.core.launcher.JobStatus attribute) down() (in module oumi.launcher) (oumi.core.launcher.BaseCluster method) (oumi.launcher.clients.sky_client.SkyClient method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) (oumi.launcher.Launcher method) DROP_ALWAYS (oumi.core.configs.internal.internal_model_config.InternalFeatureFirstDimAction attribute) DROP_IF_DUMMY (oumi.core.configs.internal.internal_model_config.InternalFeatureFirstDimAction attribute) E embedding_params (oumi.utils.torch_utils.ModelParameterCount attribute) empty_device_cache_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) enable_cpu_profiling (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) enable_cuda_profiling (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) enable_deepspeed (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) enable_fsdp (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) enable_gradient_checkpointing (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) enable_liger_kernel (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) enable_mlflow (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) enable_schedule (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) enable_tensorboard (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) enable_wandb (oumi.core.configs.EvaluationConfig attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) engine (oumi.core.configs.InferenceConfig attribute) ENUM (oumi.core.configs.JudgeOutputType attribute) (oumi.core.configs.params.judge_params.JudgeOutputType attribute) envs (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) epsilon (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) error_threshold (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) estimate_sample_dict_size_in_bytes() (in module oumi.utils.torch_utils) EVA (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) eval_kwargs (oumi.core.configs.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.AlpacaEvalTaskParams attribute) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.LMHarnessTaskParams attribute) eval_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) eval_strategy (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) evaluate() (in module oumi) (in module oumi.core.evaluation.backends.alpaca_eval) (in module oumi.core.evaluation.backends.lm_harness) (oumi.core.evaluation.Evaluator method) (oumi.core.trainers.Trainer method) evaluate_async() (in module oumi) evaluate_task() (oumi.core.evaluation.Evaluator method) evaluation (oumi.core.configs.AsyncEvaluationConfig attribute) evaluation_backend (oumi.core.configs.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams attribute) evaluation_direction (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) EVALUATION_FUNCTION (oumi.core.registry.RegistryType attribute) evaluation_metrics (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) EvaluationBackend (class in oumi.core.configs) (class in oumi.core.configs.params.evaluation_params) EvaluationConfig (class in oumi.core.configs) EvaluationResult (class in oumi.core.evaluation) EvaluationTaskParams (class in oumi.core.configs) (class in oumi.core.configs.params.evaluation_params) Evaluator (class in oumi.core.evaluation) examples (oumi.core.configs.ExampleSource attribute) (oumi.core.configs.params.judge_params.JudgeParams attribute) (oumi.core.configs.params.synthesis_params.ExampleSource attribute) ExampleSource (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) exclude_prompt_from_response (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) exec() (oumi.launcher.clients.sky_client.SkyClient method) exit_code (oumi.launcher.clients.polaris_client.PolarisResponse attribute) (oumi.launcher.clients.slurm_client.SlurmResponse attribute) EXTENDED (oumi.launcher.clusters.frontier_cluster.FrontierCluster.SupportedQueues attribute) extract_prompt_images_completion_from_single_turn_conversation() (in module oumi.utils.grpo_utils) F FAILED (oumi.core.launcher.JobState attribute) FalconRefinedWebDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) fan_speed (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) fan_speeds (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) FeatureGeneratorOptions (class in oumi.core.feature_generators) field_key (oumi.judges.JudgeOutputField attribute) field_scores (oumi.judges.JudgeOutput attribute) (oumi.judges.JudgeOutputField attribute) field_type (oumi.judges.JudgeOutputField attribute) field_values (oumi.judges.JudgeOutput attribute) file_mounts (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) filter() (oumi.core.analyze.DatasetAnalyzer method) filter_messages() (oumi.core.types.Conversation method) finalize_and_validate() (oumi.core.configs.BaseConfig method) (oumi.core.configs.params.base_params.BaseParams method) find_hf_token() (in module oumi.utils.hf_utils) find_internal_model_config() (in module oumi.core.configs.internal.supported_models) find_internal_model_config_using_model_name() (in module oumi.core.configs.internal.supported_models) find_model_hf_config() (in module oumi.core.configs.internal.supported_models) FineWebEduDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) first_dim_action (oumi.core.configs.internal.internal_model_config.InternalFeatureSpec attribute) FIRST_EXHAUSTED (oumi.core.configs.MixtureStrategy attribute) (oumi.core.configs.params.data_params.MixtureStrategy attribute) first_message() (oumi.core.types.Conversation method) fixed_peft_params (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) fixed_training_params (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) flatten_config() (in module oumi.utils.serialization_utils) Flickr30kDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) FLOAT (oumi.core.configs.JudgeOutputType attribute) (oumi.core.configs.params.judge_params.JudgeOutputType attribute) (oumi.core.configs.params.tuning_params.ParamType attribute) format() (oumi.core.synthesis.attribute_formatter.AttributeFormatter method) format_cudnn_version() (in module oumi.utils.torch_utils) format_size() (in module oumi.utils.hf_cache_utils) format_type (oumi.quantize.QuantizationResult attribute) forward() (oumi.core.models.BaseModel method) (oumi.models.CNNClassifier method) (oumi.models.MLPEncoder method) forward_prefetch (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) FP16 (oumi.core.configs.MixedPrecisionDtype attribute) (oumi.core.configs.params.deepspeed_params.DeepSpeedPrecision attribute) (oumi.core.configs.params.training_params.MixedPrecisionDtype attribute) freeze_layers (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) freeze_model_layers() (in module oumi.utils.torch_utils) frequency_penalty (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) from_dict() (oumi.core.types.Conversation class method) from_json() (oumi.core.types.Conversation class method) from_path() (oumi.core.configs.JudgeConfig class method) from_proto() (oumi.core.types.ContentItem static method) (oumi.core.types.Conversation static method) (oumi.core.types.Message static method) from_raw_output() (oumi.judges.JudgeOutput class method) from_str() (oumi.core.configs.BaseConfig class method) from_yaml() (oumi.core.configs.BaseConfig class method) from_yaml_and_arg_list() (oumi.core.configs.BaseConfig class method) FrontierCloud (class in oumi.launcher.clouds) FrontierCluster (class in oumi.launcher.clusters.frontier_cluster) FrontierCluster.SupportedQueues (class in oumi.launcher.clusters.frontier_cluster) frozen_params_percent (oumi.utils.torch_utils.ModelParameterCount property) fsdp (oumi.core.configs.TrainingConfig attribute) (oumi.core.configs.TuningConfig attribute) FSDPModelMerger (class in oumi.utils.verl_model_merger) FSDPParams (class in oumi.core.configs) (class in oumi.core.configs.params.fsdp_params) full_determinism (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) FULL_SHARD (oumi.core.configs.params.fsdp_params.ShardingStrategy attribute) (oumi.core.configs.ShardingStrategy attribute) FULL_STATE_DICT (oumi.core.configs.params.fsdp_params.StateDictType attribute) (oumi.core.configs.StateDictType attribute) G GAUSSIAN (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) GCP (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) GeneralSynthesisParams (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) generate_raw_output() (oumi.judges.JudgeOutput method) generated_attributes (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) GeneratedAttribute (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) GeneratedAttributePostprocessingParams (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) generation (oumi.core.configs.EvaluationConfig attribute) (oumi.core.configs.InferenceConfig attribute) GenerationParams (class in oumi.core.configs) (class in oumi.core.configs.params.generation_params) Geometry3kDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) get() (oumi.core.registry.Registry method) get_active_users() (oumi.launcher.clients.polaris_client.PolarisClient static method) (oumi.launcher.clients.slurm_client.SlurmClient static method) get_all() (oumi.core.registry.Registry method) get_all_models_map() (in module oumi.core.configs.internal.supported_models) get_alpaca_schema() (in module oumi.utils.analysis_utils) get_batch_api_url() (oumi.inference.RemoteInferenceEngine method) get_batch_results() (oumi.inference.BedrockInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) get_batch_status() (oumi.inference.BedrockInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) get_best_trial() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) get_best_trials() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) get_cloud() (in module oumi.launcher) (oumi.launcher.Launcher method) get_cluster() (oumi.core.launcher.BaseCloud method) (oumi.launcher.clouds.FrontierCloud method) (oumi.launcher.clouds.LocalCloud method) (oumi.launcher.clouds.PerlmutterCloud method) (oumi.launcher.clouds.PolarisCloud method) (oumi.launcher.clouds.SkyCloud method) (oumi.launcher.clouds.SlurmCloud method) get_conversation_schema() (in module oumi.utils.analysis_utils) get_dataset() (oumi.core.registry.Registry method) get_default_special_tokens() (in module oumi.core.tokenizers) get_default_vlm_model_config() (in module oumi.core.configs.internal.supported_models) get_device_name() (in module oumi.utils.torch_utils) get_dpo_schema() (in module oumi.utils.analysis_utils) get_dtype_size_in_bytes() (in module oumi.utils.torch_utils) get_editable_install_override_env_var() (in module oumi.utils.str_utils) get_evaluation_backend() (oumi.core.configs.EvaluationTaskParams method) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams method) get_evaluation_function() (oumi.core.registry.Registry method) get_failure_reason_from_response() (in module oumi.utils.http) get_file() (oumi.inference.RemoteInferenceEngine method) get_file_api_url() (oumi.inference.RemoteInferenceEngine method) get_file_content() (oumi.inference.RemoteInferenceEngine method) get_file_extension() (oumi.core.synthesis.dataset_ingestion.DatasetPath method) get_first_dim_len() (in module oumi.utils.torch_utils) get_git_revision_hash() (in module oumi.utils.git_utils) get_git_root_dir() (in module oumi.utils.git_utils) get_git_tag() (in module oumi.utils.git_utils) get_hf_chat_template() (in module oumi.utils.hf_utils) get_job() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clients.local_client.LocalClient method) (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) get_jobs() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) get_judge_config() (oumi.core.registry.Registry method) get_kto_schema() (in module oumi.utils.analysis_utils) get_last_eval_metrics() (oumi.core.trainers.BaseTrainer method) (oumi.core.trainers.HuggingFaceTrainer method) (oumi.core.trainers.Trainer method) (oumi.core.trainers.VerlGrpoTrainer method) get_latest_job() (oumi.launcher.clients.slurm_client.SlurmClient method) get_literal_value() (oumi.core.configs.LoraWeightInitialization method) (oumi.core.configs.MixtureStrategy method) (oumi.core.configs.params.data_params.MixtureStrategy method) (oumi.core.configs.params.peft_params.LoraWeightInitialization method) get_local_filepath_for_gguf() (in module oumi.utils.model_caching) get_logger() (in module oumi.utils.logging) get_logs_stream() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clients.sky_client.SkyClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) get_lora_rank() (in module oumi.utils.peft_utils) get_metrics_function() (oumi.core.registry.Registry method) get_model() (oumi.core.registry.Registry method) get_model_config() (oumi.core.registry.Registry method) get_nvidia_gpu_fan_speeds() (in module oumi.utils.device_utils) get_nvidia_gpu_memory_utilization() (in module oumi.utils.device_utils) get_nvidia_gpu_power_usage() (in module oumi.utils.device_utils) get_nvidia_gpu_runtime_info() (in module oumi.utils.device_utils) get_nvidia_gpu_temperature() (in module oumi.utils.device_utils) get_oumi_root_directory() (in module oumi.utils.io_utils) get_path_str() (oumi.core.synthesis.dataset_ingestion.DatasetPath method) get_placeholders() (in module oumi.utils.placeholders) (oumi.core.configs.params.judge_params.JudgeParams method) get_pretraining_schema() (in module oumi.utils.analysis_utils) get_prompt_response_schema() (in module oumi.utils.analysis_utils) get_python_package_versions() (in module oumi.utils.version_utils) get_results() (oumi.core.evaluation.EvaluationResult method) get_reward_function() (oumi.core.registry.Registry method) get_rollout_function() (oumi.core.registry.Registry method) get_sample_analyzer() (oumi.core.registry.Registry method) get_schema_for_format() (in module oumi.utils.analysis_utils) get_shape_as_list() (in module oumi.utils.torch_utils) get_slurm_connections() (oumi.launcher.clusters.slurm_cluster.SlurmCluster static method) get_split() (oumi.core.configs.DataParams method) (oumi.core.configs.params.data_params.DataParams method) get_state_dicts_from_all_ranks() (oumi.performance.telemetry.TelemetryTracker method) get_storage_type() (oumi.core.synthesis.dataset_ingestion.DatasetPath method) get_strategy() (oumi.core.configs.params.synthesis_params.TransformedAttribute method) (oumi.core.configs.TransformedAttribute method) get_summaries_from_all_ranks() (oumi.performance.telemetry.TelemetryTracker method) get_summary() (oumi.performance.telemetry.TelemetryTracker method) get_supported_formats() (oumi.quantize.BaseQuantization method) get_supported_methods() (oumi.quantize.BaseQuantization method) get_supported_params() (oumi.core.inference.BaseInferenceEngine method) (oumi.inference.AnthropicInferenceEngine method) (oumi.inference.BedrockInferenceEngine method) (oumi.inference.GoogleGeminiInferenceEngine method) (oumi.inference.GoogleVertexInferenceEngine method) (oumi.inference.LlamaCppInferenceEngine method) (oumi.inference.NativeTextInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) (oumi.inference.RemoteVLLMInferenceEngine method) (oumi.inference.SambanovaInferenceEngine method) (oumi.inference.SGLangInferenceEngine method) (oumi.inference.VLLMInferenceEngine method) get_torch_dtype() (in module oumi.utils.torch_utils) get_transformers_auto_model_class() (oumi.utils.verl_model_merger.BaseModelMerger method) get_typed_value() (oumi.judges.JudgeOutputField method) get_value_distribution() (oumi.core.configs.params.synthesis_params.SampledAttribute method) (oumi.core.configs.SampledAttribute method) gkd (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) GkdParams (class in oumi.core.configs.params.gkd_params) GOOGLE_GEMINI (oumi.core.configs.InferenceEngineType attribute) GOOGLE_VERTEX (oumi.core.configs.InferenceEngineType attribute) GoogleGeminiInferenceEngine (class in oumi.inference) GoogleVertexInferenceEngine (class in oumi.inference) gpu_memory (oumi.performance.telemetry.TelemetryState attribute) gpu_memory_logger() (in module oumi.performance.telemetry) gpu_temperature (oumi.performance.telemetry.TelemetryState attribute) gpu_utilization (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) gradient_accumulation_steps (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) gradient_checkpointing_kwargs (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) gradient_clipping (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) group_trainable_params() (in module oumi.utils.torch_naming_heuristics) grpo (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) GrpoParams (class in oumi.core.configs) (class in oumi.core.configs.params.grpo_params) gsm8k_reward() (in module oumi.datasets.grpo.rewards) Gsm8kGrpoDataset (class in oumi.datasets.grpo) guess_transformer_layer_cls() (in module oumi.utils.torch_naming_heuristics) guided_decoding (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) GuidedDecodingParams (class in oumi.core.configs) (class in oumi.core.configs.params.guided_decoding_params) H HardwareException HF (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) (oumi.core.synthesis.dataset_ingestion.DatasetStorageType attribute) hf_model_config_path (oumi.utils.verl_model_merger.ModelMergerConfig attribute) hf_model_path (oumi.utils.verl_model_merger.ModelMergerConfig attribute) hf_revision (oumi.core.configs.params.synthesis_params.DatasetSource attribute) hf_split (oumi.core.configs.params.synthesis_params.DatasetSource attribute) hf_upload (oumi.utils.verl_model_merger.ModelMergerConfig attribute) hf_upload_path (oumi.utils.verl_model_merger.ModelMergerConfig attribute) HfMfuTrainerCallback (class in oumi.core.callbacks) hostname (oumi.launcher.clusters.slurm_cluster.SlurmCluster.ConnectionInfo attribute) (oumi.performance.telemetry.TelemetryState attribute) HuggingFaceDataset (class in oumi.datasets) (class in oumi.datasets.sft) HuggingFaceTrainer (class in oumi.core.trainers) HuggingFaceVisionDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) HYBRID_SHARD (oumi.core.configs.params.fsdp_params.ShardingStrategy attribute) (oumi.core.configs.ShardingStrategy attribute) HYBRID_SHARD_ZERO2 (oumi.core.configs.params.fsdp_params.ShardingStrategy attribute) (oumi.core.configs.ShardingStrategy attribute) I id (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.DocumentSource attribute) (oumi.core.configs.GeneratedAttribute attribute) (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSource attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttribute attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.SampledAttribute attribute) (oumi.core.configs.params.synthesis_params.SampledAttributeValue attribute) (oumi.core.configs.params.synthesis_params.TransformedAttribute attribute) (oumi.core.configs.SampleAnalyzerParams attribute) (oumi.core.configs.SampledAttribute attribute) (oumi.core.configs.SampledAttributeValue attribute) (oumi.core.configs.TransformedAttribute attribute) (oumi.core.launcher.JobStatus attribute) (oumi.core.types.Message attribute) ignore_features (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) IMAGE_BINARY (oumi.core.types.Type attribute) image_content_items (oumi.core.types.Message property) image_dependent (oumi.core.configs.internal.internal_model_config.InternalFeatureSpec attribute) image_id (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) image_id_map (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) image_items (oumi.core.types.ContentItemCounts attribute) IMAGE_PATH (oumi.core.types.Type attribute) image_processor (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) image_token (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) image_token_id (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) IMAGE_URL (oumi.core.types.Type attribute) include_alternative_mfu_metrics (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) include_explanation (oumi.core.configs.params.judge_params.JudgeParams attribute) include_performance_metrics (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) infer() (in module oumi) (oumi.core.inference.BaseInferenceEngine method) infer_batch() (oumi.inference.BedrockInferenceEngine method) (oumi.inference.GoogleGeminiInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) infer_from_file() (oumi.inference.LlamaCppInferenceEngine method) (oumi.inference.NativeTextInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) (oumi.inference.VLLMInferenceEngine method) infer_interactive() (in module oumi) infer_online() (oumi.inference.LlamaCppInferenceEngine method) (oumi.inference.NativeTextInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) (oumi.inference.VLLMInferenceEngine method) inference_config (oumi.core.configs.JudgeConfig attribute) (oumi.core.configs.SynthesisConfig attribute) inference_engine (oumi.core.configs.EvaluationConfig attribute) inference_remote_params (oumi.core.configs.EvaluationConfig attribute) InferenceConfig (class in oumi.core.configs) InferenceEngineType (class in oumi.core.configs) initial_concurrency_factor (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) initialize_clusters() (oumi.launcher.clouds.FrontierCloud method) (oumi.launcher.clouds.PerlmutterCloud method) (oumi.launcher.clouds.PolarisCloud method) (oumi.launcher.clouds.SlurmCloud method) input_data (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) input_documents (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) input_examples (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) input_path (oumi.core.configs.InferenceConfig attribute) instance_type (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) instruction_messages (oumi.core.configs.GeneratedAttribute attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttribute attribute) INT (oumi.core.configs.JudgeOutputType attribute) (oumi.core.configs.params.judge_params.JudgeOutputType attribute) (oumi.core.configs.params.tuning_params.ParamType attribute) INTERACTIVE (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) InternalFeatureFirstDimAction (class in oumi.core.configs.internal.internal_model_config) InternalFeatureSpec (class in oumi.core.configs.internal.internal_model_config) InternalModelConfig (class in oumi.core.configs.internal.internal_model_config) InternalPaddingSide (class in oumi.core.configs.internal.internal_model_config) InternalVisualModelConfig (class in oumi.core.configs.internal.internal_model_config) is_cached_to_disk_hf_dataset() (in module oumi.utils.hf_utils) is_custom_model() (in module oumi.core.configs.internal.supported_models) is_dev_build() (in module oumi.utils.version_utils) is_image() (oumi.core.types.ContentItem method) is_image_text_llm() (in module oumi.builders) is_multimodal (oumi.core.configs.AnalyzeConfig attribute) is_non_retriable_status_code() (in module oumi.utils.http) is_power_of_two() (in module oumi.utils.math_utils) is_text() (oumi.core.types.ContentItem method) is_torchdata_available() (in module oumi.utils.packaging) is_using_accelerate() (in module oumi.utils.distributed_utils) is_using_accelerate_fsdp() (in module oumi.utils.distributed_utils) is_value_model (oumi.utils.verl_model_merger.ModelMergerConfig attribute) is_zero3_enabled() (oumi.core.configs.params.deepspeed_params.DeepSpeedParams method) iter_raw() (oumi.core.datasets.BaseIterableDataset method) J JobConfig (class in oumi.core.configs) (class in oumi.launcher) JobResources (class in oumi.core.configs) (class in oumi.launcher) JobState (class in oumi.core.launcher) JobStatus (class in oumi.core.launcher) json (oumi.core.configs.GuidedDecodingParams attribute) JSON (oumi.core.configs.JudgeResponseFormat attribute) json (oumi.core.configs.params.guided_decoding_params.GuidedDecodingParams attribute) JSON (oumi.core.configs.params.judge_params.JudgeResponseFormat attribute) json_serializer() (in module oumi.utils.serialization_utils) judge() (oumi.judges.BaseJudge method) JUDGE_CONFIG (oumi.core.registry.RegistryType attribute) judge_params (oumi.core.configs.JudgeConfig attribute) JudgeConfig (class in oumi.core.configs) JudgeOutput (class in oumi.judges) JudgeOutputField (class in oumi.judges) JudgeOutputType (class in oumi.core.configs) (class in oumi.core.configs.params.judge_params) JudgeParams (class in oumi.core.configs.params.judge_params) JudgeResponseFormat (class in oumi.core.configs) (class in oumi.core.configs.params.judge_params) judgment_scores (oumi.core.configs.params.judge_params.JudgeParams attribute) judgment_type (oumi.core.configs.params.judge_params.JudgeParams attribute) JUPYTER (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) K K8S (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) KEEP (oumi.core.configs.internal.internal_model_config.InternalFeatureFirstDimAction attribute) keep_original_text (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) keep_original_text_attribute (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) KtoMix14kDataset (class in oumi.datasets.preference_tuning) L label_ignore_index (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) LAMBDA (oumi.core.configs.InferenceEngineType attribute) (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) LambdaInferenceEngine (class in oumi.inference) last_accessed (oumi.utils.hf_cache_utils.CachedItem attribute) last_message() (oumi.core.types.Conversation method) last_modified (oumi.utils.hf_cache_utils.CachedItem attribute) launch() (oumi.launcher.clients.sky_client.SkyClient method) Launcher (class in oumi.launcher) learning_rate (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) LengthAnalyzer (class in oumi.core.analyze) LetterCountGrpoDataset (class in oumi.datasets) (class in oumi.datasets.grpo) limit_per_process_memory() (in module oumi.utils.torch_utils) LINEAR (oumi.core.configs.params.training_params.SchedulerType attribute) (oumi.core.configs.SchedulerType attribute) LIST (oumi.core.configs.params.synthesis_params.TransformationType attribute) (oumi.core.configs.TransformationType attribute) list_batches() (oumi.inference.BedrockInferenceEngine method) (oumi.inference.RemoteInferenceEngine method) list_clusters() (oumi.core.launcher.BaseCloud method) (oumi.launcher.clouds.FrontierCloud method) (oumi.launcher.clouds.LocalCloud method) (oumi.launcher.clouds.PerlmutterCloud method) (oumi.launcher.clouds.PolarisCloud method) (oumi.launcher.clouds.SkyCloud method) (oumi.launcher.clouds.SlurmCloud method) list_evaluation_backends() (oumi.core.configs.EvaluationTaskParams static method) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams static method) list_files() (oumi.inference.RemoteInferenceEngine method) list_hf_cache() (in module oumi.utils.hf_cache_utils) list_jobs() (oumi.launcher.clients.local_client.LocalClient method) (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) list_transform (oumi.core.configs.params.synthesis_params.TransformationStrategy attribute) (oumi.core.configs.TransformationStrategy attribute) LLAMACPP (oumi.core.configs.InferenceEngineType attribute) LlamaCppInferenceEngine (class in oumi.inference) LlavaInstructMixVsftDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) llm_int8_skip_modules (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) LM_HARNESS (oumi.core.configs.EvaluationBackend attribute) (oumi.core.configs.params.evaluation_params.EvaluationBackend attribute) lmbda (oumi.core.configs.params.gkd_params.GkdParams attribute) LMHarnessTaskParams (class in oumi.core.configs) (class in oumi.core.configs.params.evaluation_params) LmmsLabMultimodalOpenR1Dataset (class in oumi.datasets) (class in oumi.datasets.vision_language) load_dataset_from_config() (in module oumi.utils.analysis_utils) load_file() (in module oumi.utils.io_utils) load_if_exists (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) load_image_bytes_to_content_item() (in module oumi.utils.conversation_utils) load_image_png_bytes_from_path() (in module oumi.utils.image_utils) load_image_png_bytes_from_url() (in module oumi.utils.image_utils) load_infer_prob() (in module oumi.utils.saver) load_infer_prob_csv() (in module oumi.utils.saver) load_json() (in module oumi.utils.io_utils) load_jsonlines() (in module oumi.utils.io_utils) load_pdf_pages_from_bytes() (in module oumi.utils.image_utils) load_pdf_pages_from_path() (in module oumi.utils.image_utils) load_pdf_pages_from_url() (in module oumi.utils.image_utils) load_pil_image_from_bytes() (in module oumi.utils.image_utils) load_pil_image_from_content_item() (in module oumi.utils.conversation_utils) load_pil_image_from_path() (in module oumi.utils.image_utils) load_pil_image_from_url() (in module oumi.utils.image_utils) load_pretrained_weights (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) load_state_dict() (oumi.performance.telemetry.TelemetryTracker method) LOCAL (oumi.core.synthesis.dataset_ingestion.DatasetStorageType attribute) local_dir (oumi.utils.verl_model_merger.ModelMergerConfig attribute) LOCAL_STATE_DICT (oumi.core.configs.params.fsdp_params.StateDictType attribute) (oumi.core.configs.StateDictType attribute) LocalClient (class in oumi.launcher.clients.local_client) LocalCloud (class in oumi.launcher.clouds) LocalCluster (class in oumi.launcher.clusters.local_cluster) LOFTQ (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) log() (oumi.core.trainers.Trainer method) log_completions (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) log_devices_info() (in module oumi.utils.torch_utils) log_example_for_debugging() (in module oumi.utils.debug_utils) log_examples (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) log_gpu_memory() (oumi.performance.telemetry.TelemetryTracker method) log_level (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) log_metrics() (oumi.core.trainers.Trainer method) log_model_summary (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) log_model_summary() (in module oumi.utils.torch_utils) log_number_of_model_parameters() (in module oumi.utils.torch_utils) log_nvidia_gpu_fan_speeds() (in module oumi.utils.device_utils) log_nvidia_gpu_memory_utilization() (in module oumi.utils.device_utils) log_nvidia_gpu_power_usage() (in module oumi.utils.device_utils) log_nvidia_gpu_runtime_info() (in module oumi.utils.device_utils) log_nvidia_gpu_temperature() (in module oumi.utils.device_utils) log_peak_gpu_memory() (in module oumi.utils.torch_utils) log_samples (oumi.core.configs.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams attribute) log_versioning_info() (in module oumi.utils.torch_utils) logging_dir (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) logging_first_step (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) logging_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) logging_strategy (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) logit_bias (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) LOGUNIFORM (oumi.core.configs.params.tuning_params.ParamType attribute) lora_alpha (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_bias (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_dropout (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_init_weights (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_modules_to_save (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_r (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_target_modules (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_target_parameters (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) lora_task_type (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) LoraWeightInitialization (class in oumi.core.configs) (class in oumi.core.configs.params.peft_params) lr_scheduler_kwargs (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) lr_scheduler_type (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) M MagpieProDataset (class in oumi.datasets) (class in oumi.datasets.sft) main() (in module oumi.utils.verl_model_merger) main_image_feature (oumi.core.configs.internal.internal_model_config.InternalVisualModelConfig attribute) make_hashable() (in module oumi.utils.cache_utils) max_completion_length (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) max_concurrency (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) max_grad_norm (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) max_new_tokens (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) (oumi.core.configs.params.gkd_params.GkdParams attribute) max_package_version (oumi.utils.packaging.PackagePrerequisites attribute) max_prompt_length (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) max_retries (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) max_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) measurements (oumi.performance.telemetry.TelemetryState attribute) MegatronModelMerger (class in oumi.utils.verl_model_merger) memory (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) memory_efficient_linear (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) memory_utilization (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) merge_and_save() (oumi.utils.verl_model_merger.BaseModelMerger method) (oumi.utils.verl_model_merger.FSDPModelMerger method) (oumi.utils.verl_model_merger.MegatronModelMerger method) MERGED (oumi.core.configs.params.peft_params.PeftSaveMode attribute) (oumi.core.configs.PeftSaveMode attribute) Message (class in oumi.core.types) message (oumi.core.types.TemplatedMessage property) message_df (oumi.core.analyze.DatasetAnalyzer property) messages (oumi.core.configs.params.synthesis_params.TextConversation attribute) (oumi.core.configs.TextConversation attribute) (oumi.core.types.Conversation attribute) metadata (oumi.core.configs.params.synthesis_params.TextConversation attribute) (oumi.core.configs.TextConversation attribute) (oumi.core.launcher.JobStatus attribute) (oumi.core.types.Conversation attribute) method (oumi.core.configs.QuantizationConfig attribute) metrics_function (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) METRICS_FUNCTION (oumi.core.registry.RegistryType attribute) MfuTrainerCallback (class in oumi.core.callbacks) min_concurrency (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) min_num_params (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) min_p (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) min_package_version (oumi.utils.packaging.PackagePrerequisites attribute) min_update_time (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) min_window_size (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) mixed_precision (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) mixed_precision_dtype (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) MixedPrecisionDtype (class in oumi.core.configs) (class in oumi.core.configs.params.training_params) mixture_proportion (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) mixture_strategy (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) MixtureStrategy (class in oumi.core.configs) (class in oumi.core.configs.params.data_params) MLPEncoder (class in oumi.models) MnistSftDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) model (oumi.core.configs.EvaluationConfig attribute) (oumi.core.configs.InferenceConfig attribute) (oumi.core.configs.QuantizationConfig attribute) (oumi.core.configs.TrainingConfig attribute) (oumi.core.configs.TuningConfig attribute) MODEL (oumi.core.registry.RegistryType attribute) MODEL_CONFIG (oumi.core.registry.RegistryType attribute) model_config (oumi.core.types.ContentItem attribute) (oumi.core.types.Conversation attribute) (oumi.core.types.Message attribute) (oumi.core.types.TemplatedMessage attribute) (oumi.judges.JudgeOutput attribute) (oumi.judges.JudgeOutputField attribute) (oumi.performance.telemetry.TelemetryState attribute) model_init_kwargs (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) model_input_features (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) model_kwargs (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) model_max_length (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) model_name (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) model_post_init() (oumi.core.types.ContentItem method) (oumi.core.types.Message method) model_revision (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) model_type (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) ModelMergerConfig (class in oumi.utils.verl_model_merger) ModelParameterCount (class in oumi.utils.torch_utils) ModelParams (class in oumi.core.configs) (class in oumi.core.configs.params.model_params) module oumi oumi.builders oumi.cli oumi.core oumi.core.analyze oumi.core.callbacks oumi.core.collators oumi.core.collators.text_collator_with_padding oumi.core.collators.text_completions_collator_with_padding oumi.core.collators.trl_data_collator_for_completion_only_lm oumi.core.collators.vision_language_collator_with_padding oumi.core.collators.vision_language_sft_collator oumi.core.configs oumi.core.configs.internal oumi.core.configs.internal.internal_model_config oumi.core.configs.internal.supported_models oumi.core.configs.params oumi.core.configs.params.base_params oumi.core.configs.params.data_params oumi.core.configs.params.deepspeed_params oumi.core.configs.params.evaluation_params oumi.core.configs.params.fsdp_params oumi.core.configs.params.generation_params oumi.core.configs.params.gkd_params oumi.core.configs.params.grpo_params oumi.core.configs.params.guided_decoding_params oumi.core.configs.params.judge_params oumi.core.configs.params.model_params oumi.core.configs.params.peft_params oumi.core.configs.params.profiler_params oumi.core.configs.params.remote_params oumi.core.configs.params.synthesis_params oumi.core.configs.params.telemetry_params oumi.core.configs.params.training_params oumi.core.configs.params.tuning_params oumi.core.datasets oumi.core.evaluation oumi.core.evaluation.backends oumi.core.evaluation.backends.alpaca_eval oumi.core.evaluation.backends.lm_harness oumi.core.evaluation.utils oumi.core.evaluation.utils.platform_prerequisites oumi.core.evaluation.utils.save_utils oumi.core.feature_generators oumi.core.inference oumi.core.launcher oumi.core.models oumi.core.processors oumi.core.processors.base_image_processor oumi.core.processors.base_processor oumi.core.processors.default_image_processor oumi.core.processors.default_processor oumi.core.registry oumi.core.synthesis oumi.core.synthesis.attribute_formatter oumi.core.synthesis.attribute_synthesizer oumi.core.synthesis.attribute_transformation oumi.core.synthesis.data_synthesizer oumi.core.synthesis.dataset_ingestion oumi.core.synthesis.dataset_planner oumi.core.synthesis.document_ingestion oumi.core.synthesis.synthesis_pipeline oumi.core.tokenizers oumi.core.trainers oumi.core.tuners oumi.core.types oumi.core.types.proto oumi.core.types.proto.generated oumi.datasets oumi.datasets.evaluation oumi.datasets.grpo oumi.datasets.grpo.rewards oumi.datasets.grpo.rollouts oumi.datasets.preference_tuning oumi.datasets.pretraining oumi.datasets.sft oumi.datasets.vision_language oumi.evaluation oumi.evaluation.registry oumi.inference oumi.judges oumi.launcher oumi.launcher.clients oumi.launcher.clients.local_client oumi.launcher.clients.polaris_client oumi.launcher.clients.sky_client oumi.launcher.clients.slurm_client oumi.launcher.clouds oumi.launcher.clusters oumi.launcher.clusters.frontier_cluster oumi.launcher.clusters.local_cluster oumi.launcher.clusters.perlmutter_cluster oumi.launcher.clusters.polaris_cluster oumi.launcher.clusters.sky_cluster oumi.launcher.clusters.slurm_cluster oumi.models oumi.performance oumi.performance.mfu oumi.performance.telemetry oumi.performance.torch_profiler_utils oumi.quantize oumi.utils oumi.utils.analysis_utils oumi.utils.batching oumi.utils.cache_utils oumi.utils.conversation_utils oumi.utils.debug_utils oumi.utils.device_utils oumi.utils.distributed_utils oumi.utils.git_utils oumi.utils.grpo_utils oumi.utils.hf_cache_utils oumi.utils.hf_utils oumi.utils.http oumi.utils.image_utils oumi.utils.io_utils oumi.utils.logging oumi.utils.math_utils oumi.utils.model_caching oumi.utils.packaging oumi.utils.peft_utils oumi.utils.placeholders oumi.utils.saver oumi.utils.serialization_utils oumi.utils.str_utils oumi.utils.torch_naming_heuristics oumi.utils.torch_utils oumi.utils.verl_model_merger oumi.utils.version_utils N n_trials (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) name (oumi.core.configs.internal.internal_model_config.InternalFeatureSpec attribute) (oumi.core.configs.JobConfig attribute) (oumi.core.configs.params.synthesis_params.SampledAttribute attribute) (oumi.core.configs.params.synthesis_params.SampledAttributeValue attribute) (oumi.core.configs.SampledAttribute attribute) (oumi.core.configs.SampledAttributeValue attribute) (oumi.core.launcher.JobStatus attribute) (oumi.launcher.clusters.slurm_cluster.SlurmCluster.ConnectionInfo property) (oumi.launcher.JobConfig attribute) name() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) NanInfDetectionCallback (class in oumi.core.callbacks) NATIVE (oumi.core.configs.InferenceEngineType attribute) NativeTextInferenceEngine (class in oumi.inference) nb_files (oumi.utils.hf_cache_utils.CachedItem attribute) nccl_default_timeout_minutes (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) NO_PREFETCH (oumi.core.configs.BackwardPrefetch attribute) (oumi.core.configs.params.fsdp_params.BackwardPrefetch attribute) NO_SHARD (oumi.core.configs.params.fsdp_params.ShardingStrategy attribute) (oumi.core.configs.ShardingStrategy attribute) NO_WRAP (oumi.core.configs.AutoWrapPolicy attribute) (oumi.core.configs.params.fsdp_params.AutoWrapPolicy attribute) NONE (oumi.core.configs.MixedPrecisionDtype attribute) (oumi.core.configs.params.training_params.MixedPrecisionDtype attribute) num_beams (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) num_fewshot (oumi.core.configs.LMHarnessTaskParams attribute) (oumi.core.configs.params.evaluation_params.LMHarnessTaskParams attribute) num_generations (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) num_nodes (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) num_retries (oumi.core.configs.AsyncEvaluationConfig attribute) num_samples (oumi.core.configs.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams attribute) (oumi.core.configs.SynthesisConfig attribute) num_train_epochs (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) num_workers (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) NVidiaGpuRuntimeInfo (class in oumi.utils.device_utils) NVME (oumi.core.configs.params.deepspeed_params.DeepSpeedOffloadDevice attribute) O offload_optimizer (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) offload_param (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) OffloadConfig (class in oumi.core.configs.params.deepspeed_params) OLORA (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) on_epoch_begin() (oumi.core.callbacks.TelemetryCallback method) on_epoch_end() (oumi.core.callbacks.TelemetryCallback method) on_log() (oumi.core.callbacks.HfMfuTrainerCallback method) (oumi.core.callbacks.MfuTrainerCallback method) (oumi.core.callbacks.NanInfDetectionCallback method) (oumi.core.callbacks.TelemetryCallback method) on_save() (oumi.core.callbacks.BitNetCallback method) on_step_begin() (oumi.core.callbacks.HfMfuTrainerCallback method) (oumi.core.callbacks.MfuTrainerCallback method) (oumi.core.callbacks.ProfilerStepCallback method) (oumi.core.callbacks.TelemetryCallback method) on_step_end() (oumi.core.callbacks.HfMfuTrainerCallback method) (oumi.core.callbacks.MfuTrainerCallback method) (oumi.core.callbacks.ProfilerStepCallback method) (oumi.core.callbacks.TelemetryCallback method) on_substep_end() (oumi.core.callbacks.ProfilerStepCallback method) (oumi.core.callbacks.TelemetryCallback method) on_train_end() (oumi.core.callbacks.TelemetryCallback method) OPENAI (oumi.core.configs.InferenceEngineType attribute) OpenAIInferenceEngine (class in oumi.inference) OpenbmbRlaifVDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) OpenO1SFTDataset (class in oumi.datasets) (class in oumi.datasets.sft) operation (oumi.utils.verl_model_merger.ModelMergerConfig attribute) optimize() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) optimizer (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) OPTUNA (oumi.core.configs.params.tuning_params.TunerType attribute) (oumi.core.configs.TunerType attribute) OptunaTuner (class in oumi.core.tuners) OrpoDpoMix40kDataset (class in oumi.datasets) (class in oumi.datasets.preference_tuning) oumi module OUMI (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) (oumi.core.synthesis.dataset_ingestion.DatasetStorageType attribute) oumi.builders module oumi.cli module oumi.core module oumi.core.analyze module oumi.core.callbacks module oumi.core.collators module oumi.core.collators.text_collator_with_padding module oumi.core.collators.text_completions_collator_with_padding module oumi.core.collators.trl_data_collator_for_completion_only_lm module oumi.core.collators.vision_language_collator_with_padding module oumi.core.collators.vision_language_sft_collator module oumi.core.configs module oumi.core.configs.internal module oumi.core.configs.internal.internal_model_config module oumi.core.configs.internal.supported_models module oumi.core.configs.params module oumi.core.configs.params.base_params module oumi.core.configs.params.data_params module oumi.core.configs.params.deepspeed_params module oumi.core.configs.params.evaluation_params module oumi.core.configs.params.fsdp_params module oumi.core.configs.params.generation_params module oumi.core.configs.params.gkd_params module oumi.core.configs.params.grpo_params module oumi.core.configs.params.guided_decoding_params module oumi.core.configs.params.judge_params module oumi.core.configs.params.model_params module oumi.core.configs.params.peft_params module oumi.core.configs.params.profiler_params module oumi.core.configs.params.remote_params module oumi.core.configs.params.synthesis_params module oumi.core.configs.params.telemetry_params module oumi.core.configs.params.training_params module oumi.core.configs.params.tuning_params module oumi.core.datasets module oumi.core.evaluation module oumi.core.evaluation.backends module oumi.core.evaluation.backends.alpaca_eval module oumi.core.evaluation.backends.lm_harness module oumi.core.evaluation.utils module oumi.core.evaluation.utils.platform_prerequisites module oumi.core.evaluation.utils.save_utils module oumi.core.feature_generators module oumi.core.inference module oumi.core.launcher module oumi.core.models module oumi.core.processors module oumi.core.processors.base_image_processor module oumi.core.processors.base_processor module oumi.core.processors.default_image_processor module oumi.core.processors.default_processor module oumi.core.registry module oumi.core.synthesis module oumi.core.synthesis.attribute_formatter module oumi.core.synthesis.attribute_synthesizer module oumi.core.synthesis.attribute_transformation module oumi.core.synthesis.data_synthesizer module oumi.core.synthesis.dataset_ingestion module oumi.core.synthesis.dataset_planner module oumi.core.synthesis.document_ingestion module oumi.core.synthesis.synthesis_pipeline module oumi.core.tokenizers module oumi.core.trainers module oumi.core.tuners module oumi.core.types module oumi.core.types.proto module oumi.core.types.proto.generated module oumi.datasets module oumi.datasets.evaluation module oumi.datasets.grpo module oumi.datasets.grpo.rewards module oumi.datasets.grpo.rollouts module oumi.datasets.preference_tuning module oumi.datasets.pretraining module oumi.datasets.sft module oumi.datasets.vision_language module oumi.evaluation module oumi.evaluation.registry module oumi.inference module oumi.judges module oumi.launcher module oumi.launcher.clients module oumi.launcher.clients.local_client module oumi.launcher.clients.polaris_client module oumi.launcher.clients.sky_client module oumi.launcher.clients.slurm_client module oumi.launcher.clouds module oumi.launcher.clusters module oumi.launcher.clusters.frontier_cluster module oumi.launcher.clusters.local_cluster module oumi.launcher.clusters.perlmutter_cluster module oumi.launcher.clusters.polaris_cluster module oumi.launcher.clusters.sky_cluster module oumi.launcher.clusters.slurm_cluster module oumi.models module oumi.performance module oumi.performance.mfu module oumi.performance.telemetry module oumi.performance.torch_profiler_utils module oumi.quantize module oumi.utils module oumi.utils.analysis_utils module oumi.utils.batching module oumi.utils.cache_utils module oumi.utils.conversation_utils module oumi.utils.debug_utils module oumi.utils.device_utils module oumi.utils.distributed_utils module oumi.utils.git_utils module oumi.utils.grpo_utils module oumi.utils.hf_cache_utils module oumi.utils.hf_utils module oumi.utils.http module oumi.utils.image_utils module oumi.utils.io_utils module oumi.utils.logging module oumi.utils.math_utils module oumi.utils.model_caching module oumi.utils.packaging module oumi.utils.peft_utils module oumi.utils.placeholders module oumi.utils.saver module oumi.utils.serialization_utils module oumi.utils.str_utils module oumi.utils.torch_naming_heuristics module oumi.utils.torch_utils module oumi.utils.verl_model_merger module oumi.utils.version_utils module output_dir (oumi.core.configs.EvaluationConfig attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) output_fields (oumi.judges.JudgeOutput attribute) output_format (oumi.core.configs.QuantizationConfig attribute) output_path (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.InferenceConfig attribute) (oumi.core.configs.QuantizationConfig attribute) (oumi.core.configs.SynthesisConfig attribute) (oumi.quantize.QuantizationResult attribute) overlap_comm (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) overlap_events (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) OVERRUN (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) P pack (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) package_name (oumi.utils.packaging.PackagePrerequisites attribute) PackagePrerequisites (class in oumi.utils.packaging) PackedSftDataset (class in oumi.core.datasets) PAD_LEFT (oumi.core.configs.internal.internal_model_config.InternalPaddingSide attribute) PAD_RIGHT (oumi.core.configs.internal.internal_model_config.InternalPaddingSide attribute) pad_sequences() (in module oumi.utils.torch_utils) pad_sequences_left_side() (in module oumi.utils.torch_utils) pad_sequences_right_side() (in module oumi.utils.torch_utils) pad_to_max_dim_and_stack() (in module oumi.utils.torch_utils) padding_side (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) params (oumi.core.configs.SampleAnalyzerParams attribute) ParamType (class in oumi.core.configs.params.tuning_params) PARASAIL (oumi.core.configs.InferenceEngineType attribute) ParasailInferenceEngine (class in oumi.inference) parse_cluster_name() (oumi.launcher.clusters.slurm_cluster.SlurmCluster static method) parsed_output (oumi.judges.JudgeOutput attribute) passthrough_attributes (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) patch_model_generation_config() (oumi.utils.verl_model_merger.BaseModelMerger method) path (oumi.core.configs.DocumentSource attribute) (oumi.core.configs.params.synthesis_params.DatasetSource attribute) (oumi.core.configs.params.synthesis_params.DocumentSource attribute) peft (oumi.core.configs.TrainingConfig attribute) peft_save_mode (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) PeftParams (class in oumi.core.configs) (class in oumi.core.configs.params.peft_params) PeftSaveMode (class in oumi.core.configs) (class in oumi.core.configs.params.peft_params) PENDING (oumi.core.launcher.JobState attribute) per_device_eval_batch_size (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) per_device_train_batch_size (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) performance_state (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) PerlmutterCloud (class in oumi.launcher.clouds) PerlmutterCluster (class in oumi.launcher.clusters.perlmutter_cluster) PerlmutterCluster.SupportedQueues (class in oumi.launcher.clusters.perlmutter_cluster) PileV1Dataset (class in oumi.datasets) (class in oumi.datasets.pretraining) pin_memory (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) PISA (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) PISSA_NITER (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) PixmoAskModelAnythingDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) PixmoCapDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) PixmoCapQADataset (class in oumi.datasets) (class in oumi.datasets.vision_language) plan() (oumi.core.synthesis.dataset_planner.DatasetPlanner method) PolarisClient (class in oumi.launcher.clients.polaris_client) PolarisClient.SupportedQueues (class in oumi.launcher.clients.polaris_client) PolarisCloud (class in oumi.launcher.clouds) PolarisCluster (class in oumi.launcher.clusters.polaris_cluster) PolarisResponse (class in oumi.launcher.clients.polaris_client) politeness_policy (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) polling_interval (oumi.core.configs.AsyncEvaluationConfig attribute) possible_values (oumi.core.configs.params.synthesis_params.SampledAttribute attribute) (oumi.core.configs.SampledAttribute attribute) postprocessing_params (oumi.core.configs.GeneratedAttribute attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttribute attribute) power_limit_watts (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) power_usage_watts (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) precision (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) PREEMPT (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) PREEMPTABLE (oumi.launcher.clients.polaris_client.PolarisClient.SupportedQueues attribute) PREMIUM (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) presence_penalty (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) PretrainingAsyncTextDataset (class in oumi.core.datasets) print_config() (oumi.core.configs.BaseConfig method) print_summary() (oumi.performance.telemetry.TelemetryTracker method) private (oumi.utils.verl_model_merger.ModelMergerConfig attribute) processor_kwargs (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) processor_name (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) PROD (oumi.launcher.clients.polaris_client.PolarisClient.SupportedQueues attribute) profile_memory (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) profiler (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) ProfilerParams (class in oumi.core.configs) (class in oumi.core.configs.params.profiler_params) ProfilerScheduleParams (class in oumi.core.configs.params.profiler_params) ProfilerStepCallback (class in oumi.core.callbacks) prompt() (oumi.core.datasets.BaseSftDataset method) prompt_template (oumi.core.configs.params.judge_params.JudgeParams attribute) prompt_template_placeholders (oumi.core.configs.params.judge_params.JudgeParams attribute) PromptResponseDataset (class in oumi.datasets) (class in oumi.datasets.sft) put() (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) put_recursive() (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) Q q_lora (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) q_lora_bits (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) quantization_method (oumi.quantize.QuantizationResult attribute) QuantizationConfig (class in oumi.core.configs) QuantizationResult (class in oumi.quantize) quantize() (in module oumi) (in module oumi.quantize) (oumi.quantize.AwqQuantization method) (oumi.quantize.BaseQuantization method) (oumi.quantize.BitsAndBytesQuantization method) quantized_size_bytes (oumi.quantize.QuantizationResult attribute) query() (oumi.core.analyze.DatasetAnalyzer method) query_conversations() (oumi.core.analyze.DatasetAnalyzer method) queue() (oumi.launcher.clients.sky_client.SkyClient method) queue_depth (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) R raise_if_requirements_not_met() (oumi.quantize.AwqQuantization method) (oumi.quantize.BaseQuantization method) (oumi.quantize.BitsAndBytesQuantization method) RANDOM (oumi.core.configs.LoraWeightInitialization attribute) (oumi.core.configs.params.peft_params.LoraWeightInitialization attribute) RAW (oumi.core.configs.JudgeResponseFormat attribute) (oumi.core.configs.params.judge_params.JudgeResponseFormat attribute) raw() (oumi.core.datasets.BaseMapDataset method) raw_output (oumi.judges.JudgeOutput attribute) raw_processor (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) read() (oumi.core.synthesis.dataset_ingestion.DatasetReader method) (oumi.core.synthesis.document_ingestion.DocumentReader method) readline() (oumi.launcher.clients.sky_client.SkyLogStream method) (oumi.launcher.clients.slurm_client.SlurmLogStream method) REALTIME (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) record_gpu_temperature() (oumi.performance.telemetry.TelemetryTracker method) record_shapes (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) recovery_threshold (oumi.core.configs.params.remote_params.AdaptiveConcurrencyParams attribute) RedPajamaDataV1Dataset (class in oumi.datasets) (class in oumi.datasets.pretraining) RedPajamaDataV2Dataset (class in oumi.datasets) (class in oumi.datasets.pretraining) reduce_bucket_size (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) reduce_scatter (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) regex (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.GuidedDecodingParams attribute) (oumi.core.configs.params.guided_decoding_params.GuidedDecodingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) region (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) register() (in module oumi.core.registry) (oumi.core.registry.Registry method) register_cloud_builder() (in module oumi.core.registry) register_dataset() (in module oumi.core.registry) register_evaluation_function() (in module oumi.core.registry) register_sample_analyzer() (in module oumi.core.registry) Registry (class in oumi.core.registry) RegistryType (class in oumi.core.registry) REGULAR (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) REMOTE (oumi.core.configs.InferenceEngineType attribute) remote_params (oumi.core.configs.InferenceConfig attribute) REMOTE_VLLM (oumi.core.configs.InferenceEngineType attribute) RemoteInferenceEngine (class in oumi.inference) RemoteParams (class in oumi.core.configs) (class in oumi.core.configs.params.remote_params) RemoteVLLMInferenceEngine (class in oumi.inference) remove_excessive_images() (in module oumi.utils.conversation_utils) remove_excessive_images_from_conversation() (in module oumi.utils.conversation_utils) remove_unused_columns (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) repeat (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) repetition_penalty (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) replace_template_variables() (oumi.core.configs.params.judge_params.JudgeParams method) repo_id (oumi.utils.hf_cache_utils.CachedItem attribute) repo_path (oumi.utils.hf_cache_utils.CachedItem attribute) repo_type (oumi.utils.hf_cache_utils.CachedItem attribute) require_torchdata() (in module oumi.utils.packaging) required (oumi.core.configs.internal.internal_model_config.InternalFeatureSpec attribute) requires_optimizer_offloading() (oumi.core.configs.params.deepspeed_params.DeepSpeedParams method) requires_param_offloading() (oumi.core.configs.params.deepspeed_params.DeepSpeedParams method) resolve_placeholders() (in module oumi.utils.placeholders) resolve_transformer_layer_cls_string_as_module_set() (in module oumi.utils.torch_naming_heuristics) resources (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) response_format (oumi.core.configs.params.judge_params.JudgeParams attribute) (oumi.judges.JudgeOutput attribute) resume_from_checkpoint (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) retry_auth() (in module oumi.launcher.clients.polaris_client) (in module oumi.launcher.clients.slurm_client) retry_backoff_base (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) retry_backoff_max (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) REWARD_FUNCTION (oumi.core.registry.RegistryType attribute) reward_functions (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) Role (class in oumi.core.types) role (oumi.core.configs.params.synthesis_params.TextMessage attribute) (oumi.core.configs.TextMessage attribute) (oumi.core.types.Message attribute) (oumi.core.types.TemplatedMessage attribute) rollout_function (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) ROLLOUT_FUNCTION (oumi.core.registry.RegistryType attribute) round_robin_gradients (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) row_limit (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) run (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) run() (in module oumi.launcher) (oumi.launcher.Launcher method) run_commands() (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) run_job() (oumi.core.launcher.BaseCluster method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) run_name (oumi.core.configs.EvaluationConfig attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) RUNNING (oumi.core.launcher.JobState attribute) RUNPOD (oumi.launcher.clients.sky_client.SkyClient.SupportedClouds attribute) S SafeDict (class in oumi.utils.placeholders) SAMBANOVA (oumi.core.configs.InferenceEngineType attribute) SambanovaInferenceEngine (class in oumi.inference) SAMPLE_ANALYZER (oumi.core.registry.RegistryType attribute) sample_count (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) sample_rate (oumi.core.configs.AttributeCombination attribute) (oumi.core.configs.params.synthesis_params.AttributeCombination attribute) (oumi.core.configs.params.synthesis_params.SampledAttributeValue attribute) (oumi.core.configs.SampledAttributeValue attribute) SampleAnalyzer (class in oumi.core.analyze) SampleAnalyzerParams (class in oumi.core.configs) sampled_attributes (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) SampledAttribute (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) SampledAttributeValue (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) sanitize_negative_labels (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) sanitize_run_name() (in module oumi.utils.str_utils) save_config() (oumi.core.processors.base_processor.BaseProcessor method) (oumi.core.processors.default_processor.DefaultProcessor method) save_dir (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) save_epoch (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) save_evaluation_output() (in module oumi.core.evaluation.utils.save_utils) save_final_model (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) save_hf_model_and_tokenizer() (oumi.utils.verl_model_merger.BaseModelMerger method) save_infer_prob() (in module oumi.utils.saver) save_infer_prob_csv() (in module oumi.utils.saver) save_json() (in module oumi.utils.io_utils) save_jsonlines() (in module oumi.utils.io_utils) save_model() (oumi.core.trainers.BaseTrainer method) (oumi.core.trainers.HuggingFaceTrainer method) (oumi.core.trainers.Trainer method) (oumi.core.trainers.VerlGrpoTrainer method) save_output() (oumi.core.evaluation.Evaluator method) save_state() (oumi.core.trainers.BaseTrainer method) (oumi.core.trainers.HuggingFaceTrainer method) (oumi.core.trainers.Trainer method) (oumi.core.trainers.VerlGrpoTrainer method) save_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) save_study() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) schedule (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) SchedulerType (class in oumi.core.configs) (class in oumi.core.configs.params.training_params) seed (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) segment() (oumi.core.synthesis.document_ingestion.DocumentSegmenter method) segment_batch() (oumi.core.synthesis.document_ingestion.DocumentSegmenter method) segment_length (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) segment_overlap (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) segmentation_params (oumi.core.configs.DocumentSource attribute) (oumi.core.configs.params.synthesis_params.DocumentSource attribute) segmentation_strategy (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) SegmentationStrategy (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) seq_kd (oumi.core.configs.params.gkd_params.GkdParams attribute) set_oumi_install_editable() (in module oumi.utils.str_utils) setup (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) sgd_momentum (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) SGLANG (oumi.core.configs.InferenceEngineType attribute) SGLangInferenceEngine (class in oumi.inference) shard_for_eval (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) SHARD_GRAD_OP (oumi.core.configs.params.fsdp_params.ShardingStrategy attribute) (oumi.core.configs.ShardingStrategy attribute) SHARDED_STATE_DICT (oumi.core.configs.params.fsdp_params.StateDictType attribute) (oumi.core.configs.StateDictType attribute) sharding_strategy (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) ShardingStrategy (class in oumi.core.configs) (class in oumi.core.configs.params.fsdp_params) SHARED (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) SHARED_INTERACTIVE (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) SHARED_OVERRUN (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster.SupportedQueues attribute) should_use_rich_logging() (in module oumi.utils.logging) shuffle (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) shuffle_buffer_size (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) SimpleJudge (class in oumi.judges) simplify_transformer_layer_cls_string() (in module oumi.utils.torch_naming_heuristics) single_submit (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) size (oumi.utils.hf_cache_utils.CachedItem attribute) SIZE_BASED_WRAP (oumi.core.configs.AutoWrapPolicy attribute) (oumi.core.configs.params.fsdp_params.AutoWrapPolicy attribute) size_bytes (oumi.utils.hf_cache_utils.CachedItem attribute) skip_first (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) skip_special_tokens (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) SkyClient (class in oumi.launcher.clients.sky_client) SkyClient.SupportedClouds (class in oumi.launcher.clients.sky_client) SkyCloud (class in oumi.launcher.clouds) SkyCluster (class in oumi.launcher.clusters.sky_cluster) SkyLogStream (class in oumi.launcher.clients.sky_client) SlimPajamaDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) SlurmClient (class in oumi.launcher.clients.slurm_client) SlurmCloud (class in oumi.launcher.clouds) SlurmCluster (class in oumi.launcher.clusters.slurm_cluster) SlurmCluster.ConnectionInfo (class in oumi.launcher.clusters.slurm_cluster) SlurmLogStream (class in oumi.launcher.clients.slurm_client) SlurmResponse (class in oumi.launcher.clients.slurm_client) source (oumi.core.configs.StorageMount attribute) (oumi.launcher.StorageMount attribute) split (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) stage3_gather_16bit_weights_on_model_save (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) stage3_max_live_parameters (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) stage3_max_reuse_distance (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) stage3_param_persistence_threshold (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) stage3_prefetch_bucket_size (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) StarCoderDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) start_time (oumi.performance.telemetry.TelemetryState attribute) state (oumi.core.launcher.JobStatus attribute) state_dict() (oumi.performance.telemetry.TelemetryTracker method) state_dict_type (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) StateDictType (class in oumi.core.configs) (class in oumi.core.configs.params.fsdp_params) status (oumi.core.launcher.JobStatus attribute) status() (in module oumi.launcher) (oumi.launcher.clients.sky_client.SkyClient method) (oumi.launcher.Launcher method) stderr (oumi.launcher.clients.polaris_client.PolarisResponse attribute) (oumi.launcher.clients.slurm_client.SlurmResponse attribute) stdout (oumi.launcher.clients.polaris_client.PolarisResponse attribute) (oumi.launcher.clients.slurm_client.SlurmResponse attribute) steps_per_print (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) stop() (in module oumi.launcher) (oumi.core.launcher.BaseCluster method) (oumi.launcher.clients.sky_client.SkyClient method) (oumi.launcher.clusters.frontier_cluster.FrontierCluster method) (oumi.launcher.clusters.local_cluster.LocalCluster method) (oumi.launcher.clusters.perlmutter_cluster.PerlmutterCluster method) (oumi.launcher.clusters.polaris_cluster.PolarisCluster method) (oumi.launcher.clusters.sky_cluster.SkyCluster method) (oumi.launcher.clusters.slurm_cluster.SlurmCluster method) (oumi.launcher.Launcher method) stop_strings (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) stop_token_ids (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) storage (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) storage_mounts (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) StorageMount (class in oumi.core.configs) (class in oumi.launcher) store (oumi.core.configs.StorageMount attribute) (oumi.launcher.StorageMount attribute) str_to_bool() (in module oumi.utils.str_utils) str_to_float_list() (in module oumi.utils.saver) strategy (oumi.core.configs.SynthesisConfig attribute) strategy_params (oumi.core.configs.SynthesisConfig attribute) stream (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) STRING (oumi.core.configs.params.synthesis_params.TransformationType attribute) (oumi.core.configs.TransformationType attribute) string_transform (oumi.core.configs.params.synthesis_params.TransformationStrategy attribute) (oumi.core.configs.TransformationStrategy attribute) strip_whitespace (oumi.core.configs.GeneratedAttributePostprocessingParams attribute) (oumi.core.configs.params.synthesis_params.GeneratedAttributePostprocessingParams attribute) sub_group_size (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) submit_job() (oumi.launcher.clients.local_client.LocalClient method) (oumi.launcher.clients.polaris_client.PolarisClient method) (oumi.launcher.clients.slurm_client.SlurmClient method) subset (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) SUCCEEDED (oumi.core.launcher.JobState attribute) suggest_parameters() (oumi.core.tuners.BaseTuner method) (oumi.core.tuners.OptunaTuner method) supported_formats (oumi.quantize.AwqQuantization attribute) (oumi.quantize.BaseQuantization attribute) (oumi.quantize.BitsAndBytesQuantization attribute) supported_methods (oumi.quantize.AwqQuantization attribute) (oumi.quantize.BaseQuantization attribute) (oumi.quantize.BitsAndBytesQuantization attribute) supports_format() (oumi.quantize.BaseQuantization method) supports_method() (oumi.quantize.BaseQuantization method) supports_multiple_images (oumi.core.configs.internal.internal_model_config.InternalVisualModelConfig attribute) sync_module_states (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) SynthesisConfig (class in oumi.core.configs) SynthesisPipeline (class in oumi.core.synthesis.synthesis_pipeline) synthesize() (in module oumi) (oumi.core.synthesis.attribute_synthesizer.AttributeSynthesizer method) (oumi.core.synthesis.data_synthesizer.DataSynthesizer method) (oumi.core.synthesis.synthesis_pipeline.SynthesisPipeline method) SYSTEM (oumi.core.types.Role attribute) system_instruction (oumi.core.configs.params.judge_params.JudgeParams attribute) system_prompt_with_context (oumi.datasets.AlpacaDataset attribute) (oumi.datasets.AlpacaEvalDataset attribute) (oumi.datasets.evaluation.AlpacaEvalDataset attribute) (oumi.datasets.sft.AlpacaDataset attribute) system_prompt_without_context (oumi.datasets.AlpacaDataset attribute) (oumi.datasets.AlpacaEvalDataset attribute) (oumi.datasets.evaluation.AlpacaEvalDataset attribute) (oumi.datasets.sft.AlpacaDataset attribute) T target_col (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) target_dir (oumi.utils.verl_model_merger.ModelMergerConfig attribute) task (oumi.core.datasets.BaseSftDataset property) task_name (oumi.core.configs.EvaluationTaskParams attribute) (oumi.core.configs.params.evaluation_params.EvaluationTaskParams attribute) tasks (oumi.core.configs.EvaluationConfig attribute) teacher_model_init_kwargs (oumi.core.configs.params.gkd_params.GkdParams attribute) teacher_model_name_or_path (oumi.core.configs.params.gkd_params.GkdParams attribute) telemetry (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) telemetry_dir (oumi.core.configs.params.telemetry_params.TelemetryParams attribute) (oumi.core.configs.params.training_params.TrainingParams property) (oumi.core.configs.params.tuning_params.TuningParams property) (oumi.core.configs.TelemetryParams attribute) (oumi.core.configs.TrainingParams property) (oumi.core.configs.TuningParams property) TelemetryCallback (class in oumi.core.callbacks) TelemetryParams (class in oumi.core.configs) (class in oumi.core.configs.params.telemetry_params) TelemetryState (class in oumi.performance.telemetry) TelemetryTracker (class in oumi.performance.telemetry) temperature (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) (oumi.core.configs.params.gkd_params.GkdParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) template (oumi.core.types.TemplatedMessage attribute) template_variables (oumi.core.configs.params.judge_params.JudgeParams attribute) TemplatedMessage (class in oumi.core.types) test (oumi.core.configs.DataParams attribute) TEST (oumi.core.configs.DatasetSplit attribute) test (oumi.core.configs.params.data_params.DataParams attribute) TEST (oumi.core.configs.params.data_params.DatasetSplit attribute) test_hf_dir (oumi.utils.verl_model_merger.ModelMergerConfig attribute) TEXT (oumi.core.configs.JudgeOutputType attribute) (oumi.core.configs.params.judge_params.JudgeOutputType attribute) (oumi.core.types.Type attribute) text_col (oumi.core.datasets.BaseSftDataset property) text_content_items (oumi.core.types.Message property) text_items (oumi.core.types.ContentItemCounts attribute) TextCollatorWithPadding (class in oumi.core.collators.text_collator_with_padding) TextCompletionsCollatorWithPadding (class in oumi.core.collators.text_completions_collator_with_padding) TextConversation (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) TextMessage (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) TextSftJsonLinesDataset (class in oumi.datasets) (class in oumi.datasets.sft) TheCauldronDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) TheStackDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) thread_count (oumi.core.configs.params.deepspeed_params.OffloadConfig attribute) tie_word_embedding (oumi.utils.verl_model_merger.ModelMergerConfig attribute) timer() (oumi.performance.telemetry.TelemetryTracker method) TimerContext (class in oumi.performance.telemetry) TinyStoriesDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) TinyTextbooksDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) TldrGrpoDataset (class in oumi.datasets) (class in oumi.datasets.grpo) to_bits_and_bytes() (oumi.core.configs.params.peft_params.PeftParams method) (oumi.core.configs.PeftParams method) to_conversation() (oumi.core.configs.params.synthesis_params.TextConversation method) (oumi.core.configs.TextConversation method) to_deepspeed() (oumi.core.configs.params.deepspeed_params.DeepSpeedParams method) to_dict() (oumi.core.evaluation.EvaluationResult method) (oumi.core.types.Conversation method) to_hf() (oumi.core.configs.params.training_params.TrainingParams method) (oumi.core.configs.TrainingParams method) (oumi.core.datasets.BaseIterableDataset method) (oumi.core.datasets.BaseMapDataset method) to_hf_trainer_kwargs() (oumi.core.configs.GrpoParams method) (oumi.core.configs.params.gkd_params.GkdParams method) (oumi.core.configs.params.grpo_params.GrpoParams method) to_json() (oumi.core.types.Conversation method) (oumi.judges.JudgeOutput method) to_lora() (oumi.core.configs.params.peft_params.PeftParams method) (oumi.core.configs.PeftParams method) to_message() (oumi.core.configs.params.synthesis_params.TextMessage method) (oumi.core.configs.TextMessage method) to_proto() (oumi.core.types.ContentItem method) (oumi.core.types.Conversation method) (oumi.core.types.Message method) to_torch() (oumi.core.configs.BackwardPrefetch method) (oumi.core.configs.params.fsdp_params.BackwardPrefetch method) (oumi.core.configs.params.fsdp_params.ShardingStrategy method) (oumi.core.configs.params.fsdp_params.StateDictType method) (oumi.core.configs.ShardingStrategy method) (oumi.core.configs.StateDictType method) to_yaml() (oumi.core.configs.BaseConfig method) TOGETHER (oumi.core.configs.InferenceEngineType attribute) TogetherInferenceEngine (class in oumi.inference) tokenize() (oumi.core.datasets.BasePretrainingDataset method) (oumi.core.datasets.BaseSftDataset method) tokenizer (oumi.core.configs.DocumentSegmentationParams attribute) (oumi.core.configs.params.synthesis_params.DocumentSegmentationParams attribute) (oumi.core.processors.base_processor.BaseProcessor property) (oumi.core.processors.default_processor.DefaultProcessor property) tokenizer_config (oumi.core.configs.AnalyzeConfig attribute) tokenizer_kwargs (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) tokenizer_name (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) tokenizer_pad_token (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) TOKENS (oumi.core.configs.params.synthesis_params.SegmentationStrategy attribute) (oumi.core.configs.SegmentationStrategy attribute) TOOL (oumi.core.types.Role attribute) top_p (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) torch_call() (oumi.core.collators.trl_data_collator_for_completion_only_lm.DataCollatorForCompletionOnlyLM method) torch_dtype_str (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) torch_profile() (in module oumi.performance.torch_profiler_utils) TorchJsonEncoder (class in oumi.utils.serialization_utils) total_items (oumi.core.types.ContentItemCounts attribute) track_gpu_temperature (oumi.core.configs.params.telemetry_params.TelemetryParams attribute) (oumi.core.configs.TelemetryParams attribute) train (oumi.core.configs.DataParams attribute) TRAIN (oumi.core.configs.DatasetSplit attribute) train (oumi.core.configs.params.data_params.DataParams attribute) TRAIN (oumi.core.configs.params.data_params.DatasetSplit attribute) train() (in module oumi) (oumi.core.trainers.BaseTrainer method) (oumi.core.trainers.HuggingFaceTrainer method) (oumi.core.trainers.Trainer method) (oumi.core.trainers.VerlGrpoTrainer method) train_batch_size (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) train_micro_batch_size_per_gpu (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) trainable_params (oumi.utils.torch_utils.ModelParameterCount attribute) trainable_params_percent (oumi.utils.torch_utils.ModelParameterCount property) Trainer (class in oumi.core.trainers) trainer_kwargs (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) trainer_type (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TrainingParams attribute) (oumi.core.configs.TuningParams attribute) TrainerType (class in oumi.core.configs) (class in oumi.core.configs.params.training_params) training (oumi.core.configs.TrainingConfig attribute) TrainingConfig (class in oumi.core.configs) TrainingParams (class in oumi.core.configs) (class in oumi.core.configs.params.training_params) transform() (oumi.core.datasets.BaseDpoDataset method) (oumi.core.datasets.BaseExperimentalGrpoDataset method) (oumi.core.datasets.BaseExperimentalKtoDataset method) (oumi.core.datasets.BaseIterableDataset method) (oumi.core.datasets.BaseMapDataset method) (oumi.core.datasets.BasePretrainingDataset method) (oumi.core.datasets.BaseSftDataset method) (oumi.core.datasets.PackedSftDataset method) (oumi.core.datasets.VisionLanguageDpoDataset method) (oumi.core.datasets.VisionLanguageSftDataset method) (oumi.core.synthesis.attribute_transformation.AttributeTransformer method) (oumi.datasets.grpo.BerryBenchGrpoDataset method) (oumi.datasets.grpo.CountdownGrpoDataset method) (oumi.datasets.grpo.Gsm8kGrpoDataset method) (oumi.datasets.grpo.LetterCountGrpoDataset method) (oumi.datasets.LetterCountGrpoDataset method) transform_conversation() (oumi.core.datasets.BaseExperimentalGrpoDataset method) (oumi.core.datasets.BaseSftDataset method) (oumi.core.datasets.VisionLanguageSftDataset method) (oumi.core.feature_generators.BaseConversationFeatureGenerator method) (oumi.core.feature_generators.VisionLanguageConversationFeatureGenerator method) (oumi.datasets.AlpacaDataset method) (oumi.datasets.AlpacaEvalDataset method) (oumi.datasets.ArgillaDollyDataset method) (oumi.datasets.ArgillaMagpieUltraDataset method) (oumi.datasets.AyaDataset method) (oumi.datasets.ChatqaDataset method) (oumi.datasets.ChatRAGBenchDataset method) (oumi.datasets.COCOCaptionsDataset method) (oumi.datasets.DebugSftDataset method) (oumi.datasets.evaluation.AlpacaEvalDataset method) (oumi.datasets.Flickr30kDataset method) (oumi.datasets.Geometry3kDataset method) (oumi.datasets.grpo.BerryBenchGrpoDataset method) (oumi.datasets.grpo.CountdownGrpoDataset method) (oumi.datasets.grpo.Gsm8kGrpoDataset method) (oumi.datasets.grpo.LetterCountGrpoDataset method) (oumi.datasets.HuggingFaceDataset method) (oumi.datasets.HuggingFaceVisionDataset method) (oumi.datasets.LetterCountGrpoDataset method) (oumi.datasets.LlavaInstructMixVsftDataset method) (oumi.datasets.MagpieProDataset method) (oumi.datasets.MnistSftDataset method) (oumi.datasets.PixmoAskModelAnythingDataset method) (oumi.datasets.PixmoCapDataset method) (oumi.datasets.PixmoCapQADataset method) (oumi.datasets.PromptResponseDataset method) (oumi.datasets.sft.AlpacaDataset method) (oumi.datasets.sft.ArgillaDollyDataset method) (oumi.datasets.sft.ArgillaMagpieUltraDataset method) (oumi.datasets.sft.AyaDataset method) (oumi.datasets.sft.ChatqaDataset method) (oumi.datasets.sft.ChatRAGBenchDataset method) (oumi.datasets.sft.HuggingFaceDataset method) (oumi.datasets.sft.MagpieProDataset method) (oumi.datasets.sft.PromptResponseDataset method) (oumi.datasets.sft.TextSftJsonLinesDataset method) (oumi.datasets.sft.Tulu3MixtureDataset method) (oumi.datasets.sft.UltrachatH4Dataset method) (oumi.datasets.sft.WildChatDataset method) (oumi.datasets.TextSftJsonLinesDataset method) (oumi.datasets.TheCauldronDataset method) (oumi.datasets.Tulu3MixtureDataset method) (oumi.datasets.UltrachatH4Dataset method) (oumi.datasets.vision_language.COCOCaptionsDataset method) (oumi.datasets.vision_language.Flickr30kDataset method) (oumi.datasets.vision_language.Geometry3kDataset method) (oumi.datasets.vision_language.HuggingFaceVisionDataset method) (oumi.datasets.vision_language.LlavaInstructMixVsftDataset method) (oumi.datasets.vision_language.MnistSftDataset method) (oumi.datasets.vision_language.PixmoAskModelAnythingDataset method) (oumi.datasets.vision_language.PixmoCapDataset method) (oumi.datasets.vision_language.PixmoCapQADataset method) (oumi.datasets.vision_language.TheCauldronDataset method) (oumi.datasets.vision_language.VLJsonlinesDataset method) (oumi.datasets.vision_language.Vqav2SmallDataset method) (oumi.datasets.VLJsonlinesDataset method) (oumi.datasets.Vqav2SmallDataset method) (oumi.datasets.WildChatDataset method) transform_conversations() (oumi.core.feature_generators.BaseConversationFeatureGenerator method) (oumi.core.feature_generators.VisionLanguageConversationFeatureGenerator method) transform_num_workers (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) (oumi.core.datasets.BaseMapDataset attribute) transform_preference() (oumi.core.datasets.BaseDpoDataset method) (oumi.core.datasets.VisionLanguageDpoDataset method) transformation_strategy (oumi.core.configs.params.synthesis_params.TransformedAttribute attribute) (oumi.core.configs.TransformedAttribute attribute) TransformationStrategy (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) TransformationType (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) transformed_attributes (oumi.core.configs.GeneralSynthesisParams attribute) (oumi.core.configs.params.synthesis_params.GeneralSynthesisParams attribute) TransformedAttribute (class in oumi.core.configs) (class in oumi.core.configs.params.synthesis_params) TRANSFORMER_BASED_WRAP (oumi.core.configs.AutoWrapPolicy attribute) (oumi.core.configs.params.fsdp_params.AutoWrapPolicy attribute) transformer_layer_cls (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) TRL_DPO (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) TRL_GKD (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) TRL_GRPO (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) TRL_KTO (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) TRL_SFT (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) TrlDpoTrainer (class in oumi.core.trainers) truncate_text() (oumi.core.processors.base_processor.BaseProcessor method) (oumi.core.processors.default_processor.DefaultProcessor method) truncate_text_in_content_items() (in module oumi.utils.conversation_utils) truncate_text_pieces_to_max_tokens_limit() (in module oumi.utils.str_utils) truncate_to_max_tokens_limit() (in module oumi.utils.str_utils) trust_remote_code (oumi.core.configs.AnalyzeConfig attribute) (oumi.core.configs.DatasetParams attribute) (oumi.core.configs.ModelParams attribute) (oumi.core.configs.params.data_params.DatasetParams attribute) (oumi.core.configs.params.model_params.ModelParams attribute) (oumi.core.datasets.BaseDpoDataset attribute) (oumi.core.datasets.BaseExperimentalDpoDataset attribute) (oumi.core.datasets.BaseExperimentalGrpoDataset attribute) (oumi.core.datasets.BaseExperimentalKtoDataset attribute) (oumi.core.datasets.BaseIterableDataset attribute) (oumi.core.datasets.BaseMapDataset attribute) (oumi.core.datasets.BaseSftDataset attribute) (oumi.core.datasets.PackedSftDataset attribute) (oumi.core.datasets.VisionLanguageDpoDataset attribute) (oumi.core.datasets.VisionLanguageSftDataset attribute) (oumi.datasets.AlpacaDataset attribute) (oumi.datasets.AlpacaEvalDataset attribute) (oumi.datasets.ArgillaDollyDataset attribute) (oumi.datasets.ArgillaMagpieUltraDataset attribute) (oumi.datasets.AyaDataset attribute) (oumi.datasets.ChatqaDataset attribute) (oumi.datasets.ChatqaTatqaDataset attribute) (oumi.datasets.CoALMDataset attribute) (oumi.datasets.COCOCaptionsDataset attribute) (oumi.datasets.DebugSftDataset attribute) (oumi.datasets.DocmatixDataset attribute) (oumi.datasets.evaluation.AlpacaEvalDataset attribute) (oumi.datasets.Flickr30kDataset attribute) (oumi.datasets.Geometry3kDataset attribute) (oumi.datasets.grpo.BerryBenchGrpoDataset attribute) (oumi.datasets.grpo.CountdownGrpoDataset attribute) (oumi.datasets.grpo.Gsm8kGrpoDataset attribute) (oumi.datasets.grpo.LetterCountGrpoDataset attribute) (oumi.datasets.grpo.TldrGrpoDataset attribute) (oumi.datasets.HuggingFaceDataset attribute) (oumi.datasets.HuggingFaceVisionDataset attribute) (oumi.datasets.LetterCountGrpoDataset attribute) (oumi.datasets.LlavaInstructMixVsftDataset attribute) (oumi.datasets.LmmsLabMultimodalOpenR1Dataset attribute) (oumi.datasets.MagpieProDataset attribute) (oumi.datasets.MnistSftDataset attribute) (oumi.datasets.OpenbmbRlaifVDataset attribute) (oumi.datasets.OpenO1SFTDataset attribute) (oumi.datasets.OrpoDpoMix40kDataset attribute) (oumi.datasets.PixmoAskModelAnythingDataset attribute) (oumi.datasets.PixmoCapDataset attribute) (oumi.datasets.PixmoCapQADataset attribute) (oumi.datasets.preference_tuning.KtoMix14kDataset attribute) (oumi.datasets.preference_tuning.OrpoDpoMix40kDataset attribute) (oumi.datasets.PromptResponseDataset attribute) (oumi.datasets.sft.AlpacaDataset attribute) (oumi.datasets.sft.ArgillaDollyDataset attribute) (oumi.datasets.sft.ArgillaMagpieUltraDataset attribute) (oumi.datasets.sft.AyaDataset attribute) (oumi.datasets.sft.ChatqaDataset attribute) (oumi.datasets.sft.ChatqaTatqaDataset attribute) (oumi.datasets.sft.ChatRAGBenchDataset attribute) (oumi.datasets.sft.CoALMDataset attribute) (oumi.datasets.sft.HuggingFaceDataset attribute) (oumi.datasets.sft.MagpieProDataset attribute) (oumi.datasets.sft.OpenO1SFTDataset attribute) (oumi.datasets.sft.PromptResponseDataset attribute) (oumi.datasets.sft.TextSftJsonLinesDataset attribute) (oumi.datasets.sft.Tulu3MixtureDataset attribute) (oumi.datasets.sft.UltrachatH4Dataset attribute) (oumi.datasets.sft.WildChatDataset attribute) (oumi.datasets.TextSftJsonLinesDataset attribute) (oumi.datasets.TheCauldronDataset attribute) (oumi.datasets.TldrGrpoDataset attribute) (oumi.datasets.Tulu3MixtureDataset attribute) (oumi.datasets.UltrachatH4Dataset attribute) (oumi.datasets.vision_language.COCOCaptionsDataset attribute) (oumi.datasets.vision_language.DocmatixDataset attribute) (oumi.datasets.vision_language.Flickr30kDataset attribute) (oumi.datasets.vision_language.Geometry3kDataset attribute) (oumi.datasets.vision_language.HuggingFaceVisionDataset attribute) (oumi.datasets.vision_language.LlavaInstructMixVsftDataset attribute) (oumi.datasets.vision_language.LmmsLabMultimodalOpenR1Dataset attribute) (oumi.datasets.vision_language.MnistSftDataset attribute) (oumi.datasets.vision_language.OpenbmbRlaifVDataset attribute) (oumi.datasets.vision_language.PixmoAskModelAnythingDataset attribute) (oumi.datasets.vision_language.PixmoCapDataset attribute) (oumi.datasets.vision_language.PixmoCapQADataset attribute) (oumi.datasets.vision_language.TheCauldronDataset attribute) (oumi.datasets.vision_language.VisionDpoJsonlinesDataset attribute) (oumi.datasets.vision_language.VLJsonlinesDataset attribute) (oumi.datasets.vision_language.Vqav2SmallDataset attribute) (oumi.datasets.VisionDpoJsonlinesDataset attribute) (oumi.datasets.VLJsonlinesDataset attribute) (oumi.datasets.Vqav2SmallDataset attribute) (oumi.datasets.WildChatDataset attribute) try_prepare_trl_grpo_dataset() (in module oumi.utils.grpo_utils) try_prepare_trl_grpo_example() (in module oumi.utils.grpo_utils) try_resume_from_last_checkpoint (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) try_str_to_bool() (in module oumi.utils.str_utils) Tulu3MixtureDataset (class in oumi.datasets) (class in oumi.datasets.sft) tunable_peft_params (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) tunable_training_params (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) tune() (in module oumi) tuner_sampler (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) tuner_type (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) TunerType (class in oumi.core.configs) (class in oumi.core.configs.params.tuning_params) tuning (oumi.core.configs.TuningConfig attribute) tuning_study_name (oumi.core.configs.params.tuning_params.TuningParams attribute) (oumi.core.configs.TuningParams attribute) TuningConfig (class in oumi.core.configs) TuningParams (class in oumi.core.configs) (class in oumi.core.configs.params.tuning_params) Type (class in oumi.core.types) type (oumi.core.configs.params.synthesis_params.TransformationStrategy attribute) (oumi.core.configs.TransformationStrategy attribute) (oumi.core.types.ContentItem attribute) U UltrachatH4Dataset (class in oumi.datasets) (class in oumi.datasets.sft) unbatch() (in module oumi.utils.batching) UNIFORM (oumi.core.configs.params.tuning_params.ParamType attribute) up() (in module oumi.launcher) (oumi.launcher.Launcher method) up_cluster() (oumi.core.launcher.BaseCloud method) (oumi.launcher.clouds.FrontierCloud method) (oumi.launcher.clouds.LocalCloud method) (oumi.launcher.clouds.PerlmutterCloud method) (oumi.launcher.clouds.PolarisCloud method) (oumi.launcher.clouds.SkyCloud method) (oumi.launcher.clouds.SlurmCloud method) update_logger_level() (in module oumi.utils.logging) upload_to_huggingface() (oumi.utils.verl_model_merger.BaseModelMerger method) use_adaptive_concurrency (oumi.core.configs.params.remote_params.RemoteParams attribute) (oumi.core.configs.RemoteParams attribute) use_bnb_nested_quant (oumi.core.configs.params.peft_params.PeftParams attribute) (oumi.core.configs.PeftParams attribute) use_cache (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) use_deterministic (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) use_orig_params (oumi.core.configs.FSDPParams attribute) (oumi.core.configs.params.fsdp_params.FSDPParams attribute) use_peft (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) use_sampling (oumi.core.configs.GenerationParams attribute) (oumi.core.configs.params.generation_params.GenerationParams attribute) use_spot (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute) use_torchdata (oumi.core.configs.DatasetSplitParams attribute) (oumi.core.configs.params.data_params.DatasetSplitParams attribute) use_vllm (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) used_memory_mb (oumi.utils.device_utils.NVidiaGpuRuntimeInfo attribute) user (oumi.core.configs.JobConfig attribute) USER (oumi.core.types.Role attribute) user (oumi.launcher.clusters.slurm_cluster.SlurmCluster.ConnectionInfo attribute) (oumi.launcher.JobConfig attribute) V validate_config() (oumi.quantize.BaseQuantization method) validate_dataset() (oumi.judges.BaseJudge method) validate_requirements() (oumi.quantize.BaseQuantization method) validation (oumi.core.configs.DataParams attribute) VALIDATION (oumi.core.configs.DatasetSplit attribute) validation (oumi.core.configs.params.data_params.DataParams attribute) VALIDATION (oumi.core.configs.params.data_params.DatasetSplit attribute) variable_shape (oumi.core.configs.internal.internal_model_config.InternalFeatureSpec attribute) variable_shape_image_features (oumi.core.configs.internal.internal_model_config.InternalVisualModelConfig attribute) verbose (oumi.core.configs.QuantizationConfig attribute) verify_param_spec() (oumi.core.configs.params.tuning_params.ParamType static method) verl_config_overrides (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) VERL_GRPO (oumi.core.configs.params.training_params.TrainerType attribute) (oumi.core.configs.TrainerType attribute) VerlGrpoTrainer (class in oumi.core.trainers) version (oumi.core.configs.AlpacaEvalTaskParams attribute) (oumi.core.configs.params.evaluation_params.AlpacaEvalTaskParams attribute) VisionDpoJsonlinesDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) VisionLanguageCollatorWithPadding (class in oumi.core.collators.vision_language_collator_with_padding) VisionLanguageConversationFeatureGenerator (class in oumi.core.feature_generators) VisionLanguageDpoDataset (class in oumi.core.datasets) VisionLanguageSftCollator (class in oumi.core.collators.vision_language_sft_collator) VisionLanguageSftDataset (class in oumi.core.datasets) visual_config (oumi.core.configs.internal.internal_model_config.InternalModelConfig attribute) VLJsonlinesDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) VLLM (oumi.core.configs.InferenceEngineType attribute) vllm_gpu_memory_utilization (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) vllm_mode (oumi.core.configs.GrpoParams attribute) (oumi.core.configs.params.grpo_params.GrpoParams attribute) VLLMInferenceEngine (class in oumi.inference) Vqav2SmallDataset (class in oumi.datasets) (class in oumi.datasets.vision_language) W wait (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) wall_clock_breakdown (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) warmup (oumi.core.configs.params.profiler_params.ProfilerScheduleParams attribute) warmup_ratio (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) warmup_steps (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) weight_decay (oumi.core.configs.params.training_params.TrainingParams attribute) (oumi.core.configs.TrainingParams attribute) which_clouds() (in module oumi.launcher) (oumi.launcher.Launcher method) WikipediaDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) WikiTextDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) WildChatDataset (class in oumi.datasets) (class in oumi.datasets.sft) with_flops (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) with_modules (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) with_stack (oumi.core.configs.params.profiler_params.ProfilerParams attribute) (oumi.core.configs.ProfilerParams attribute) working_dir (oumi.core.configs.JobConfig attribute) (oumi.launcher.JobConfig attribute) X XML (oumi.core.configs.JudgeResponseFormat attribute) (oumi.core.configs.params.judge_params.JudgeResponseFormat attribute) Y YouTubeCommonsDataset (class in oumi.datasets) (class in oumi.datasets.pretraining) Z ZERO_0 (oumi.core.configs.params.deepspeed_params.ZeRORuntimeStage attribute) ZERO_1 (oumi.core.configs.params.deepspeed_params.ZeRORuntimeStage attribute) ZERO_2 (oumi.core.configs.params.deepspeed_params.ZeRORuntimeStage attribute) ZERO_3 (oumi.core.configs.params.deepspeed_params.ZeRORuntimeStage attribute) zero_allow_untested_optimizer (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) zero_force_ds_cpu_optimizer (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) zero_stage (oumi.core.configs.params.deepspeed_params.DeepSpeedParams attribute) ZeRORuntimeStage (class in oumi.core.configs.params.deepspeed_params) zone (oumi.core.configs.JobResources attribute) (oumi.launcher.JobResources attribute)
