oumi.core#
Core package for the Oumi (Open Universal Machine Intelligence) library.
This package contains fundamental components and abstractions used throughout the Oumi framework, including configurations, datasets, models, and more.
Subpackages#
- oumi.core.analyze
- oumi.core.callbacks
- oumi.core.collators
- Submodules
- 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
AlpacaEvalTaskParamsAnalyzeConfigAsyncEvaluationConfigAutoWrapPolicyBackwardPrefetchBaseConfigDataParamsDatasetParamsDatasetSplitDatasetSplitParamsEvaluationBackendEvaluationConfigEvaluationTaskParamsFSDPParamsGenerationParamsGrpoParamsGuidedDecodingParamsInferenceConfigInferenceEngineTypeJobConfigJobResourcesJudgeConfigJudgeOutputTypeJudgeResponseFormatLMHarnessTaskParamsLoraWeightInitializationMixedPrecisionDtypeMixtureStrategyModelParamsPeftParamsPeftSaveModeProfilerParamsQuantizationConfigRemoteParamsSampleAnalyzerParamsSchedulerTypeShardingStrategyStateDictTypeStorageMountSynthesisConfigTelemetryParamsTrainerTypeTrainingConfigTrainingParams- Subpackages
- oumi.core.datasets
- oumi.core.evaluation
- oumi.core.feature_generators
- oumi.core.inference
- oumi.core.launcher
- oumi.core.models
- oumi.core.processors
- oumi.core.registry
- oumi.core.synthesis
- Submodules
- 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
- oumi.core.trainers
- oumi.core.types