Index

_

__call__() (medlatents.sampling.CFGDiffuser method) (medlatents.sampling.CFGFlowMatcher method) (medlatents.sampling.CompiledSampler method) (medlatents.sampling.GuidedSampler method) __class_getitem__() (medlatents.data.ContinuousLatentDataset class method) (medlatents.data.PairedDataset class method) (medlatents.data.PreTokenizedDataset class method) (medlatents.data.TokenizedImageDataset class method) (medlatents.post_training.PreferenceDataset class method) __init__() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiDraftSpeculativeDecoder method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.autoregressive.SpeculativeDecoder method) (medlatents.bayesian_flow.BaseBFNSolver method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.DPMSolver2 method) (medlatents.bayesian_flow.DPMSolver3 method) (medlatents.bayesian_flow.EulerSolver method) (medlatents.bayesian_flow.ExponentialIntegrator method) (medlatents.bayesian_flow.HeunSolver method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.bayesian_flow.ScoreGuidedSampler method) (medlatents.bayesian_flow.StochasticHeun method) (medlatents.conditioning.ConditionalBatchConfig method) (medlatents.conditioning.ConditioningBundle method) (medlatents.conditioning.ConditioningConfig method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.configs.ModelSize method) (medlatents.data.ContinuousLatentDataset method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.data.PairedDataset method) (medlatents.data.PreTokenizedDataset method) (medlatents.data.TokenizedImageDataset method) (medlatents.diffusion.ContinuousGaussianDiffusion method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.D3PM method) (medlatents.diffusion.D3PMKLASS method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.evaluation.FIDCalculator method) (medlatents.evaluation.InceptionPool3FeatureExtractor method) (medlatents.evaluation.PrecisionRecallCalculator method) (medlatents.flow_matching.AdaptiveStepSampler method) (medlatents.flow_matching.DiscreteFlowTrainer method) (medlatents.flow_matching.MaskedSourceDistribution method) (medlatents.flow_matching.MixtureDiscreteEulerSolver method) (medlatents.flow_matching.MixtureDiscreteProbPath method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.PolynomialConvexScheduler method) (medlatents.flow_matching.RectifiedFlow method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.UniformSourceDistribution method) (medlatents.flow_matching.WrappedModel method) (medlatents.generation.ContinuousLatentGenerator method) (medlatents.generation.DiscreteLatentGenerator method) (medlatents.inference.kv_cache.QuantizedKVCache method) (medlatents.inference.ModelPruner method) (medlatents.inference.ModelQuantizer method) (medlatents.inference.PagedKVCache method) (medlatents.inference.PruningConfig method) (medlatents.inference.QuantizationConfig method) (medlatents.inference.QuantizedKVCache method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTConfig method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.SPRINTScheduler method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.distillation.ConsistencyTrainer method) (medlatents.post_training.distillation.ReflowPairGenerator method) (medlatents.post_training.distillation.ReflowTrainer method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.DPOOutput method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.PostTrainingConfig method) (medlatents.post_training.preference.AIPreferenceScorer method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.PreferenceDataset method) (medlatents.post_training.PreferencePair method) (medlatents.post_training.RankedSamples method) (medlatents.post_training.RewardModel method) (medlatents.post_training.RewardModelTrainer method) (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) (medlatents.post_training.rl.Trajectory method) (medlatents.post_training.rl.TrajectoryBuffer method) (medlatents.post_training.self_play.RFTTrainer method) (medlatents.post_training.self_play.SPINTrainer method) (medlatents.post_training.SPOStepOutput method) (medlatents.post_training.SPOTrainer method) (medlatents.post_training.StepPreferenceModel method) (medlatents.post_training.StepwisePreference method) (medlatents.sampling.AncestralSamplingScheduler method) (medlatents.sampling.BayesianFlowSampler method) (medlatents.sampling.CFGDiffuser method) (medlatents.sampling.CFGFlowMatcher method) (medlatents.sampling.CompiledSampler method) (medlatents.sampling.DDIMScheduler method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.DPMSolverScheduler method) (medlatents.sampling.EulerDiscreteScheduler method) (medlatents.sampling.GuidedSampler method) (medlatents.sampling.KLASSGenerator method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.MaskGITScheduler method) (medlatents.sampling.RCRGenerator method) (medlatents.sampling.ReinMax method) (medlatents.sampling.RunningConfidenceRemasker method) (medlatents.sampling.RunningConfidenceScheduler method) (medlatents.sampling.TemperatureScheduler method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.ConditionalTrainingConfig method) (medlatents.training.ContinuousLatentTrainer method) (medlatents.training.CurriculumScheduler method) (medlatents.training.DataLoaderConfig method) (medlatents.training.DiscreteLatentTrainer method) (medlatents.training.DistributedConfig method) (medlatents.training.MaskingRatioCurriculum method) (medlatents.training.MaskingSchedule method) (medlatents.training.MemoryProfiler method) (medlatents.training.NoiseLevelCurriculum method) (medlatents.training.NoiseSchedule method) (medlatents.training.REPAEConfig method) (medlatents.training.REPAETrainer method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) (medlatents.training.SequenceLengthCurriculum method) (medlatents.training.TrainingProfiler method)

__post_init__() (medlatents.post_training.PostTrainingConfig method) (medlatents.post_training.PreferencePair method) (medlatents.post_training.RankedSamples method)

A

accuracy (medlatents.post_training.DPOOutput attribute) (medlatents.post_training.SPOStepOutput attribute) actions (medlatents.post_training.rl.Trajectory attribute) adaptive_masking() (in module medlatents.sampling) (in module medlatents.sampling.maskgit) AdaptiveStepSampler (class in medlatents.flow_matching) add() (medlatents.post_training.rl.TrajectoryBuffer method) add_module() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) add_noise() (medlatents.training.NoiseSchedule method) add_pair() (medlatents.post_training.PreferenceDataset method) add_ranked() (medlatents.post_training.PreferenceDataset method) advantages (medlatents.post_training.rl.Trajectory attribute) AIPreferenceScorer (class in medlatents.post_training.preference)

alignment_layers (medlatents.training.REPAEConfig attribute) allocate_sequence() (medlatents.inference.PagedKVCache method) alpha_schedule() (medlatents.diffusion.ContinuousTimeMDLM method) AncestralSamplingScheduler (class in medlatents.sampling) anisotropic_super_resolution() (in module medlatents.inference) append() (medlatents.inference.PagedKVCache method) apply() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) apply_cfg_dropout() (medlatents.conditioning.ConditioningBundle method) apply_rotary_emb() (in module medlatents.networks) apply_token_mask() (in module medlatents.inference) Attention (class in medlatents.networks) AttentionWithValueResidual (class in medlatents.networks) AutoregressiveDPOTrainer (class in medlatents.post_training) AutoregressiveTransformer (class in medlatents.autoregressive)

B

backend (medlatents.training.DistributedConfig attribute) BaseBFNSolver (class in medlatents.bayesian_flow) BaseDPOTrainer (class in medlatents.post_training) BaseRLTrainer (class in medlatents.post_training.rl) batch_size (medlatents.conditioning.ConditioningBundle property) (medlatents.post_training.PostTrainingConfig attribute) (medlatents.training.DataLoaderConfig attribute) BayesianFlowSampler (class in medlatents.sampling) BayesianFlowTransformer (class in medlatents.bayesian_flow) beta (medlatents.post_training.PostTrainingConfig attribute) bfloat16() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

block_conditioning (medlatents.conditioning.ConditioningConfig attribute) bootstrap_fid_noise_floor() (in module medlatents.evaluation) bootstrap_fid_real_vs_gen() (in module medlatents.evaluation) buffer (medlatents.post_training.rl.TrajectoryBuffer attribute) buffers() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) build_grayscale_resnet18() (in module medlatents.evaluation)

C

calculate_fid() (in module medlatents.evaluation) calculate_nll() (in module medlatents.evaluation) calculate_precision_recall() (in module medlatents.evaluation) calculate_psnr() (in module medlatents.evaluation) calculate_ssim() (in module medlatents.evaluation) calibration_batches (medlatents.inference.QuantizationConfig attribute) call_super_init (medlatents.autoregressive.AutoregressiveTransformer attribute) (medlatents.autoregressive.MedusaHead attribute) (medlatents.autoregressive.MedusaModel attribute) (medlatents.autoregressive.MedusaTrainer attribute) (medlatents.autoregressive.MultiTokenPredictionLoss attribute) (medlatents.autoregressive.speculative.medusa.MedusaModel attribute) (medlatents.bayesian_flow.BayesianFlowTransformer attribute) (medlatents.bayesian_flow.ResidualLossWrapper attribute) (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenEncoder attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) (medlatents.conditioning.SliceWiseEncoder attribute) (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) (medlatents.diffusion.ContinuousTimeMDLM attribute) (medlatents.diffusion.MDLMLoss attribute) (medlatents.diffusion.ScoreEntropyLoss attribute) (medlatents.diffusion.SEDDLoss attribute) (medlatents.flow_matching.MixturePathGeneralizedKL attribute) (medlatents.flow_matching.ShortcutFlowMatchingLoss attribute) (medlatents.flow_matching.ShortcutFlowMatchingModel attribute) (medlatents.flow_matching.StepInvariantModel attribute) (medlatents.flow_matching.TimeInvariantVectorField attribute) (medlatents.flow_matching.WrappedModel attribute) (medlatents.maskgit.MaskGIT attribute) (medlatents.networks.Attention attribute) (medlatents.networks.AttentionWithValueResidual attribute) (medlatents.networks.ContinuousDiT attribute) (medlatents.networks.ContinuousSequenceEmbedder attribute) (medlatents.networks.ContinuousTransformer attribute) (medlatents.networks.DiscreteDiT attribute) (medlatents.networks.DiscreteSequenceEmbedder attribute) (medlatents.networks.DiscreteTransformer attribute) (medlatents.networks.DiTAttention attribute) (medlatents.networks.DiTBlock attribute) (medlatents.networks.DiTBlockWithCrossAttention attribute) (medlatents.networks.DiTCrossAttention attribute) (medlatents.networks.FinalLayer attribute) (medlatents.networks.LabelEmbedder attribute) (medlatents.networks.MMDiTBlock attribute) (medlatents.networks.RMSNorm attribute) (medlatents.networks.RMSNormZero attribute) (medlatents.networks.SPRINTDiT attribute) (medlatents.networks.TimestepEmbedder attribute) (medlatents.networks.TokenRestorer attribute) (medlatents.networks.TokenSelector attribute) (medlatents.networks.TransformerBlock attribute) (medlatents.post_training.DPOLoss attribute) (medlatents.post_training.preference.DiscriminatorScorer attribute) (medlatents.post_training.preference.StepAwarePreferenceModel attribute) (medlatents.post_training.RewardModel attribute) (medlatents.post_training.StepPreferenceModel attribute) (medlatents.sampling.DecoupledSTGumbelSoftmax attribute) (medlatents.sampling.maskgit.TokenCritic attribute) (medlatents.sampling.ReinMax attribute) (medlatents.training.ConditionalLossWrapper attribute) (medlatents.training.REPALoss attribute) (medlatents.training.REPAProjection attribute) cfg_zero_guidance() (in module medlatents.sampling) cfg_zero_star_guidance() (in module medlatents.sampling) CFGDiffuser (class in medlatents.sampling) CFGFlowMatcher (class in medlatents.sampling) children() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) chosen (medlatents.post_training.PreferencePair attribute) chosen_rewards (medlatents.post_training.DPOOutput attribute) chosen_score (medlatents.post_training.StepwisePreference attribute) chosen_state (medlatents.post_training.SPOStepOutput attribute) (medlatents.post_training.StepwisePreference attribute) ckpt_wrapper() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteTransformer method) class_dropout_prob (medlatents.conditioning.ConditioningConfig attribute) (medlatents.training.ConditionalTrainingConfig attribute) class_key (medlatents.conditioning.ConditionalBatchConfig attribute) class_labels (medlatents.conditioning.ConditioningBundle attribute) classifier_fid() (in module medlatents.evaluation) clear() (medlatents.inference.kv_cache.QuantizedKVCache method) (medlatents.inference.QuantizedKVCache method) (medlatents.post_training.rl.TrajectoryBuffer method) clip_range (medlatents.post_training.PostTrainingConfig attribute) collate_conditional_batch() (in module medlatents.conditioning) collate_preference_pairs() (in module medlatents.post_training) collate_variable_length() (in module medlatents.training) compare_with_interpolation() (in module medlatents.inference) compile() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

compile_function() (in module medlatents.sampling) compile_model() (in module medlatents.sampling) compile_sampling() (medlatents.diffusion.D3PM method) CompiledSampler (class in medlatents.sampling) compute() (medlatents.evaluation.FIDCalculator method) (medlatents.evaluation.PrecisionRecallCalculator method) compute_conditional_loss() (in module medlatents.training) compute_entropy() (in module medlatents.bayesian_flow) (in module medlatents.flow_matching) compute_kl_penalty() (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) compute_log_snr() (in module medlatents.diffusion) (in module medlatents.flow_matching) compute_logprobs() (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) compute_loss() (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.diffusion.ContinuousGaussianDiffusion method) (medlatents.diffusion.D3PM method) (medlatents.flow_matching.DiscreteFlowTrainer method) (medlatents.flow_matching.RectifiedFlow method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.training.ContinuousLatentTrainer method) (medlatents.training.DiscreteLatentTrainer method) (medlatents.training.REPAETrainer method) compute_ot_coupling() (in module medlatents.flow_matching) compute_policy_loss() (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) compute_preference_loss() (medlatents.post_training.RewardModel method) compute_score_function() (medlatents.bayesian_flow.ScoreGuidedSampler method) compute_snr() (in module medlatents.diffusion) compute_transition_matrices() (in module medlatents.diffusion) compute_transition_rates() (medlatents.diffusion.SEDDLoss method) ConditionalBatchConfig (class in medlatents.conditioning) ConditionalLossWrapper (class in medlatents.training) ConditionalTrainingConfig (class in medlatents.training) conditioning_config (medlatents.training.ConditionalTrainingConfig attribute) ConditioningBundle (class in medlatents.conditioning) ConditioningConfig (class in medlatents.conditioning) confidence_based_schedule() (in module medlatents.sampling) confidence_mask() (in module medlatents.inference) config (medlatents.data.ContinuousTokenizer attribute) config_name (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) ConsistencyTrainer (class in medlatents.post_training.distillation) constant_guidance() (in module medlatents.sampling) continuous_time_loss() (medlatents.bayesian_flow.BayesianFlowTransformer method) ContinuousDiT (class in medlatents.networks) ContinuousGaussianDiffusion (class in medlatents.diffusion) ContinuousLatentDataset (class in medlatents.data) ContinuousLatentGenerator (class in medlatents.generation) ContinuousLatentTrainer (class in medlatents.training) ContinuousSequenceEmbedder (class in medlatents.networks) ContinuousTimeMDLM (class in medlatents.diffusion) ContinuousTokenizer (class in medlatents.data) ContinuousTransformer (class in medlatents.networks) cosine_decay_guidance() (in module medlatents.sampling) cosine_distance_matrix() (in module medlatents.evaluation) cosine_guidance() (in module medlatents.sampling) cosine_schedule() (in module medlatents.bayesian_flow) cpu() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) create_curriculum_from_config() (in module medlatents.training) create_encoder() (in module medlatents.conditioning) create_fused_adamw() (in module medlatents.training) create_mask() (medlatents.training.MaskingRatioCurriculum method) create_model_variants() (in module medlatents.configs) create_optimized_dataloader() (in module medlatents.training) create_optimizer_groups() (medlatents.training.REPAETrainer method) create_pairs() (medlatents.post_training.preference.AIPreferenceScorer method) (medlatents.post_training.preference.DiscriminatorScorer method) create_reward_model() (in module medlatents.post_training) create_spatial_mask() (in module medlatents.inference) create_time_grid() (in module medlatents.flow_matching) cuda() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) CurriculumScheduler (class in medlatents.training)

D

D3PM (class in medlatents.diffusion) D3PMDPOTrainer (class in medlatents.post_training) D3PMKLASS (class in medlatents.diffusion) DataLoaderConfig (class in medlatents.training) DDIMScheduler (class in medlatents.sampling) DDPO_STANDARD (in module medlatents.post_training) DDPODiscreteFlowTrainer (class in medlatents.post_training.rl) DDPOTrainer (class in medlatents.post_training.rl) decode() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) decode_batch() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) decode_large_volume() (in module medlatents.inference) decode_latents() (medlatents.generation.ContinuousLatentGenerator method) DecoupledSTGumbelSoftmax (class in medlatents.sampling) dense_layers (medlatents.networks.SPRINTConfig attribute) depth (medlatents.configs.ModelSize attribute) detect_anomalies() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteTransformer method) detokenize() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) device (medlatents.autoregressive.AutoregressiveTransformer property) (medlatents.bayesian_flow.BayesianFlowTransformer property) (medlatents.conditioning.ConditioningBundle property) (medlatents.maskgit.MaskGIT property) (medlatents.networks.ContinuousDiT property) (medlatents.networks.ContinuousTransformer property) (medlatents.networks.DiscreteDiT property) (medlatents.networks.DiscreteTransformer property) diffusion_weight (medlatents.training.REPAEConfig attribute) DiffusionDPOTrainer (class in medlatents.post_training) discrete_time_loss() (medlatents.bayesian_flow.BayesianFlowTransformer method) DiscreteDiT (class in medlatents.networks) DiscreteDPOTrainer (class in medlatents.post_training) DiscreteFlowDPOTrainer (class in medlatents.post_training) DiscreteFlowTrainer (class in medlatents.flow_matching) DiscreteLatentGenerator (class in medlatents.generation) DiscreteLatentTrainer (class in medlatents.training) DiscreteSequenceEmbedder (class in medlatents.networks) DiscreteTokenizer (class in medlatents.data) DiscreteTransformer (class in medlatents.networks) discretization_steps (medlatents.post_training.PostTrainingConfig attribute) DiscriminatorScorer (class in medlatents.post_training.preference) distance_penalty (medlatents.post_training.PostTrainingConfig attribute) DistributedConfig (class in medlatents.training) DiTAttention (class in medlatents.networks) DiTBlock (class in medlatents.networks) DiTBlockWithCrossAttention (class in medlatents.networks) DiTCrossAttention (class in medlatents.networks) double() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

DPMSolver2 (class in medlatents.bayesian_flow) DPMSolver3 (class in medlatents.bayesian_flow) DPMSolverScheduler (class in medlatents.sampling) DPO_LARGE (in module medlatents.post_training) DPO_SMALL (in module medlatents.post_training) DPO_STANDARD (in module medlatents.post_training) DPOLoss (class in medlatents.post_training) DPOOutput (class in medlatents.post_training) drop_last (medlatents.training.DataLoaderConfig attribute) drop_ratio (medlatents.networks.SPRINTConfig attribute) drop_schedule (medlatents.networks.SPRINTConfig attribute) dtype (medlatents.inference.QuantizationConfig attribute) dump_patches (medlatents.autoregressive.AutoregressiveTransformer attribute) (medlatents.autoregressive.MedusaHead attribute) (medlatents.autoregressive.MedusaModel attribute) (medlatents.autoregressive.MedusaTrainer attribute) (medlatents.autoregressive.MultiTokenPredictionLoss attribute) (medlatents.autoregressive.speculative.medusa.MedusaModel attribute) (medlatents.bayesian_flow.BayesianFlowTransformer attribute) (medlatents.bayesian_flow.ResidualLossWrapper attribute) (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenEncoder attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) (medlatents.conditioning.SliceWiseEncoder attribute) (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) (medlatents.diffusion.ContinuousTimeMDLM attribute) (medlatents.diffusion.MDLMLoss attribute) (medlatents.diffusion.ScoreEntropyLoss attribute) (medlatents.diffusion.SEDDLoss attribute) (medlatents.flow_matching.MixturePathGeneralizedKL attribute) (medlatents.flow_matching.ShortcutFlowMatchingLoss attribute) (medlatents.flow_matching.ShortcutFlowMatchingModel attribute) (medlatents.flow_matching.StepInvariantModel attribute) (medlatents.flow_matching.TimeInvariantVectorField attribute) (medlatents.flow_matching.WrappedModel attribute) (medlatents.maskgit.MaskGIT attribute) (medlatents.networks.Attention attribute) (medlatents.networks.AttentionWithValueResidual attribute) (medlatents.networks.ContinuousDiT attribute) (medlatents.networks.ContinuousSequenceEmbedder attribute) (medlatents.networks.ContinuousTransformer attribute) (medlatents.networks.DiscreteDiT attribute) (medlatents.networks.DiscreteSequenceEmbedder attribute) (medlatents.networks.DiscreteTransformer attribute) (medlatents.networks.DiTAttention attribute) (medlatents.networks.DiTBlock attribute) (medlatents.networks.DiTBlockWithCrossAttention attribute) (medlatents.networks.DiTCrossAttention attribute) (medlatents.networks.FinalLayer attribute) (medlatents.networks.LabelEmbedder attribute) (medlatents.networks.MMDiTBlock attribute) (medlatents.networks.RMSNorm attribute) (medlatents.networks.RMSNormZero attribute) (medlatents.networks.SPRINTDiT attribute) (medlatents.networks.TimestepEmbedder attribute) (medlatents.networks.TokenRestorer attribute) (medlatents.networks.TokenSelector attribute) (medlatents.networks.TransformerBlock attribute) (medlatents.post_training.DPOLoss attribute) (medlatents.post_training.preference.DiscriminatorScorer attribute) (medlatents.post_training.preference.StepAwarePreferenceModel attribute) (medlatents.post_training.RewardModel attribute) (medlatents.post_training.StepPreferenceModel attribute) (medlatents.sampling.DecoupledSTGumbelSoftmax attribute) (medlatents.sampling.maskgit.TokenCritic attribute) (medlatents.sampling.ReinMax attribute) (medlatents.training.ConditionalLossWrapper attribute) (medlatents.training.REPALoss attribute) (medlatents.training.REPAProjection attribute) dynamic_guidance() (in module medlatents.sampling)

E

ema_decay (medlatents.post_training.PostTrainingConfig attribute) embed_dim (medlatents.conditioning.FrozenDINOv2 property) (medlatents.conditioning.FrozenEncoder property) (medlatents.conditioning.FrozenMedSigLIP property) (medlatents.conditioning.FrozenNeuroVFM property) (medlatents.conditioning.FrozenSigLIP property) (medlatents.conditioning.SliceWiseEncoder property) enable_cuda_optimizations() (in module medlatents.training) enabled (medlatents.training.TrainingProfiler attribute) encode() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) encode_batch() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) encode_large_volume() (in module medlatents.inference) encode_with_entropy() (in module medlatents.bayesian_flow) enforce_zero_terminal_snr() (in module medlatents.diffusion) estimate_likelihood() (in module medlatents.flow_matching) estimate_model_size() (in module medlatents.inference) EulerDiscreteScheduler (class in medlatents.sampling) EulerSolver (class in medlatents.bayesian_flow) eval() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

eval_every (medlatents.post_training.PostTrainingConfig attribute) evaluate_auc() (in module medlatents.evaluation) exponential_schedule() (in module medlatents.bayesian_flow) ExponentialIntegrator (class in medlatents.bayesian_flow) export_to_onnx() (in module medlatents.inference) extra_repr() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) extract() (medlatents.evaluation.InceptionPool3FeatureExtractor method)

F

fid_from_features() (in module medlatents.evaluation) FIDCalculator (class in medlatents.evaluation) filter() (medlatents.post_training.PreferenceDataset method) final_sample (medlatents.post_training.rl.Trajectory attribute) FinalLayer (class in medlatents.networks) float() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) FlowDPOTrainer (class in medlatents.post_training) forward() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.CompiledSampler method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

forward_features() (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) forward_with_cache() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.networks.Attention method) (medlatents.networks.TransformerBlock method) forward_with_cfg() (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.DiscreteDiT method) forward_with_pdg() (medlatents.networks.SPRINTDiT method) free_sequence() (medlatents.inference.PagedKVCache method) freeze() (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) freeze_vae_decoder (medlatents.training.REPAEConfig attribute) freeze_vae_encoder (medlatents.training.REPAEConfig attribute) from_batch() (medlatents.conditioning.ConditioningBundle class method) from_checkpoints() (medlatents.generation.DiscreteLatentGenerator class method) from_dict() (medlatents.post_training.PostTrainingConfig class method) from_discriminator() (medlatents.post_training.preference.AIPreferenceScorer class method) from_env() (medlatents.training.DistributedConfig class method) from_generations() (medlatents.post_training.PreferenceDataset class method) from_json() (medlatents.post_training.PreferenceDataset class method) from_preset() (medlatents.post_training.PostTrainingConfig class method) from_pretrained() (medlatents.data.ContinuousTokenizer class method) (medlatents.data.DiscreteTokenizer class method) from_pretrained_encoder_decoder() (medlatents.data.ContinuousTokenizer class method) (medlatents.data.DiscreteTokenizer class method) from_reward_model() (medlatents.post_training.preference.AIPreferenceScorer class method) FrozenDINOv2 (class in medlatents.conditioning) FrozenEncoder (class in medlatents.conditioning) FrozenMedSigLIP (class in medlatents.conditioning) FrozenNeuroVFM (class in medlatents.conditioning) FrozenSigLIP (class in medlatents.conditioning) fsdp_sharding_strategy (medlatents.post_training.PostTrainingConfig attribute) (medlatents.training.DistributedConfig attribute) fusion_method (medlatents.conditioning.ConditioningConfig attribute) (medlatents.networks.SPRINTConfig attribute)

G

gae_lambda (medlatents.post_training.PostTrainingConfig attribute) GARDOTrainer (class in medlatents.post_training.rl) generate() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MultiDraftSpeculativeDecoder method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.autoregressive.SpeculativeDecoder method) (medlatents.generation.ContinuousLatentGenerator method) (medlatents.generation.DiscreteLatentGenerator method) (medlatents.maskgit.MaskGIT method) (medlatents.sampling.CompiledSampler method) (medlatents.sampling.KLASSGenerator method) (medlatents.sampling.RCRGenerator method) generate_counterfactual() (in module medlatents.flow_matching) generate_pairs() (medlatents.post_training.distillation.ReflowPairGenerator method) generate_pairs_from_generator() (medlatents.post_training.preference.AIPreferenceScorer method) generate_samples() (in module medlatents.flow_matching) generate_speculative() (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) generate_trajectories() (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) generate_volumes() (medlatents.generation.ContinuousLatentGenerator method) (medlatents.generation.DiscreteLatentGenerator method) generate_with_cfg() (in module medlatents.sampling) get() (medlatents.inference.kv_cache.QuantizedKVCache method) (medlatents.inference.PagedKVCache method) (medlatents.inference.QuantizedKVCache method) get_absorbing_transition_mat() (in module medlatents.diffusion) get_accuracy() (medlatents.bayesian_flow.BayesianFlowTransformer method) get_beta_schedule() (in module medlatents.diffusion) get_beta_schedule_with_zero_snr() (in module medlatents.diffusion) get_bfn_schedule() (in module medlatents.bayesian_flow) get_bfn_solver() (in module medlatents.bayesian_flow) get_buffer() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) get_codebook_size() (medlatents.data.DiscreteTokenizer method) get_combined_embedding() (medlatents.conditioning.ConditioningBundle method) get_confidence() (in module medlatents.sampling) get_cosine_schedule_with_warmup() (in module medlatents.training) get_discretized_gaussian_transition_mat() (in module medlatents.diffusion) get_dit_parameters() (medlatents.training.REPAETrainer method) get_drop_ratio() (medlatents.networks.SPRINTScheduler method) get_extra_state() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) get_guidance_schedule() (in module medlatents.sampling) get_latent_dataloaders() (in module medlatents.data)

get_latent_shape() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) get_length() (medlatents.training.SequenceLengthCurriculum method) get_loss_function() (in module medlatents.flow_matching) get_mask() (medlatents.sampling.MaskGITScheduler method) (medlatents.sampling.RunningConfidenceScheduler method) get_mask_ratio() (in module medlatents.training) (medlatents.sampling.MaskGITScheduler method) (medlatents.sampling.RunningConfidenceScheduler method) (medlatents.training.MaskingRatioCurriculum method) (medlatents.training.MaskingSchedule method) get_noise_level() (medlatents.training.NoiseSchedule method) get_null_bundle() (medlatents.conditioning.ConditioningBundle method) get_num_steps() (medlatents.flow_matching.AdaptiveStepSampler method) get_optimal_dtype() (in module medlatents.training) get_parameter() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) get_path() (in module medlatents.flow_matching) get_percentages() (medlatents.training.TrainingProfiler method) get_prior_params() (medlatents.bayesian_flow.BayesianFlowTransformer method) get_quant_dtype() (medlatents.inference.QuantizationConfig method) get_random_num_steps() (medlatents.flow_matching.StepInvariantModel method) get_source_distribution() (in module medlatents.flow_matching) get_stats() (medlatents.training.MemoryProfiler method) (medlatents.training.TrainingProfiler method) get_submodule() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) get_temperature() (medlatents.sampling.MaskGITScheduler method) (medlatents.sampling.RunningConfidenceScheduler method) (medlatents.sampling.TemperatureScheduler method) get_time_weights() (medlatents.diffusion.MDLMLoss method) get_timestep_sampler() (in module medlatents.flow_matching) get_tokenized_dataloaders() (in module medlatents.data) get_tokenizer_info() (in module medlatents.data) get_uniform_transition_mat() (in module medlatents.diffusion) get_vae_parameters() (medlatents.training.REPAETrainer method) get_value() (medlatents.training.CurriculumScheduler method) (medlatents.training.MaskingRatioCurriculum method) (medlatents.training.NoiseLevelCurriculum method) (medlatents.training.SequenceLengthCurriculum method) get_wandb_metrics() (medlatents.training.TrainingProfiler method) grad_clip (medlatents.post_training.PostTrainingConfig attribute) gradient_accumulation_steps (medlatents.post_training.PostTrainingConfig attribute) gradient_checkpointing (medlatents.training.DistributedConfig attribute) granularity (medlatents.inference.PruningConfig attribute) GRPO_EFFICIENT (in module medlatents.post_training) GRPOTrainer (class in medlatents.post_training.rl) GuidedSampler (class in medlatents.sampling) gumbel_max_sampling() (in module medlatents.sampling)

H

half() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

halton_schedule_1d() (in module medlatents.sampling.maskgit) haste_termination_step (medlatents.training.ConditionalTrainingConfig attribute) haste_termination_type (medlatents.training.ConditionalTrainingConfig attribute) HeunSolver (class in medlatents.bayesian_flow) hidden_size (medlatents.configs.ModelSize attribute) (medlatents.networks.SPRINTDiT attribute) HilbertCurve (class in medlatents.rasterization) huber_c (medlatents.post_training.PostTrainingConfig attribute)

I

importance_scores (medlatents.inference.PruningConfig attribute) importance_type (medlatents.networks.SPRINTConfig attribute) InceptionPool3FeatureExtractor (class in medlatents.evaluation) inference_mode() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) init_weights() (in module medlatents.networks) initialize_weights() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.DiscreteDiT method) inpaint_anomalies() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteTransformer method) inpaint_autoregressive() (in module medlatents.inference) inpaint_bayesian_flow() (in module medlatents.inference) inpaint_diffusion_repaint() (in module medlatents.inference) inpaint_flow_matching() (in module medlatents.inference) inpaint_maskgit() (in module medlatents.inference) inpaint_volume() (in module medlatents.inference) inpaint_volumes() (medlatents.generation.DiscreteLatentGenerator method) interpolate_spatial_upsampling() (in module medlatents.inference) ipu() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

is_compile_available() (in module medlatents.sampling) is_compiled (medlatents.sampling.CompiledSampler property) is_distributed (medlatents.training.DistributedConfig property) is_main_process (medlatents.training.DistributedConfig property) iterative_steps (medlatents.inference.PruningConfig attribute)

K

kappa() (medlatents.flow_matching.PolynomialConvexScheduler method) kappa_inverse() (medlatents.flow_matching.PolynomialConvexScheduler method)

kl_coeff (medlatents.post_training.PostTrainingConfig attribute) kl_weight (medlatents.training.REPAEConfig attribute) KLASSGenerator (class in medlatents.sampling)

L

label_smoothing (medlatents.post_training.PostTrainingConfig attribute) LabelEmbedder (class in medlatents.networks) length (medlatents.inference.kv_cache.QuantizedKVCache property) (medlatents.inference.QuantizedKVCache property) linear_entropy_schedule() (in module medlatents.bayesian_flow) linear_guidance() (in module medlatents.sampling) load() (medlatents.post_training.PreferenceDataset class method) load_checkpoint() (in module medlatents.training) (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.distillation.ConsistencyTrainer method) (medlatents.post_training.distillation.ReflowTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) (medlatents.post_training.self_play.RFTTrainer method) (medlatents.post_training.self_play.SPINTrainer method) (medlatents.post_training.SPOTrainer method) (medlatents.training.ContinuousLatentTrainer method) load_encoder_decoder_weights() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) load_state_dict() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

local_rank (medlatents.training.DistributedConfig attribute) log_every (medlatents.post_training.PostTrainingConfig attribute) log_probs (medlatents.post_training.rl.Trajectory attribute) logdir (medlatents.post_training.PostTrainingConfig attribute) loss (medlatents.post_training.DPOOutput attribute) (medlatents.post_training.SPOStepOutput attribute) loss_type (medlatents.post_training.PostTrainingConfig attribute) lr (medlatents.post_training.PostTrainingConfig attribute)

M

margin (medlatents.post_training.PreferencePair attribute) masked (medlatents.flow_matching.MaskedSourceDistribution property) (medlatents.flow_matching.UniformSourceDistribution property) MaskedSourceDistribution (class in medlatents.flow_matching) MaskGIT (class in medlatents.maskgit) MaskGITDPOTrainer (class in medlatents.post_training) MaskGITScheduler (class in medlatents.sampling) MaskingRatioCurriculum (class in medlatents.training) MaskingSchedule (class in medlatents.training) master_addr (medlatents.training.DistributedConfig attribute) master_port (medlatents.training.DistributedConfig attribute) max_seq_length (medlatents.conditioning.ConditionalBatchConfig attribute) max_steps (medlatents.post_training.PostTrainingConfig attribute) max_text_length (medlatents.conditioning.ConditioningConfig attribute) MDLMLoss (class in medlatents.diffusion) MEAN (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) medlatents.autoregressive module medlatents.bayesian_flow.BFN_models (built-in variable) medlatents.configs.MODEL_CONFIGS (built-in variable) medlatents.configs.MODEL_SIZES (built-in variable) medlatents.configs.MODEL_TYPES (built-in variable) medlatents.maskgit module medlatents.networks module medlatents.sampling module MedusaHead (class in medlatents.autoregressive) MedusaModel (class in medlatents.autoregressive) (class in medlatents.autoregressive.speculative.medusa) MedusaTrainer (class in medlatents.autoregressive) memorization_metrics() (in module medlatents.evaluation) memorization_pairs() (in module medlatents.evaluation) memory_stats() (medlatents.inference.PagedKVCache method) memory_usage() (medlatents.inference.kv_cache.QuantizedKVCache method) (medlatents.inference.QuantizedKVCache method) MemoryProfiler (class in medlatents.training) metadata (medlatents.conditioning.ConditioningBundle attribute) (medlatents.post_training.PreferencePair attribute) (medlatents.post_training.RankedSamples attribute) metadata_dim (medlatents.conditioning.ConditioningConfig attribute) metadata_key (medlatents.conditioning.ConditionalBatchConfig attribute) method (medlatents.inference.PruningConfig attribute) (medlatents.inference.QuantizationConfig attribute) (medlatents.post_training.PostTrainingConfig attribute) min_snr_weighting() (in module medlatents.diffusion) mixed_precision (medlatents.post_training.PostTrainingConfig attribute) (medlatents.training.DistributedConfig attribute) MixtureDiscreteEulerSolver (class in medlatents.flow_matching) MixtureDiscreteProbPath (class in medlatents.flow_matching) MixturePathGeneralizedKL (class in medlatents.flow_matching) MMDiTBlock (class in medlatents.networks) ModelPruner (class in medlatents.inference) ModelQuantizer (class in medlatents.inference) ModelSize (class in medlatents.configs) module medlatents.autoregressive medlatents.maskgit medlatents.networks medlatents.sampling modules() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

modules_to_quantize (medlatents.inference.QuantizationConfig attribute) modules_to_skip (medlatents.inference.QuantizationConfig attribute) mtia() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) MultiDraftSpeculativeDecoder (class in medlatents.autoregressive) MultiTokenPredictionLoss (class in medlatents.autoregressive)

N

named_buffers() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) named_children() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) named_modules() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

named_parameters() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) nearest_neighbor_gallery_pairs() (in module medlatents.evaluation) NoiseLevelCurriculum (class in medlatents.training) NoiseSchedule (class in medlatents.training) normalize_rewards (medlatents.post_training.PostTrainingConfig attribute) null_class (medlatents.conditioning.ConditioningBundle attribute) null_text (medlatents.conditioning.ConditioningBundle attribute) num_candidates (medlatents.post_training.PostTrainingConfig attribute) num_classes (medlatents.conditioning.ConditioningConfig attribute) num_distillation_steps (medlatents.post_training.PostTrainingConfig attribute) num_heads (medlatents.configs.ModelSize attribute) num_iterations (medlatents.sampling.MaskGITScheduler property) (medlatents.sampling.RunningConfidenceScheduler property) num_parameters() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) num_reflow_iterations (medlatents.post_training.PostTrainingConfig attribute) num_samples_per_prompt (medlatents.post_training.PostTrainingConfig attribute) num_segmentation_classes (medlatents.conditioning.ConditioningConfig attribute) num_self_play_iterations (medlatents.post_training.PostTrainingConfig attribute) num_workers (medlatents.training.DataLoaderConfig attribute)

O

ot_flow_sample_path() (in module medlatents.flow_matching)

P

p_sample() (medlatents.diffusion.ContinuousGaussianDiffusion method) (medlatents.diffusion.D3PM method) pad_token_id (medlatents.conditioning.ConditionalBatchConfig attribute) PagedKVCache (class in medlatents.inference) PairedDataset (class in medlatents.data) pairs_per_iteration (medlatents.post_training.PostTrainingConfig attribute) parameters() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

patch_size (medlatents.conditioning.FrozenDINOv2 property) (medlatents.conditioning.FrozenEncoder property) (medlatents.conditioning.FrozenMedSigLIP property) (medlatents.conditioning.FrozenNeuroVFM property) (medlatents.conditioning.FrozenSigLIP property) path_drop_guidance (medlatents.networks.SPRINTConfig attribute) pdg_scale (medlatents.networks.SPRINTConfig attribute) per_channel (medlatents.inference.QuantizationConfig attribute) persistent_workers (medlatents.training.DataLoaderConfig attribute) pin_memory (medlatents.training.DataLoaderConfig attribute) PolynomialConvexScheduler (class in medlatents.flow_matching) PostTrainingConfig (class in medlatents.post_training) PrecisionRecallCalculator (class in medlatents.evaluation) precompute_freqs_cis() (in module medlatents.networks) predict_x0() (medlatents.diffusion.ContinuousGaussianDiffusion method) PreferenceDataset (class in medlatents.post_training) PreferencePair (class in medlatents.post_training) prefetch_factor (medlatents.training.DataLoaderConfig attribute) prepare_batch() (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) prepare_batch_with_conditioning() (in module medlatents.training) preprocess() (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) pretokenize_dataset() (in module medlatents.data) PreTokenizedDataset (class in medlatents.data) print_report() (medlatents.training.MemoryProfiler method) (medlatents.training.TrainingProfiler method) priorities (medlatents.post_training.rl.TrajectoryBuffer attribute) progressive_super_resolution() (in module medlatents.inference) project_target() (medlatents.training.REPAProjection method) projection (medlatents.training.REPAProjection attribute) prompt (medlatents.post_training.PreferencePair attribute) (medlatents.post_training.RankedSamples attribute) (medlatents.post_training.rl.Trajectory attribute) prune() (medlatents.inference.ModelPruner method) prune_attention_heads() (in module medlatents.inference) PruningConfig (class in medlatents.inference) pseudo_huber_loss() (in module medlatents.post_training.distillation) push_to_hub() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method)

Q

q_posterior() (medlatents.diffusion.D3PM method) q_sample() (medlatents.diffusion.ContinuousGaussianDiffusion method) (medlatents.diffusion.D3PM method)

QuantizationConfig (class in medlatents.inference) quantize() (medlatents.inference.ModelQuantizer method) QuantizedKVCache (class in medlatents.inference) (class in medlatents.inference.kv_cache)

R

random_masking() (medlatents.maskgit.MaskGIT method) rank (medlatents.training.DistributedConfig attribute) rank() (medlatents.post_training.preference.AIPreferenceScorer method) rank_samples() (medlatents.post_training.preference.DiscriminatorScorer method) RankedSamples (class in medlatents.post_training) rasterize_2d() (in module medlatents.rasterization) RasterScan (class in medlatents.rasterization) raw_image_key (medlatents.conditioning.ConditionalBatchConfig attribute) RCRGenerator (class in medlatents.sampling) reconstruct() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) reconstruct_large_volume() (in module medlatents.inference) RectifiedFlow (class in medlatents.flow_matching) reference_free (medlatents.post_training.PostTrainingConfig attribute) reflow_iteration() (medlatents.post_training.distillation.ReflowTrainer method) ReflowPairGenerator (class in medlatents.post_training.distillation) ReflowTrainer (class in medlatents.post_training.distillation) register_backward_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_buffer() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_forward_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_forward_pre_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_full_backward_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_full_backward_pre_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_load_state_dict_post_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

register_load_state_dict_pre_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_module() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_parameter() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_schedule() (medlatents.diffusion.ContinuousGaussianDiffusion method) register_state_dict_post_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) register_state_dict_pre_hook() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) ReinMax (class in medlatents.sampling) rejected (medlatents.post_training.PreferencePair attribute) rejected_rewards (medlatents.post_training.DPOOutput attribute) rejected_score (medlatents.post_training.StepwisePreference attribute) rejected_state (medlatents.post_training.StepwisePreference attribute) repa_encoder_name (medlatents.training.ConditionalTrainingConfig attribute) repa_timestep_threshold (medlatents.training.ConditionalTrainingConfig attribute) repa_weight (medlatents.training.ConditionalTrainingConfig attribute) (medlatents.training.REPAEConfig attribute) REPAEConfig (class in medlatents.training) REPAETrainer (class in medlatents.training) REPALoss (class in medlatents.training) REPAProjection (class in medlatents.training) requires_grad_() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) rescale_cfg() (in module medlatents.sampling) reset() (medlatents.diffusion.D3PMKLASS method) (medlatents.evaluation.FIDCalculator method) (medlatents.evaluation.PrecisionRecallCalculator method) (medlatents.sampling.RunningConfidenceRemasker method) (medlatents.sampling.RunningConfidenceScheduler method) (medlatents.training.TrainingProfiler method) ResidualLossWrapper (class in medlatents.bayesian_flow) reslice_volume() (in module medlatents.inference) resume_from_checkpoint() (in module medlatents.training) (medlatents.training.ContinuousLatentTrainer method) returns (medlatents.post_training.rl.Trajectory attribute) reward_margin (medlatents.post_training.DPOOutput attribute) reward_model_config (medlatents.post_training.PostTrainingConfig attribute) reward_model_path (medlatents.post_training.PostTrainingConfig attribute) reward_model_type (medlatents.post_training.PostTrainingConfig attribute) reward_threshold (medlatents.post_training.PostTrainingConfig attribute) RewardModel (class in medlatents.post_training) RewardModelTrainer (class in medlatents.post_training) rewards (medlatents.post_training.rl.Trajectory attribute) RFTTrainer (class in medlatents.post_training.self_play) RMSNorm (class in medlatents.networks) RMSNormZero (class in medlatents.networks) run_name (medlatents.post_training.PostTrainingConfig attribute) RunningConfidenceRemasker (class in medlatents.sampling) RunningConfidenceScheduler (class in medlatents.sampling)

S

sample() (medlatents.bayesian_flow.BaseBFNSolver method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.DPMSolver2 method) (medlatents.bayesian_flow.DPMSolver3 method) (medlatents.bayesian_flow.EulerSolver method) (medlatents.bayesian_flow.ExponentialIntegrator method) (medlatents.bayesian_flow.HeunSolver method) (medlatents.bayesian_flow.ScoreGuidedSampler method) (medlatents.bayesian_flow.StochasticHeun method) (medlatents.diffusion.ContinuousGaussianDiffusion method) (medlatents.diffusion.D3PM method) (medlatents.diffusion.D3PMKLASS method) (medlatents.flow_matching.MaskedSourceDistribution method) (medlatents.flow_matching.MixtureDiscreteEulerSolver method) (medlatents.flow_matching.MixtureDiscreteProbPath method) (medlatents.flow_matching.RectifiedFlow method) (medlatents.flow_matching.SourceDistribution method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.UniformSourceDistribution method) (medlatents.post_training.rl.TrajectoryBuffer method) (medlatents.sampling.BayesianFlowSampler method) sample_autoregressive() (in module medlatents.sampling) sample_base() (medlatents.flow_matching.RectifiedFlow method) sample_bayesian_flow() (in module medlatents.sampling) sample_from_coupling() (in module medlatents.flow_matching) sample_like() (medlatents.flow_matching.MaskedSourceDistribution method) (medlatents.flow_matching.SourceDistribution method) (medlatents.flow_matching.UniformSourceDistribution method) sample_min_p() (in module medlatents.sampling) sample_noise_like() (medlatents.diffusion.ContinuousGaussianDiffusion method) sample_nucleus() (in module medlatents.sampling) sample_path() (medlatents.flow_matching.RectifiedFlow method) sample_random_candidate() (medlatents.post_training.preference.StepAwarePreferenceModel method) sample_receiver_distribution() (medlatents.bayesian_flow.BayesianFlowTransformer method) sample_sender_distribution() (medlatents.bayesian_flow.BayesianFlowTransformer method) sample_shortcut() (medlatents.flow_matching.ShortcutFlowMatchingModel method) sample_timesteps() (medlatents.flow_matching.DiscreteFlowTrainer method) (medlatents.training.NoiseLevelCurriculum method) sample_timesteps_log_snr_importance() (in module medlatents.diffusion) sample_timesteps_logit_normal() (in module medlatents.flow_matching) sample_timesteps_u_shaped() (in module medlatents.flow_matching) sample_timesteps_uniform() (in module medlatents.flow_matching) sample_with_cfg() (in module medlatents.sampling) (medlatents.sampling.BayesianFlowSampler method) sample_with_cfg_diffusion() (in module medlatents.sampling) sample_with_cfg_flow() (in module medlatents.sampling) sample_with_dynamic_cfg() (in module medlatents.sampling) sample_with_early_stopping() (in module medlatents.sampling) (medlatents.sampling.BayesianFlowSampler method) sample_with_schedule() (medlatents.sampling.BayesianFlowSampler method) samples (medlatents.post_training.RankedSamples attribute) save() (medlatents.post_training.PreferenceDataset method) save_checkpoint() (in module medlatents.training) (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.distillation.ConsistencyTrainer method) (medlatents.post_training.distillation.ReflowTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) (medlatents.post_training.self_play.RFTTrainer method) (medlatents.post_training.self_play.SPINTrainer method) (medlatents.post_training.SPOTrainer method) (medlatents.training.ContinuousLatentTrainer method) save_every (medlatents.post_training.PostTrainingConfig attribute) save_pretrained() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) score() (medlatents.post_training.preference.AIPreferenceScorer method) ScoreEntropyLoss (class in medlatents.diffusion) ScoreGuidedSampler (class in medlatents.bayesian_flow) scores (medlatents.post_training.RankedSamples attribute) SCurve (class in medlatents.rasterization) section() (medlatents.training.TrainingProfiler method) SEDDLoss (class in medlatents.diffusion) seed (medlatents.post_training.PostTrainingConfig attribute) segmentation_key (medlatents.conditioning.ConditionalBatchConfig attribute) segmentation_mask (medlatents.conditioning.ConditioningBundle attribute) select_winner_loser() (medlatents.post_training.preference.StepAwarePreferenceModel method) sequence_to_spatial_2d() (medlatents.rasterization.HilbertCurve static method) (medlatents.rasterization.RasterScan static method) (medlatents.rasterization.SCurve static method) (medlatents.rasterization.ZOrderCurve static method) sequence_to_spatial_3d() (medlatents.rasterization.HilbertCurve static method) (medlatents.rasterization.RasterScan static method) (medlatents.rasterization.SCurve static method) (medlatents.rasterization.ZOrderCurve static method) SequenceLengthCurriculum (class in medlatents.training) set_chunk_size() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteTransformer method) set_extra_state() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) set_step() (medlatents.training.CurriculumScheduler method) (medlatents.training.MaskingRatioCurriculum method) (medlatents.training.NoiseLevelCurriculum method) (medlatents.training.SequenceLengthCurriculum method) set_submodule() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

set_timesteps() (medlatents.sampling.AncestralSamplingScheduler method) (medlatents.sampling.DDIMScheduler method) (medlatents.sampling.DPMSolverScheduler method) (medlatents.sampling.EulerDiscreteScheduler method) setup_distributed() (in module medlatents.training) share_memory() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) ShortcutFlowMatchingLoss (class in medlatents.flow_matching) ShortcutFlowMatchingModel (class in medlatents.flow_matching) should_stop() (medlatents.diffusion.D3PMKLASS method) (medlatents.sampling.KLASSGenerator method) shuffle (medlatents.training.DataLoaderConfig attribute) shuffle() (medlatents.post_training.PreferenceDataset method) SliceWiseEncoder (class in medlatents.conditioning) smoothing_alpha (medlatents.inference.QuantizationConfig attribute) SourceDistribution (class in medlatents.flow_matching) sparse_layers (medlatents.networks.SPRINTConfig attribute) sparsity (medlatents.inference.PruningConfig attribute) spatial_channels (medlatents.conditioning.ConditioningConfig attribute) spatial_condition (medlatents.conditioning.ConditioningBundle attribute) spatial_key (medlatents.conditioning.ConditionalBatchConfig attribute) spatial_mask (medlatents.conditioning.ConditioningBundle attribute) spatial_mask_key (medlatents.conditioning.ConditionalBatchConfig attribute) spatial_method (medlatents.conditioning.ConditioningConfig attribute) spatial_to_sequence_2d() (medlatents.rasterization.HilbertCurve static method) (medlatents.rasterization.RasterScan static method) (medlatents.rasterization.SCurve static method) (medlatents.rasterization.ZOrderCurve static method) spatial_to_sequence_3d() (medlatents.rasterization.HilbertCurve static method) (medlatents.rasterization.RasterScan static method) (medlatents.rasterization.SCurve static method) (medlatents.rasterization.ZOrderCurve static method) spatial_to_sequence_mask() (in module medlatents.inference) SpeculativeDecoder (class in medlatents.autoregressive) SPINTrainer (class in medlatents.post_training.self_play) split() (medlatents.post_training.PreferenceDataset method) SPO_STANDARD (in module medlatents.post_training) SPOStepOutput (class in medlatents.post_training) SPOTrainer (class in medlatents.post_training) SPRINTConfig (class in medlatents.networks) SPRINTDiT (class in medlatents.networks) SPRINTScheduler (class in medlatents.networks) state_dict() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) states (medlatents.post_training.rl.Trajectory attribute) STD (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) step() (medlatents.bayesian_flow.BaseBFNSolver method) (medlatents.bayesian_flow.DPMSolver2 method) (medlatents.bayesian_flow.DPMSolver3 method) (medlatents.bayesian_flow.EulerSolver method) (medlatents.bayesian_flow.ExponentialIntegrator method) (medlatents.bayesian_flow.HeunSolver method) (medlatents.bayesian_flow.StochasticHeun method) (medlatents.sampling.AncestralSamplingScheduler method) (medlatents.sampling.DDIMScheduler method) (medlatents.sampling.DPMSolverScheduler method) (medlatents.sampling.EulerDiscreteScheduler method) (medlatents.sampling.MaskGITScheduler method) (medlatents.sampling.RunningConfidenceScheduler method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.CurriculumScheduler method) (medlatents.training.MaskingRatioCurriculum method) (medlatents.training.MaskingSchedule method) (medlatents.training.NoiseLevelCurriculum method) (medlatents.training.SequenceLengthCurriculum method) step_idx (medlatents.post_training.SPOStepOutput attribute) (medlatents.post_training.StepwisePreference attribute) step_preference_model (medlatents.post_training.PostTrainingConfig attribute) StepAwarePreferenceModel (class in medlatents.post_training.preference) StepInvariantModel (class in medlatents.flow_matching) StepPreferenceModel (class in medlatents.post_training) StepwisePreference (class in medlatents.post_training) StochasticHeun (class in medlatents.bayesian_flow) structured_dim (medlatents.inference.PruningConfig attribute) super_resolve_slices() (in module medlatents.inference) supports_3d (medlatents.conditioning.FrozenDINOv2 property) (medlatents.conditioning.FrozenEncoder property) (medlatents.conditioning.FrozenMedSigLIP property) (medlatents.conditioning.FrozenNeuroVFM property) (medlatents.conditioning.FrozenSigLIP property) symmetric (medlatents.inference.QuantizationConfig attribute) sync_cuda (medlatents.training.TrainingProfiler attribute)

T

T_destination (medlatents.autoregressive.AutoregressiveTransformer attribute) (medlatents.autoregressive.MedusaHead attribute) (medlatents.autoregressive.MedusaModel attribute) (medlatents.autoregressive.MedusaTrainer attribute) (medlatents.autoregressive.MultiTokenPredictionLoss attribute) (medlatents.autoregressive.speculative.medusa.MedusaModel attribute) (medlatents.bayesian_flow.BayesianFlowTransformer attribute) (medlatents.bayesian_flow.ResidualLossWrapper attribute) (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenEncoder attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) (medlatents.conditioning.SliceWiseEncoder attribute) (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) (medlatents.diffusion.ContinuousTimeMDLM attribute) (medlatents.diffusion.MDLMLoss attribute) (medlatents.diffusion.ScoreEntropyLoss attribute) (medlatents.diffusion.SEDDLoss attribute) (medlatents.flow_matching.MixturePathGeneralizedKL attribute) (medlatents.flow_matching.ShortcutFlowMatchingLoss attribute) (medlatents.flow_matching.ShortcutFlowMatchingModel attribute) (medlatents.flow_matching.StepInvariantModel attribute) (medlatents.flow_matching.TimeInvariantVectorField attribute) (medlatents.flow_matching.WrappedModel attribute) (medlatents.maskgit.MaskGIT attribute) (medlatents.networks.Attention attribute) (medlatents.networks.AttentionWithValueResidual attribute) (medlatents.networks.ContinuousDiT attribute) (medlatents.networks.ContinuousSequenceEmbedder attribute) (medlatents.networks.ContinuousTransformer attribute) (medlatents.networks.DiscreteDiT attribute) (medlatents.networks.DiscreteSequenceEmbedder attribute) (medlatents.networks.DiscreteTransformer attribute) (medlatents.networks.DiTAttention attribute) (medlatents.networks.DiTBlock attribute) (medlatents.networks.DiTBlockWithCrossAttention attribute) (medlatents.networks.DiTCrossAttention attribute) (medlatents.networks.FinalLayer attribute) (medlatents.networks.LabelEmbedder attribute) (medlatents.networks.MMDiTBlock attribute) (medlatents.networks.RMSNorm attribute) (medlatents.networks.RMSNormZero attribute) (medlatents.networks.SPRINTDiT attribute) (medlatents.networks.TimestepEmbedder attribute) (medlatents.networks.TokenRestorer attribute) (medlatents.networks.TokenSelector attribute) (medlatents.networks.TransformerBlock attribute) (medlatents.post_training.DPOLoss attribute) (medlatents.post_training.preference.DiscriminatorScorer attribute) (medlatents.post_training.preference.StepAwarePreferenceModel attribute) (medlatents.post_training.RewardModel attribute) (medlatents.post_training.StepPreferenceModel attribute) (medlatents.sampling.DecoupledSTGumbelSoftmax attribute) (medlatents.sampling.maskgit.TokenCritic attribute) (medlatents.sampling.ReinMax attribute) (medlatents.training.ConditionalLossWrapper attribute) (medlatents.training.REPALoss attribute) (medlatents.training.REPAProjection attribute) target_proj (medlatents.training.REPAProjection attribute) teacher_path (medlatents.post_training.PostTrainingConfig attribute) TemperatureScheduler (class in medlatents.sampling) text_dropout_prob (medlatents.training.ConditionalTrainingConfig attribute) text_embed_dim (medlatents.conditioning.ConditioningConfig attribute) text_embeddings (medlatents.conditioning.ConditioningBundle attribute) text_embeddings_key (medlatents.conditioning.ConditionalBatchConfig attribute) text_mask (medlatents.conditioning.ConditioningBundle attribute) text_mask_key (medlatents.conditioning.ConditionalBatchConfig attribute) text_pooled (medlatents.conditioning.ConditioningBundle attribute) text_pooled_dim (medlatents.conditioning.ConditioningConfig attribute) text_pooled_key (medlatents.conditioning.ConditionalBatchConfig attribute) TimeInvariantVectorField (class in medlatents.flow_matching) timestep (medlatents.post_training.SPOStepOutput attribute) (medlatents.post_training.StepwisePreference attribute) timestep_embed_dim (medlatents.conditioning.ConditioningConfig attribute) timestep_embedding() (medlatents.networks.TimestepEmbedder static method) timestep_threshold (medlatents.training.REPAEConfig attribute) TimestepEmbedder (class in medlatents.networks) timesteps (medlatents.conditioning.ConditioningBundle attribute) (medlatents.post_training.rl.Trajectory attribute) to() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.ConditioningBundle method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.PreferencePair method) (medlatents.post_training.RewardModel method) (medlatents.post_training.rl.Trajectory method) (medlatents.post_training.StepPreferenceModel method) (medlatents.post_training.StepwisePreference method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) to_dict() (medlatents.post_training.DPOOutput method) (medlatents.post_training.PostTrainingConfig method) to_empty() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method) to_pairs() (medlatents.post_training.RankedSamples method) token_drop() (medlatents.networks.LabelEmbedder method) TokenCritic (class in medlatents.sampling.maskgit) tokenize() (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) TokenizedImageDataset (class in medlatents.data) Tokenizer (in module medlatents.data) TokenRestorer (class in medlatents.networks) tokens_key (medlatents.conditioning.ConditionalBatchConfig attribute) TokenSelector (class in medlatents.networks) top_k_samples (medlatents.post_training.PostTrainingConfig attribute) track() (medlatents.training.MemoryProfiler method) train() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.distillation.ConsistencyTrainer method) (medlatents.post_training.distillation.ReflowTrainer method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.RewardModelTrainer method) (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) (medlatents.post_training.self_play.RFTTrainer method) (medlatents.post_training.self_play.SPINTrainer method) (medlatents.post_training.SPOTrainer method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.DiscreteLatentTrainer method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

train_classifier() (in module medlatents.evaluation) train_epoch() (medlatents.training.ContinuousLatentTrainer method) (medlatents.training.DiscreteLatentTrainer method) train_iteration() (medlatents.post_training.self_play.SPINTrainer method) train_on_pairs() (medlatents.post_training.distillation.ReflowTrainer method) train_step() (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.rl.BaseRLTrainer method) (medlatents.post_training.rl.DDPODiscreteFlowTrainer method) (medlatents.post_training.rl.DDPOTrainer method) (medlatents.post_training.rl.GARDOTrainer method) (medlatents.post_training.rl.GRPOTrainer method) (medlatents.post_training.self_play.RFTTrainer method) (medlatents.post_training.self_play.SPINTrainer method) (medlatents.post_training.SPOTrainer method) train_step_distillation() (medlatents.post_training.distillation.ConsistencyTrainer method) train_step_training() (medlatents.post_training.distillation.ConsistencyTrainer method) training (medlatents.autoregressive.AutoregressiveTransformer attribute) (medlatents.autoregressive.MedusaHead attribute) (medlatents.autoregressive.MedusaModel attribute) (medlatents.autoregressive.MedusaTrainer attribute) (medlatents.autoregressive.MultiTokenPredictionLoss attribute) (medlatents.autoregressive.speculative.medusa.MedusaModel attribute) (medlatents.bayesian_flow.BayesianFlowTransformer attribute) (medlatents.bayesian_flow.ResidualLossWrapper attribute) (medlatents.conditioning.FrozenDINOv2 attribute) (medlatents.conditioning.FrozenEncoder attribute) (medlatents.conditioning.FrozenMedSigLIP attribute) (medlatents.conditioning.FrozenNeuroVFM attribute) (medlatents.conditioning.FrozenSigLIP attribute) (medlatents.conditioning.SliceWiseEncoder attribute) (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) (medlatents.diffusion.ContinuousTimeMDLM attribute) (medlatents.diffusion.MDLMLoss attribute) (medlatents.diffusion.ScoreEntropyLoss attribute) (medlatents.diffusion.SEDDLoss attribute) (medlatents.flow_matching.MixturePathGeneralizedKL attribute) (medlatents.flow_matching.ShortcutFlowMatchingLoss attribute) (medlatents.flow_matching.ShortcutFlowMatchingModel attribute) (medlatents.flow_matching.StepInvariantModel attribute) (medlatents.flow_matching.TimeInvariantVectorField attribute) (medlatents.flow_matching.WrappedModel attribute) (medlatents.maskgit.MaskGIT attribute) (medlatents.networks.Attention attribute) (medlatents.networks.AttentionWithValueResidual attribute) (medlatents.networks.ContinuousDiT attribute) (medlatents.networks.ContinuousSequenceEmbedder attribute) (medlatents.networks.ContinuousTransformer attribute) (medlatents.networks.DiscreteDiT attribute) (medlatents.networks.DiscreteSequenceEmbedder attribute) (medlatents.networks.DiscreteTransformer attribute) (medlatents.networks.DiTAttention attribute) (medlatents.networks.DiTBlock attribute) (medlatents.networks.DiTBlockWithCrossAttention attribute) (medlatents.networks.DiTCrossAttention attribute) (medlatents.networks.FinalLayer attribute) (medlatents.networks.LabelEmbedder attribute) (medlatents.networks.MMDiTBlock attribute) (medlatents.networks.RMSNorm attribute) (medlatents.networks.RMSNormZero attribute) (medlatents.networks.SPRINTDiT attribute) (medlatents.networks.TimestepEmbedder attribute) (medlatents.networks.TokenRestorer attribute) (medlatents.networks.TokenSelector attribute) (medlatents.networks.TransformerBlock attribute) (medlatents.post_training.DPOLoss attribute) (medlatents.post_training.preference.DiscriminatorScorer attribute) (medlatents.post_training.preference.StepAwarePreferenceModel attribute) (medlatents.post_training.RewardModel attribute) (medlatents.post_training.StepPreferenceModel attribute) (medlatents.sampling.DecoupledSTGumbelSoftmax attribute) (medlatents.sampling.maskgit.TokenCritic attribute) (medlatents.sampling.ReinMax attribute) (medlatents.training.ConditionalLossWrapper attribute) (medlatents.training.REPALoss attribute) (medlatents.training.REPAProjection attribute) TrainingProfiler (class in medlatents.training) Trajectory (class in medlatents.post_training.rl) TrajectoryBuffer (class in medlatents.post_training.rl) TransformerBlock (class in medlatents.networks) triangular_guidance() (in module medlatents.sampling) truncate_batch() (medlatents.training.SequenceLengthCurriculum method) type() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

U

UniformSourceDistribution (class in medlatents.flow_matching) unrasterize_2d() (in module medlatents.rasterization) update() (medlatents.inference.kv_cache.QuantizedKVCache method) (medlatents.inference.QuantizedKVCache method) (medlatents.sampling.RunningConfidenceRemasker method) update_generated() (medlatents.evaluation.FIDCalculator method) (medlatents.evaluation.PrecisionRecallCalculator method) update_params_bayesian() (medlatents.bayesian_flow.BayesianFlowTransformer method) update_real() (medlatents.evaluation.FIDCalculator method) (medlatents.evaluation.PrecisionRecallCalculator method) update_rotary_cache() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteTransformer method)

use_class (medlatents.conditioning.ConditioningConfig attribute) use_ema (medlatents.post_training.PostTrainingConfig attribute) use_fsdp (medlatents.post_training.PostTrainingConfig attribute) (medlatents.training.DistributedConfig attribute) use_haste (medlatents.training.ConditionalTrainingConfig attribute) use_metadata (medlatents.conditioning.ConditioningConfig attribute) use_repa (medlatents.training.ConditionalTrainingConfig attribute) use_segmentation (medlatents.conditioning.ConditioningConfig attribute) use_spatial (medlatents.conditioning.ConditioningConfig attribute) use_text (medlatents.conditioning.ConditioningConfig attribute) use_timestep (medlatents.conditioning.ConditioningConfig attribute) use_vecor (medlatents.training.ConditionalTrainingConfig attribute)

V

vae_grad_scale (medlatents.training.REPAEConfig attribute) vae_recon_weight (medlatents.training.REPAEConfig attribute) vae_warmup_steps (medlatents.training.REPAEConfig attribute) validate() (medlatents.conditioning.ConditioningConfig method) (medlatents.post_training.AutoregressiveDPOTrainer method) (medlatents.post_training.BaseDPOTrainer method) (medlatents.post_training.D3PMDPOTrainer method) (medlatents.post_training.DiffusionDPOTrainer method) (medlatents.post_training.DiscreteDPOTrainer method) (medlatents.post_training.DiscreteFlowDPOTrainer method) (medlatents.post_training.FlowDPOTrainer method) (medlatents.post_training.MaskGITDPOTrainer method) (medlatents.post_training.RewardModelTrainer method) (medlatents.training.ContinuousLatentTrainer method) (medlatents.training.DiscreteLatentTrainer method)

vecor_temperature (medlatents.training.ConditionalTrainingConfig attribute) vecor_weight (medlatents.training.ConditionalTrainingConfig attribute)

W

wandb_entity (medlatents.post_training.PostTrainingConfig attribute) wandb_project (medlatents.post_training.PostTrainingConfig attribute) warmup() (medlatents.sampling.CompiledSampler method) warmup_steps (medlatents.post_training.PostTrainingConfig attribute) weight_decay (medlatents.post_training.PostTrainingConfig attribute)

weights_name (medlatents.data.ContinuousTokenizer attribute) (medlatents.data.DiscreteTokenizer attribute) world_size (medlatents.training.DistributedConfig attribute) wrap_model_ddp() (in module medlatents.training) wrap_model_fsdp() (in module medlatents.training) WrappedModel (class in medlatents.flow_matching)

X

xpu() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

Z

zero_grad() (medlatents.autoregressive.AutoregressiveTransformer method) (medlatents.autoregressive.MedusaHead method) (medlatents.autoregressive.MedusaModel method) (medlatents.autoregressive.MedusaTrainer method) (medlatents.autoregressive.MultiTokenPredictionLoss method) (medlatents.autoregressive.speculative.medusa.MedusaModel method) (medlatents.bayesian_flow.BayesianFlowTransformer method) (medlatents.bayesian_flow.ResidualLossWrapper method) (medlatents.conditioning.FrozenDINOv2 method) (medlatents.conditioning.FrozenEncoder method) (medlatents.conditioning.FrozenMedSigLIP method) (medlatents.conditioning.FrozenNeuroVFM method) (medlatents.conditioning.FrozenSigLIP method) (medlatents.conditioning.SliceWiseEncoder method) (medlatents.data.ContinuousTokenizer method) (medlatents.data.DiscreteTokenizer method) (medlatents.diffusion.ContinuousTimeMDLM method) (medlatents.diffusion.MDLMLoss method) (medlatents.diffusion.ScoreEntropyLoss method) (medlatents.diffusion.SEDDLoss method) (medlatents.flow_matching.MixturePathGeneralizedKL method) (medlatents.flow_matching.ShortcutFlowMatchingLoss method) (medlatents.flow_matching.ShortcutFlowMatchingModel method) (medlatents.flow_matching.StepInvariantModel method) (medlatents.flow_matching.TimeInvariantVectorField method) (medlatents.flow_matching.WrappedModel method) (medlatents.maskgit.MaskGIT method) (medlatents.networks.Attention method) (medlatents.networks.AttentionWithValueResidual method) (medlatents.networks.ContinuousDiT method) (medlatents.networks.ContinuousSequenceEmbedder method) (medlatents.networks.ContinuousTransformer method) (medlatents.networks.DiscreteDiT method) (medlatents.networks.DiscreteSequenceEmbedder method) (medlatents.networks.DiscreteTransformer method) (medlatents.networks.DiTAttention method) (medlatents.networks.DiTBlock method) (medlatents.networks.DiTBlockWithCrossAttention method) (medlatents.networks.DiTCrossAttention method) (medlatents.networks.FinalLayer method) (medlatents.networks.LabelEmbedder method) (medlatents.networks.MMDiTBlock method) (medlatents.networks.RMSNorm method) (medlatents.networks.RMSNormZero method) (medlatents.networks.SPRINTDiT method) (medlatents.networks.TimestepEmbedder method) (medlatents.networks.TokenRestorer method) (medlatents.networks.TokenSelector method) (medlatents.networks.TransformerBlock method) (medlatents.post_training.DPOLoss method) (medlatents.post_training.preference.DiscriminatorScorer method) (medlatents.post_training.preference.StepAwarePreferenceModel method) (medlatents.post_training.RewardModel method) (medlatents.post_training.StepPreferenceModel method) (medlatents.sampling.DecoupledSTGumbelSoftmax method) (medlatents.sampling.maskgit.TokenCritic method) (medlatents.sampling.ReinMax method) (medlatents.training.ConditionalLossWrapper method) (medlatents.training.REPALoss method) (medlatents.training.REPAProjection method)

ZOrderCurve (class in medlatents.rasterization)