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.callbacks
- oumi.core.collators
- oumi.core.configs
AlpacaEvalTaskParamsAsyncEvaluationConfigAutoWrapPolicyBackwardPrefetchBaseConfigCustomEvaluationParamsDataParamsDatasetParamsDatasetSplitDatasetSplitParamsEvaluationConfigEvaluationPlatformEvaluationTaskParamsFSDPParamsGenerationParamsGuidedDecodingParamsInferenceConfigInferenceEngineTypeJobConfigJobResourcesJudgeAttributeJudgeAttributeValueTypeJudgeConfigLMHarnessTaskParamsLoraWeightInitializationMixedPrecisionDtypeMixtureStrategyModelParamsPeftParamsPeftSaveModeProfilerParamsRemoteParamsSchedulerTypeShardingStrategyStateDictTypeStorageMountTelemetryParamsTrainerTypeTrainingConfigTrainingParams- Subpackages
- oumi.core.datasets
- oumi.core.inference
- oumi.core.launcher
- oumi.core.models
- oumi.core.processors
- oumi.core.registry
- oumi.core.tokenizers
- oumi.core.trainers
- oumi.core.types