Index

A

adaptive (ls_mlkit.optimizer.sam.SAMConfig attribute) add_callback() (ls_mlkit.pipeline.callback.CallbackManager method) add_callbacks() (ls_mlkit.pipeline.callback.CallbackManager method) (ls_mlkit.pipeline.pipeline.BasePipeline method) add_maybe_special_tokens() (in module ls_mlkit.util.llm) add_special_tokens() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) alpha (ls_mlkit.optimizer.kfa.kfa_optimizer.UserConfig attribute) (ls_mlkit.optimizer.kfa.UserConfig attribute)

ANM (class in ls_mlkit.util.nma.anm) apply() (ls_mlkit.util.se3.T method) apply_inpainting_mask() (ls_mlkit.util.mask.masker.Masker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) apply_mask() (ls_mlkit.util.mask.masker.Masker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) attention() (ls_mlkit.model.decoder_tf.causal_transformer.MultiHeadAttention method) AttentionBlock (class in ls_mlkit.model.decoder_tf.causal_transformer)

B

back_to() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) base_model_prefix (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto attribute) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto attribute) BaseCallback (class in ls_mlkit.pipeline.callback) BaseDiffuser (class in ls_mlkit.diffusion.base_diffuser) BaseDiffuserConfig (class in ls_mlkit.diffusion.base_diffuser) BaseFlow (class in ls_mlkit.flow_matching.base_fm)

BaseFlowConfig (class in ls_mlkit.flow_matching.base_fm) BaseGenerativeModel (class in ls_mlkit.util.base_class.base_gm_class) BaseGenerativeModelConfig (class in ls_mlkit.util.base_class.base_gm_class) BaseLoss (class in ls_mlkit.util.base_class.base_loss_class) BaseLossConfig (class in ls_mlkit.util.base_class.base_loss_class) BasePipeline (class in ls_mlkit.pipeline.pipeline) BaseTimeScheduler (class in ls_mlkit.util.base_class.base_time_class) build_vocab() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method)

C

cache_to_disk() (in module ls_mlkit.util.decorators) calculate_C_inverse_mult_d() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) calculate_fisher_inverse_mult() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) calculate_fisher_inverse_mult_V() (ls_mlkit.optimizer.kfa.KFA static method) (ls_mlkit.optimizer.kfa.kfa.KFA static method) calculate_igso3() (in module ls_mlkit.util.manifold.so3_utils) CallbackEvent (class in ls_mlkit.pipeline.callback) CallbackManager (class in ls_mlkit.pipeline.callback) callbacks (ls_mlkit.pipeline.callback.CallbackManager attribute) CausalLanguageModel (class in ls_mlkit.model.decoder_tf) (class in ls_mlkit.model.decoder_tf.causal_transformer) CausalLanguageModelConfig (class in ls_mlkit.model.decoder_tf) (class in ls_mlkit.model.decoder_tf.causal_transformer) CausalLanguageModelConfigForAuto (class in ls_mlkit.model.decoder_tf) (class in ls_mlkit.model.decoder_tf.causal_transformer) CausalLanguageModelForAuto (class in ls_mlkit.model.decoder_tf) (class in ls_mlkit.model.decoder_tf.causal_transformer) check_cuda() (in module ls_mlkit.util.cuda) check_mask_shape() (ls_mlkit.util.mask.image_masker.ImageMasker method) (ls_mlkit.util.mask.masker.Masker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) class_decorator() (in module ls_mlkit.util.decorators) complete_micro_shape() (ls_mlkit.util.base_class.base_loss_class.BaseLoss method) (ls_mlkit.util.shape_class.Shape method) compose() (ls_mlkit.util.se3.T method) compute_conditional_loss() (ls_mlkit.diffusion.conditioner.conditioner.LGDConditioner method) (ls_mlkit.diffusion.conditioner.LGDConditioner method)

compute_hessian() (ls_mlkit.util.nma.anm.ANM method) compute_loss() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) (ls_mlkit.util.base_class.base_loss_class.BaseLoss method) compute_metrics() (in module ls_mlkit.util.llm) concat() (ls_mlkit.util.se3.T static method) Conditioner (class in ls_mlkit.diffusion.conditioner) (class in ls_mlkit.diffusion.conditioner.conditioner) Config (class in ls_mlkit.util.dequantize) config_class (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto attribute) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto attribute) CONSTANT_WITH_WARMUP (ls_mlkit.util.scheduler.SchedulerType attribute) constant_with_warmup() (in module ls_mlkit.util.scheduler) contact_pair_off (ls_mlkit.util.nma.force_fields.ForceField property) contact_pair_on (ls_mlkit.util.nma.force_fields.ForceField property) contact_shutdown (ls_mlkit.util.nma.force_fields.ForceField property) continuous_time_to_discrete_time() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) convert_id_to_token() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) convert_token_to_id() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) COSINE_DECAY_WITH_WARMUP (ls_mlkit.util.scheduler.SchedulerType attribute) cosine_decay_with_warmup() (in module ls_mlkit.util.scheduler) count_bright_area() (ls_mlkit.util.mask.masker.Masker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) cutoff_distance (ls_mlkit.util.nma.force_fields.ForceField property) (ls_mlkit.util.nma.force_fields.HinsenForceField property) (ls_mlkit.util.nma.force_fields.InvariantForceField property) CyclicGroupGenerator (class in ls_mlkit.dataset.lda_dataset)

D

d_logf_d_omega() (in module ls_mlkit.util.manifold.so3_utils) data_list_to_tensor() (ls_mlkit.dataset.lda_dataset.LDADataset method) dequantize() (in module ls_mlkit.util.dequantize) device (ls_mlkit.util.dequantize.Config attribute) DeviceConfig (class in ls_mlkit.util.base_class.base_config_class) DiffusionTimeScheduler (class in ls_mlkit.diffusion.time_scheduler)

disable() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) (ls_mlkit.util.hook.base_hook.HookHandler method) disable_hook() (ls_mlkit.util.hook.base_hook.HookManager method) discrete_time_to_continuous_time() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) DistributedPipeline (class in ls_mlkit.pipeline.distributed_pipeline) DistributedTrainingConfig (class in ls_mlkit.pipeline.distributed_pipeline)

E

eigen() (ls_mlkit.util.nma.anm.ANM method) enable() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) (ls_mlkit.util.hook.base_hook.HookHandler method) enable_hook() (ls_mlkit.util.hook.base_hook.HookManager method) EPOCH_END (ls_mlkit.pipeline.callback.CallbackEvent attribute) EPOCH_START (ls_mlkit.pipeline.callback.CallbackEvent attribute) eps (ls_mlkit.optimizer.kfa.KFA attribute) (ls_mlkit.optimizer.kfa.kfa.KFA attribute) epsilon (ls_mlkit.optimizer.sam.SAMConfig attribute) epsilon_perturb() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) EuclideanDDIMConfig (class in ls_mlkit.diffusion.euclidean_ddim_diffuser) EuclideanDDIMDiffuser (class in ls_mlkit.diffusion.euclidean_ddim_diffuser) EuclideanDDPMConfig (class in ls_mlkit.diffusion.euclidean_ddpm_diffuser) EuclideanDDPMDiffuser (class in ls_mlkit.diffusion.euclidean_ddpm_diffuser)

EuclideanDiffuser (class in ls_mlkit.diffusion.euclidean_diffuser) EuclideanDiffuserConfig (class in ls_mlkit.diffusion.euclidean_diffuser) EuclideanOTFlow (class in ls_mlkit.flow_matching.euclidean_ot_fm) EuclideanOTFlowConfig (class in ls_mlkit.flow_matching.euclidean_ot_fm) eval() (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) eval_a_step() (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) exp() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) EXPONENTIAL_DECAY_WITH_WARMUP (ls_mlkit.util.scheduler.SchedulerType attribute) exponential_decay_with_warmup() (in module ls_mlkit.util.scheduler) exponential_map() (in module ls_mlkit.util.manifold.so3_utils) extract_number() (ls_mlkit.dataset.MT19937.MT12Bit method) (ls_mlkit.dataset.MT19937.MT16Bit method) (ls_mlkit.dataset.MT19937.MT19937 method) (ls_mlkit.dataset.MT19937.MT2Bit method) (ls_mlkit.dataset.MT19937.MT8Bit method)

F

f_igso3() (in module ls_mlkit.util.manifold.so3_utils) FeedForwardBlock (class in ls_mlkit.model.decoder_tf.causal_transformer) find_linear_modules() (in module ls_mlkit.util.lora) find_tensor_devices() (in module ls_mlkit.util.show) flatten_batch_dimension() (in module ls_mlkit.util.manifold.so3_utils) FlowMatchingTimeScheduler (class in ls_mlkit.flow_matching.time_scheduler) force_constant() (ls_mlkit.util.nma.force_fields.ForceField method) (ls_mlkit.util.nma.force_fields.HinsenForceField method) (ls_mlkit.util.nma.force_fields.InvariantForceField method) ForceField (class in ls_mlkit.util.nma.force_fields) forward() (ls_mlkit.model.decoder_tf.causal_transformer.AttentionBlock method) (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModel method) (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto method) (ls_mlkit.model.decoder_tf.causal_transformer.FeedForwardBlock method) (ls_mlkit.model.decoder_tf.causal_transformer.MultiHeadAttention method) (ls_mlkit.model.decoder_tf.causal_transformer.TransformerBlock method) (ls_mlkit.model.decoder_tf.CausalLanguageModel method) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto method) (ls_mlkit.model.longLinear.LongLinearModel method) (ls_mlkit.model.model_for_pipeline.ModelForPipeline method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method)

forward_process() (ls_mlkit.diffusion.base_diffuser.BaseDiffuser method) (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) forward_process_n_step() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffusion.euclidean_diffuser.EuclideanDiffuser method) ForwardBackwardOffloadHookContext (class in ls_mlkit.util.offload.forward_backward_offload) ForwardHookForDevice (class in ls_mlkit.util.offload.forward_hook) from_4x4() (ls_mlkit.util.se3.T static method) from_tensor() (ls_mlkit.util.se3.T static method) function_mapping (ls_mlkit.util.observer.Observer attribute)

G

generate() (in module ls_mlkit.model.decoder_tf.causal_transformer) generate_dataset() (ls_mlkit.dataset.lda_dataset.LDADataset static method) generate_square_subsequent_mask() (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModel method) (ls_mlkit.model.decoder_tf.CausalLanguageModel method) get() (ls_mlkit.util.scheduler.ObjectAttrsScheduler method) (ls_mlkit.util.scheduler.Scheduler method) get_accumulated_conditional_score() (in module ls_mlkit.diffusion.conditioner.utils) get_align_device_pre_forward_hook() (ls_mlkit.util.offload.forward_hook.ForwardHookForDevice static method) get_all_by_recursion() (ls_mlkit.util.sniffer.Sniffer method) get_and_create_new_log_dir() (in module ls_mlkit.util.log) get_backward_hook() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext static method) get_backward_hook_by_block() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext static method) get_causal_model() (in module ls_mlkit.model.decoder_tf) (in module ls_mlkit.model.decoder_tf.causal_transformer) get_cifar100_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.minist_cifar) get_cifar10_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.minist_cifar) get_collate_fn() (in module ls_mlkit.model.decoder_tf.tokenizer) get_compute_graph() (in module ls_mlkit.util.cgraph) get_condition_post_compute_loss_hook() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) get_condition_pre_update_in_step_fn_hook() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) get_conditional_score() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.LGDConditioner method) (ls_mlkit.diffusion.conditioner.LGDConditioner method) get_continuous_timesteps_schedule() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) get_cosines_and_amplitudes() (in module ls_mlkit.util.vector_utils) get_data_collator() (in module ls_mlkit.util.llm) get_dataset() (in module ls_mlkit.dataset.minist_cifar) get_discrete_timesteps_schedule() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) get_displacements_from_normal_modes() (ls_mlkit.util.nma.anm.ANM method) get_drift_and_diffusion() (ls_mlkit.util.sde.base_sde.SDE method) get_fashionmnist_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.minist_cifar) get_float_weight() (in module ls_mlkit.util.dequantize) get_forward_hook() (ls_mlkit.util.offload.forward_hook.ForwardHookForDevice static method) get_forward_hook_by_block() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext static method) get_full_bright_mask() (ls_mlkit.util.mask.masker.Masker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) get_full_name_list() (ls_mlkit.util.offload.forward_hook.ForwardHookForDevice static method) get_gradient_accumulation_steps() (ls_mlkit.pipeline.dist_pipeline_impl.MyTrainingConfig method) get_gradient_norm() (ls_mlkit.optimizer.sam.SAM method) (ls_mlkit.util.observer.Observer method) get_gradients() (ls_mlkit.util.observer.Observer method) get_guidance_scale() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) get_input_embeddings() (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto method) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto method) get_iris_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.iris) get_lambda_lr_scheduler() (in module ls_mlkit.scheduler.lr_scheduler_factory) get_latest_checkpoint_dir() (ls_mlkit.pipeline.pipeline.BasePipeline method) get_lda_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.lda_dataset) get_learing_rate_scheduler() (in module ls_mlkit.util.utils_for_main) get_logger() (in module ls_mlkit.util.log) get_lora_model() (in module ls_mlkit.util.lora) get_lr_scheduler() (in module ls_mlkit.scheduler) (in module ls_mlkit.scheduler.lr_scheduler_factory)

get_macro_shape() (in module ls_mlkit.util.manifold.so3_utils) (ls_mlkit.util.base_class.base_loss_class.BaseLoss method) (ls_mlkit.util.shape_class.Shape method) get_macroscopic_shape() (in module ls_mlkit.util.shape) get_masks() (in module ls_mlkit.model.decoder_tf.tokenizer) get_minist_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.minist_cifar) get_model_device() (ls_mlkit.model.model_for_pipeline.ModelForPipeline method) get_model_memory() (in module ls_mlkit.util.offload.split) get_module_list() (ls_mlkit.util.offload.forward_hook.ForwardHookForDevice static method) get_new_save_dir() (in module ls_mlkit.util.utils_for_main) get_nma_displacement_from_node_coordinates() (in module ls_mlkit.util.nma) (in module ls_mlkit.util.nma.nma) get_offload_grad_hook() (in module ls_mlkit.util.offload.gradient_offload_v2) (ls_mlkit.util.offload.gradient_offload_v2.GradientOffloadHookContext static method) get_optimizer() (in module ls_mlkit.util.utils_for_main) get_output_embeddings() (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto method) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto method) get_partition_block() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext static method) get_posterior_mean_fn() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) get_probabilities_by_distribution() (ls_mlkit.dataset.lda_dataset.RandomGenerator static method) get_record_gradient_hook() (in module ls_mlkit.util.offload.gradient_offload) (ls_mlkit.util.offload.gradient_offload.GradientOffloadHookContext method) get_regular_language_dataset() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.regular_language) get_reverse_sde() (ls_mlkit.util.sde.base_sde.SDE method) get_rots() (ls_mlkit.util.se3.T method) get_run_name() (in module ls_mlkit.util.utils_for_main) get_save_hook_for_a() (ls_mlkit.optimizer.kfa.KFA static method) (ls_mlkit.optimizer.kfa.kfa.KFA static method) get_save_hook_for_g() (ls_mlkit.optimizer.kfa.KFA static method) (ls_mlkit.optimizer.kfa.kfa.KFA static method) get_sigma2() (ls_mlkit.diffusion.euclidean_ddim_diffuser.EuclideanDDIMDiffuser method) get_something() (ls_mlkit.util.observer.Observer method) get_something_from_targets() (ls_mlkit.util.observer.Observer method) get_something_norm() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) get_split_num() (in module ls_mlkit.util.offload.split) get_statistics() (ls_mlkit.util.observer.Observer method) get_train_class() (in module ls_mlkit.util.utils_for_main) get_trans() (ls_mlkit.util.se3.T method) get_transforms() (in module ls_mlkit.dataset.minist_cifar) get_vector_cosines() (in module ls_mlkit.util.vector_utils) get_vocab_size() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) get_weight_norm() (ls_mlkit.util.observer.Observer method) get_weights() (ls_mlkit.util.observer.Observer method) GMHook (class in ls_mlkit.util.base_class.base_gm_class) GMHookHandler (class in ls_mlkit.util.base_class.base_gm_class) GMHookManager (class in ls_mlkit.util.base_class.base_gm_class) GMHookStageType (class in ls_mlkit.util.base_class.base_gm_class) grad() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) gradient_clip() (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) gradient_norm_fn() (in module ls_mlkit.util.observer) GradientOffloadHookContext (class in ls_mlkit.util.offload.gradient_offload) (class in ls_mlkit.util.offload.gradient_offload_v2) gradients_fn() (in module ls_mlkit.util.observer) greedy_decode() (in module ls_mlkit.model.decoder_tf.generate) guidance_scale (ls_mlkit.diffusion.conditioner.Conditioner property) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner property)

H

hessian (ls_mlkit.util.nma.anm.ANM property) HF_ENDPOINT (ls_mlkit.util.huggingface.HF_MIRROR attribute) HF_MIRROR (class in ls_mlkit.util.huggingface) HinsenForceField (class in ls_mlkit.util.nma.force_fields)

Hook (class in ls_mlkit.util.hook.base_hook) hook (ls_mlkit.util.hook.base_hook.HookHandler property) HookHandler (class in ls_mlkit.util.hook.base_hook) HookManager (class in ls_mlkit.util.hook.base_hook)

I

identity() (ls_mlkit.util.manifold.lie_group.LieGroup method) (ls_mlkit.util.manifold.so3.SO3 method) (ls_mlkit.util.se3.T static method) identity_rot() (ls_mlkit.util.se3.T static method) identity_trans() (ls_mlkit.util.se3.T static method) igso3_density() (in module ls_mlkit.util.manifold.so3_utils) igso3_density_angle() (in module ls_mlkit.util.manifold.so3_utils) igso3_score() (in module ls_mlkit.util.manifold.so3_utils) ImageMasker (class in ls_mlkit.util.mask.image_masker) in_features (ls_mlkit.util.dequantize.Config attribute) inf_iterator() (in module ls_mlkit.util.iterator) inherit_docstring_from_parent() (in module ls_mlkit.util.decorators) inherit_docstrings() (in module ls_mlkit.util.decorators) initialize_timesteps_schedule() (ls_mlkit.diffusion.time_scheduler.DiffusionTimeScheduler method) (ls_mlkit.flow_matching.time_scheduler.FlowMatchingTimeScheduler method) (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method)

inpainting() (ls_mlkit.diffusion.euclidean_diffuser.EuclideanDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) interp() (in module ls_mlkit.util.interp) interp_1d() (in module ls_mlkit.util.interp) InvariantForceField (class in ls_mlkit.util.nma.force_fields) inverse() (ls_mlkit.util.manifold.lie_group.LieGroup method) (ls_mlkit.util.manifold.so3.SO3 method) inverse_transform_sampling() (in module ls_mlkit.util.manifold.so3_utils) invert() (ls_mlkit.util.se3.T method) invert_apply() (ls_mlkit.util.se3.T method) IrisDataset (class in ls_mlkit.dataset.iris) is_enabled() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method)

K

KFA (class in ls_mlkit.optimizer.kfa) (class in ls_mlkit.optimizer.kfa.kfa) KFAOptimizer (class in ls_mlkit.optimizer.kfa) (class in ls_mlkit.optimizer.kfa.kfa_optimizer)

known_modules (ls_mlkit.optimizer.kfa.KFA attribute) (ls_mlkit.optimizer.kfa.kfa.KFA attribute)

L

LDADataset (class in ls_mlkit.dataset.lda_dataset) left_translation() (ls_mlkit.util.manifold.lie_group.LieGroup method) (ls_mlkit.util.manifold.so3.SO3 method) LGDConditioner (class in ls_mlkit.diffusion.conditioner) (class in ls_mlkit.diffusion.conditioner.conditioner) LieGroup (class in ls_mlkit.util.manifold.lie_group) LieGroupDiffuser (class in ls_mlkit.diffusion.lie_group_diffuser) LieGroupDiffuserConfig (class in ls_mlkit.diffusion.lie_group_diffuser) LINEAR_DECAY_WITH_WARMUP (ls_mlkit.util.scheduler.SchedulerType attribute) linear_decay_with_warmup() (in module ls_mlkit.util.scheduler) list_hooks() (ls_mlkit.util.hook.base_hook.HookManager method) load() (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) load_alpaca_gpt4() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) load_checkpoint() (in module ls_mlkit.util.utils_for_main) load_codefeedback() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) load_gsm8k() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) load_meta_math() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) load_mt19937() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_12bits() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_12bits_with_eval() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_16bits() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_16bits_with_eval() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_32bits() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_32bits_with_eval() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_8bits() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_8bits_with_eval() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.MT19937) load_mt19937_binary_stream() (in module ls_mlkit.dataset.MT19937) load_pickle_file() (in module ls_mlkit.util.pkl) load_sst2() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) load_state_dict() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method) load_wizardlm() (in module ls_mlkit.dataset) (in module ls_mlkit.dataset.nature_language) log() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) logarithmic_map() (in module ls_mlkit.util.manifold.so3_utils) LogConfig (class in ls_mlkit.pipeline.pipeline) LongLinearModel (class in ls_mlkit.model.longLinear) ls_mlkit module ls_mlkit.dataset module ls_mlkit.dataset.iris module ls_mlkit.dataset.lda_dataset module ls_mlkit.dataset.minist_cifar module ls_mlkit.dataset.MT19937 module ls_mlkit.dataset.nature_language module ls_mlkit.dataset.regular_language module ls_mlkit.diffusion module ls_mlkit.diffusion.base_diffuser module ls_mlkit.diffusion.conditioner module ls_mlkit.diffusion.conditioner.conditioner module ls_mlkit.diffusion.conditioner.utils module ls_mlkit.diffusion.euclidean_ddim_diffuser module ls_mlkit.diffusion.euclidean_ddpm_diffuser module ls_mlkit.diffusion.euclidean_diffuser module ls_mlkit.diffusion.lie_group_diffuser module ls_mlkit.diffusion.manifold_diffuser module ls_mlkit.diffusion.time_scheduler module ls_mlkit.flow_matching module ls_mlkit.flow_matching.base_fm module ls_mlkit.flow_matching.euclidean_ot_fm module ls_mlkit.flow_matching.time_scheduler module ls_mlkit.model module ls_mlkit.model.decoder_tf module ls_mlkit.model.decoder_tf.causal_transformer module ls_mlkit.model.decoder_tf.generate module ls_mlkit.model.decoder_tf.tokenizer module ls_mlkit.model.longLinear module ls_mlkit.model.model_for_pipeline module ls_mlkit.optimizer module ls_mlkit.optimizer.kfa module ls_mlkit.optimizer.kfa.kfa module ls_mlkit.optimizer.kfa.kfa_optimizer module ls_mlkit.optimizer.sam module ls_mlkit.pipeline module ls_mlkit.pipeline.callback module ls_mlkit.pipeline.dist_pipeline_impl module ls_mlkit.pipeline.distributed_pipeline module

ls_mlkit.pipeline.pipeline module ls_mlkit.scheduler module ls_mlkit.scheduler.lr_scheduler_factory module ls_mlkit.util module ls_mlkit.util.base_class module ls_mlkit.util.base_class.base_config_class module ls_mlkit.util.base_class.base_gm_class module ls_mlkit.util.base_class.base_loss_class module ls_mlkit.util.base_class.base_time_class module ls_mlkit.util.cgraph module ls_mlkit.util.cuda module ls_mlkit.util.decorators module ls_mlkit.util.dequantize module ls_mlkit.util.hash module ls_mlkit.util.hook module ls_mlkit.util.hook.base_hook module ls_mlkit.util.hook.model_hook module ls_mlkit.util.huggingface module ls_mlkit.util.image module ls_mlkit.util.interp module ls_mlkit.util.iterator module ls_mlkit.util.llm module ls_mlkit.util.log module ls_mlkit.util.lora module ls_mlkit.util.manifold module ls_mlkit.util.manifold.lie_group module ls_mlkit.util.manifold.riemannian_manifold module ls_mlkit.util.manifold.so3 module ls_mlkit.util.manifold.so3_utils module ls_mlkit.util.mask module ls_mlkit.util.mask.image_masker module ls_mlkit.util.mask.masker module ls_mlkit.util.mask.masker_interface module ls_mlkit.util.nma module ls_mlkit.util.nma.anm module ls_mlkit.util.nma.force_fields module ls_mlkit.util.nma.nma module ls_mlkit.util.observer module ls_mlkit.util.offload module ls_mlkit.util.offload.context module ls_mlkit.util.offload.forward_backward_offload module ls_mlkit.util.offload.forward_hook module ls_mlkit.util.offload.gradient_offload module ls_mlkit.util.offload.gradient_offload_v2 module ls_mlkit.util.offload.graph_hook module ls_mlkit.util.offload.model_offload module ls_mlkit.util.offload.resource_monitor module ls_mlkit.util.offload.saved_tensor_offload module ls_mlkit.util.offload.split module ls_mlkit.util.pkl module ls_mlkit.util.plot module ls_mlkit.util.plot.box_plot module ls_mlkit.util.plot.comparison_plot module ls_mlkit.util.plot.histogram_plot module ls_mlkit.util.proxy module ls_mlkit.util.resource_monitor module ls_mlkit.util.scheduler module ls_mlkit.util.sde.base_sde module ls_mlkit.util.se3 module ls_mlkit.util.seed module ls_mlkit.util.shape module ls_mlkit.util.shape_class module ls_mlkit.util.show module ls_mlkit.util.sniffer module ls_mlkit.util.utils_for_main module ls_mlkit.util.vector_utils module

M

main() (in module ls_mlkit.util.dequantize) map_tensor_fn() (ls_mlkit.util.se3.T method) Masker (class in ls_mlkit.util.mask.masker) MaskerInterface (class in ls_mlkit.util.mask.masker_interface) masses (ls_mlkit.util.nma.anm.ANM property) merge_masks() (ls_mlkit.model.decoder_tf.causal_transformer.MultiHeadAttention method) metric() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) mode (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext attribute) (ls_mlkit.util.offload.graph_hook.OffloadSavedTensorHook attribute) model_type (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelConfigForAuto attribute) (ls_mlkit.model.decoder_tf.CausalLanguageModelConfigForAuto attribute) ModelForPipeline (class in ls_mlkit.model.model_for_pipeline) ModelHook (class in ls_mlkit.util.hook.model_hook) ModelHookHandler (class in ls_mlkit.util.hook.model_hook) ModelHookManager (class in ls_mlkit.util.hook.model_hook) ModelHookStageType (class in ls_mlkit.util.hook.model_hook) ModelOffloadHookContext (class in ls_mlkit.util.offload.model_offload) module ls_mlkit ls_mlkit.dataset ls_mlkit.dataset.iris ls_mlkit.dataset.lda_dataset ls_mlkit.dataset.minist_cifar ls_mlkit.dataset.MT19937 ls_mlkit.dataset.nature_language ls_mlkit.dataset.regular_language ls_mlkit.diffusion ls_mlkit.diffusion.base_diffuser ls_mlkit.diffusion.conditioner ls_mlkit.diffusion.conditioner.conditioner ls_mlkit.diffusion.conditioner.utils ls_mlkit.diffusion.euclidean_ddim_diffuser ls_mlkit.diffusion.euclidean_ddpm_diffuser ls_mlkit.diffusion.euclidean_diffuser ls_mlkit.diffusion.lie_group_diffuser ls_mlkit.diffusion.manifold_diffuser ls_mlkit.diffusion.time_scheduler ls_mlkit.flow_matching ls_mlkit.flow_matching.base_fm ls_mlkit.flow_matching.euclidean_ot_fm ls_mlkit.flow_matching.time_scheduler ls_mlkit.model ls_mlkit.model.decoder_tf ls_mlkit.model.decoder_tf.causal_transformer ls_mlkit.model.decoder_tf.generate ls_mlkit.model.decoder_tf.tokenizer ls_mlkit.model.longLinear ls_mlkit.model.model_for_pipeline ls_mlkit.optimizer ls_mlkit.optimizer.kfa ls_mlkit.optimizer.kfa.kfa ls_mlkit.optimizer.kfa.kfa_optimizer ls_mlkit.optimizer.sam ls_mlkit.pipeline ls_mlkit.pipeline.callback ls_mlkit.pipeline.dist_pipeline_impl ls_mlkit.pipeline.distributed_pipeline ls_mlkit.pipeline.pipeline ls_mlkit.scheduler ls_mlkit.scheduler.lr_scheduler_factory ls_mlkit.util ls_mlkit.util.base_class ls_mlkit.util.base_class.base_config_class ls_mlkit.util.base_class.base_gm_class ls_mlkit.util.base_class.base_loss_class ls_mlkit.util.base_class.base_time_class ls_mlkit.util.cgraph ls_mlkit.util.cuda ls_mlkit.util.decorators ls_mlkit.util.dequantize ls_mlkit.util.hash ls_mlkit.util.hook ls_mlkit.util.hook.base_hook ls_mlkit.util.hook.model_hook ls_mlkit.util.huggingface ls_mlkit.util.image ls_mlkit.util.interp ls_mlkit.util.iterator ls_mlkit.util.llm ls_mlkit.util.log ls_mlkit.util.lora ls_mlkit.util.manifold ls_mlkit.util.manifold.lie_group ls_mlkit.util.manifold.riemannian_manifold ls_mlkit.util.manifold.so3 ls_mlkit.util.manifold.so3_utils ls_mlkit.util.mask ls_mlkit.util.mask.image_masker ls_mlkit.util.mask.masker ls_mlkit.util.mask.masker_interface ls_mlkit.util.nma ls_mlkit.util.nma.anm ls_mlkit.util.nma.force_fields ls_mlkit.util.nma.nma ls_mlkit.util.observer ls_mlkit.util.offload ls_mlkit.util.offload.context ls_mlkit.util.offload.forward_backward_offload ls_mlkit.util.offload.forward_hook ls_mlkit.util.offload.gradient_offload ls_mlkit.util.offload.gradient_offload_v2 ls_mlkit.util.offload.graph_hook ls_mlkit.util.offload.model_offload ls_mlkit.util.offload.resource_monitor ls_mlkit.util.offload.saved_tensor_offload ls_mlkit.util.offload.split ls_mlkit.util.pkl ls_mlkit.util.plot ls_mlkit.util.plot.box_plot ls_mlkit.util.plot.comparison_plot ls_mlkit.util.plot.histogram_plot ls_mlkit.util.proxy ls_mlkit.util.resource_monitor ls_mlkit.util.scheduler ls_mlkit.util.sde.base_sde ls_mlkit.util.se3 ls_mlkit.util.seed ls_mlkit.util.shape ls_mlkit.util.shape_class ls_mlkit.util.show ls_mlkit.util.sniffer ls_mlkit.util.utils_for_main ls_mlkit.util.vector_utils

MT12Bit (class in ls_mlkit.dataset.MT19937) MT16Bit (class in ls_mlkit.dataset.MT19937) MT19937 (class in ls_mlkit.dataset.MT19937) MT2Bit (class in ls_mlkit.dataset.MT19937) MT8Bit (class in ls_mlkit.dataset.MT19937) MultiHeadAttention (class in ls_mlkit.model.decoder_tf.causal_transformer) multiply() (ls_mlkit.util.manifold.lie_group.LieGroup method) (ls_mlkit.util.manifold.so3.SO3 method) MyDistributedPipeline (class in ls_mlkit.pipeline.dist_pipeline_impl) MyTrainingConfig (class in ls_mlkit.pipeline.dist_pipeline_impl)

N

N_A (in module ls_mlkit.util.nma.anm) natoms (ls_mlkit.util.nma.force_fields.ForceField property)

next() (ls_mlkit.dataset.lda_dataset.CyclicGroupGenerator method) numpy_to_pil() (in module ls_mlkit.util.image)

O

ObjectAttrsScheduler (class in ls_mlkit.util.scheduler) Observer (class in ls_mlkit.util.observer) offload_condition() (in module ls_mlkit.util.offload.graph_hook) offload_device (ls_mlkit.util.offload.graph_hook.OffloadSavedTensorHook attribute)

OffloadContext (class in ls_mlkit.util.offload) (class in ls_mlkit.util.offload.context) OffloadSavedTensorHook (class in ls_mlkit.util.offload.graph_hook) on_event() (ls_mlkit.pipeline.callback.BaseCallback method) out_features (ls_mlkit.util.dequantize.Config attribute)

P

pack() (ls_mlkit.util.offload.graph_hook.OffloadSavedTensorHook static method) perturb() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) plot_boxplot() (in module ls_mlkit.util.plot.box_plot) plot_entity_wise_comparison() (in module ls_mlkit.util.plot.comparison_plot) plot_histogram_and_kde() (in module ls_mlkit.util.plot) (in module ls_mlkit.util.plot.histogram_plot) POST_BACKWARD (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_COMPUTE_LOSS (ls_mlkit.pipeline.callback.CallbackEvent attribute) (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) (ls_mlkit.util.hook.model_hook.ModelHookStageType attribute) POST_EVAL (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_EVAL_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_GET_MACRO_SHAPE (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) POST_LOAD (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_OPTIMIZER_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_SAMPLING_TIME_STEP (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) POST_SAVE (ls_mlkit.pipeline.callback.CallbackEvent attribute) POST_UPDATE_IN_STEP_FN (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) PRE_BACKWARD (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_COMPUTE_LOSS (ls_mlkit.pipeline.callback.CallbackEvent attribute) (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) (ls_mlkit.util.hook.model_hook.ModelHookStageType attribute)

PRE_EVAL (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_EVAL_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_LOAD (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_OPTIMIZER_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_SAVE (ls_mlkit.pipeline.callback.CallbackEvent attribute) PRE_UPDATE_IN_STEP_FN (ls_mlkit.util.base_class.base_gm_class.GMHookStageType attribute) prepare_condition_dict() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) prepare_inputs_for_generation() (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto method) (ls_mlkit.model.decoder_tf.CausalLanguageModelForAuto method) preprocess_logits_for_metrics() (in module ls_mlkit.util.llm) print_cpu_memory() (in module ls_mlkit.util.offload.resource_monitor) (in module ls_mlkit.util.resource_monitor) print_gpu_memory() (in module ls_mlkit.util.offload.resource_monitor) (in module ls_mlkit.util.resource_monitor) prior_sampling() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) pt_to_pil() (in module ls_mlkit.util.image)

Q

q_xt_x_0() (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method)

quat_to_rot() (in module ls_mlkit.util.se3)

R

random_tangent() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) RandomGenerator (class in ls_mlkit.dataset.lda_dataset) recover_bright_region() (ls_mlkit.diffusion.euclidean_diffuser.EuclideanDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) register_class_to_dict() (in module ls_mlkit.util.decorators) register_forward_hook_by_module() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext method) register_gradient_hook() (ls_mlkit.util.offload.gradient_offload.GradientOffloadHookContext method) register_hook() (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) (ls_mlkit.util.hook.base_hook.HookManager method) register_hook_by_block() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext method) register_hooks() (ls_mlkit.model.model_for_pipeline.ModelForPipeline method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) (ls_mlkit.util.hook.base_hook.HookManager method) register_model() (in module ls_mlkit.model.decoder_tf.causal_transformer) register_post_compute_loss_hook() (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) register_save_hook() (ls_mlkit.optimizer.kfa.KFA method) (ls_mlkit.optimizer.kfa.kfa.KFA method)

RegularLanguageDataset (class in ls_mlkit.dataset.regular_language) remove() (ls_mlkit.util.hook.base_hook.HookHandler method) remove_hook() (ls_mlkit.optimizer.kfa.KFA method) (ls_mlkit.optimizer.kfa.kfa.KFA method) replace_module_with_linear() (in module ls_mlkit.util.dequantize) require_keys() (in module ls_mlkit.util.decorators) rho (ls_mlkit.optimizer.kfa.kfa_optimizer.UserConfig attribute) (ls_mlkit.optimizer.kfa.UserConfig attribute) (ls_mlkit.optimizer.sam.SAMConfig attribute) rho_cov (ls_mlkit.optimizer.kfa.kfa_optimizer.UserConfig attribute) (ls_mlkit.optimizer.kfa.UserConfig attribute) RiemannianManifold (class in ls_mlkit.util.manifold.riemannian_manifold) RiemannianManifoldDiffuser (class in ls_mlkit.diffusion.manifold_diffuser) RiemannianManifoldDiffuserConfig (class in ls_mlkit.diffusion.manifold_diffuser) rigid_from_3_points() (ls_mlkit.util.se3.T static method) rot_matmul() (in module ls_mlkit.util.se3) rot_vec_mul() (in module ls_mlkit.util.se3) rotation_matrix_to_angle() (in module ls_mlkit.util.manifold.so3_utils) run_hooks() (ls_mlkit.util.hook.base_hook.HookManager method)

S

SAM (class in ls_mlkit.optimizer.sam) SAMConfig (class in ls_mlkit.optimizer.sam) sample_a_continuous_time_step_uniformly() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) sample_a_discrete_time_step_uniformly() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) sampling() (ls_mlkit.diffusion.euclidean_diffuser.EuclideanDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) save() (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) save_grad() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) save_hash_to_file() (in module ls_mlkit.util.hash) save_moving_average_and_save_d() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) save_params() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) save_pickle_file() (in module ls_mlkit.util.pkl) save_state_dict() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) SavedTensorOffloadContext (class in ls_mlkit.util.offload.saved_tensor_offload) scale_translation() (ls_mlkit.util.se3.T method) Scheduler (class in ls_mlkit.util.scheduler) SchedulerType (class in ls_mlkit.util.scheduler) SDE (class in ls_mlkit.util.sde.base_sde) seed_everything() (in module ls_mlkit.util.seed) set_closure() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method) set_condition() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) set_continuous_timesteps_schedule() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) set_discrete_timesteps_schedule() (ls_mlkit.util.base_class.base_time_class.BaseTimeScheduler method) set_eos_token() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method)

set_guidance_scale() (ls_mlkit.diffusion.conditioner.Conditioner method) (ls_mlkit.diffusion.conditioner.conditioner.Conditioner method) set_hf_mirror() (ls_mlkit.util.huggingface.HF_MIRROR static method) set_pad_token() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) set_proxy() (in module ls_mlkit.util.proxy) Shape (class in ls_mlkit.util.shape_class) shape (ls_mlkit.util.se3.T property) ShapeConfig (class in ls_mlkit.util.shape_class) show_gpu_and_cpu_memory() (in module ls_mlkit.util.offload.resource_monitor) (in module ls_mlkit.util.resource_monitor) show_info() (in module ls_mlkit.util.show) skew_symmetric_to_angle() (in module ls_mlkit.util.manifold.so3_utils) skew_symmetric_to_vector() (in module ls_mlkit.util.manifold.so3_utils) sniff_file() (ls_mlkit.util.sniffer.Sniffer method) sniff_file_by_path_pattern() (ls_mlkit.util.sniffer.Sniffer method) Sniffer (class in ls_mlkit.util.sniffer) SO3 (class in ls_mlkit.util.manifold.so3) softmax() (in module ls_mlkit.dataset.lda_dataset) state_dict() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method) step() (ls_mlkit.diffusion.euclidean_ddim_diffuser.EuclideanDDIMDiffuser method) (ls_mlkit.diffusion.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.flow_matching.euclidean_ot_fm.EuclideanOTFlow method) (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method) (ls_mlkit.util.base_class.base_gm_class.BaseGenerativeModel method) (ls_mlkit.util.scheduler.ObjectAttrsScheduler method) (ls_mlkit.util.scheduler.Scheduler method) STEP_END (ls_mlkit.pipeline.callback.CallbackEvent attribute) STEP_START (ls_mlkit.pipeline.callback.CallbackEvent attribute) stop_rot_gradient() (ls_mlkit.util.se3.T method)

T

T (class in ls_mlkit.util.se3) (ls_mlkit.util.sde.base_sde.SDE property) table_print_dict() (in module ls_mlkit.util.show) test_gen() (in module ls_mlkit.model.decoder_tf.causal_transformer) timer() (in module ls_mlkit.util.decorators) to() (ls_mlkit.util.base_class.base_config_class.DeviceConfig method) (ls_mlkit.util.se3.T method) to_4x4() (ls_mlkit.util.se3.T method) tokenize() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) Tokenizer (class in ls_mlkit.model.decoder_tf.tokenizer) trace() (in module ls_mlkit.util.manifold.so3_utils) train() (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method)

train_a_step() (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) train_an_epoch() (ls_mlkit.pipeline.pipeline.BasePipeline method) TRAINING_END (ls_mlkit.pipeline.callback.CallbackEvent attribute) TRAINING_START (ls_mlkit.pipeline.callback.CallbackEvent attribute) TrainingConfig (class in ls_mlkit.pipeline.pipeline) TrainingState (class in ls_mlkit.pipeline.pipeline) transform_dataframe_to_tensor() (in module ls_mlkit.dataset.iris) transform_label_to_integer() (in module ls_mlkit.dataset.iris) TransformerBlock (class in ls_mlkit.model.decoder_tf.causal_transformer) trigger() (ls_mlkit.pipeline.callback.CallbackManager method) trigger_callbacks() (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) twist() (ls_mlkit.dataset.MT19937.MT19937 method)

U

unit_skew_symmetric() (in module ls_mlkit.util.manifold.so3_utils) unpack() (ls_mlkit.util.offload.graph_hook.OffloadSavedTensorHook static method) unregister_hook() (ls_mlkit.util.hook.base_hook.HookManager method)

unset_hf_mirror() (ls_mlkit.util.huggingface.HF_MIRROR static method) unsqueeze() (ls_mlkit.util.se3.T method) UserConfig (class in ls_mlkit.optimizer.kfa) (class in ls_mlkit.optimizer.kfa.kfa_optimizer)

V

vector_to_skew_symmetric() (in module ls_mlkit.util.manifold.so3_utils)

W

weight_norm_fn() (in module ls_mlkit.util.observer)

weights_fn() (in module ls_mlkit.util.observer)

X

xi (ls_mlkit.optimizer.kfa.kfa_optimizer.UserConfig attribute) (ls_mlkit.optimizer.kfa.UserConfig attribute)

Z

zero_grad() (ls_mlkit.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method)