Index

A

adaptive (ls_mlkit.optimizer.sam.SAMConfig attribute) add_callback() (ls_mlkit.pipeline.callback.CallbackManager method) add_callbacks() (ls_mlkit.pipeline.BasePipeline method) (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) AFTER_BACKWARD (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_COMPUTE_LOSS (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_EVAL (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_EVAL_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_LOAD (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_OPTIMIZER_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) AFTER_SAVE (ls_mlkit.pipeline.callback.CallbackEvent attribute) 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.bio_masker.BioBackboneFrameMasker method) (ls_mlkit.util.mask.bio_masker.BioCAOnlyMasker method) (ls_mlkit.util.mask.bio_masker.BioSO3Masker method) (ls_mlkit.util.mask.image_masker.ImageMasker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) apply_mask() (ls_mlkit.util.mask.bio_masker.BioBackboneFrameMasker method) (ls_mlkit.util.mask.bio_masker.BioCAOnlyMasker method) (ls_mlkit.util.mask.bio_masker.BioSO3Masker method) (ls_mlkit.util.mask.image_masker.ImageMasker 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) BaseConfigClass (class in ls_mlkit.util.base_config_class) BaseDiffuser (class in ls_mlkit.diffuser.base_diffuser) BaseDiffuserConfig (class in ls_mlkit.diffuser.base_diffuser) BasePipeline (class in ls_mlkit.pipeline) (class in ls_mlkit.pipeline.pipeline)

BEFORE_BACKWARD (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_COMPUTE_LOSS (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_EVAL (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_EVAL_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_LOAD (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_OPTIMIZER_STEP (ls_mlkit.pipeline.callback.CallbackEvent attribute) BEFORE_SAVE (ls_mlkit.pipeline.callback.CallbackEvent attribute) BioBackboneFrameMasker (class in ls_mlkit.util.mask.bio_masker) BioCAOnlyMasker (class in ls_mlkit.util.mask.bio_masker) BioSO3Masker (class in ls_mlkit.util.mask.bio_masker) 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) 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.bio_masker.BioBackboneFrameMasker method) (ls_mlkit.util.mask.bio_masker.BioCAOnlyMasker method) (ls_mlkit.util.mask.bio_masker.BioSO3Masker method) (ls_mlkit.util.mask.image_masker.ImageMasker method) (ls_mlkit.util.mask.masker_interface.MaskerInterface method) class_decorator() (in module ls_mlkit.util.decorators) complete_micro_shape() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) compose() (ls_mlkit.util.se3.T method) compute_conditional_loss() (ls_mlkit.diffuser.conditioner.conditioner.LGDConditioner method) (ls_mlkit.diffuser.conditioner.LGDConditioner method) compute_hessian() (ls_mlkit.util.nma.anm.ANM method) compute_loss() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser method) (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method)

compute_metrics() (in module ls_mlkit.util.llm) concat() (ls_mlkit.util.se3.T static method) Conditioner (class in ls_mlkit.diffuser.conditioner) (class in ls_mlkit.diffuser.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() (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.diffuser.time_scheduler.TimeScheduler 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) Corrector (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.corrector) cosine_decay_with_warmup() (in module ls_mlkit.util.scheduler) count_bright_area() (ls_mlkit.util.mask.bio_masker.BioBackboneFrameMasker method) (ls_mlkit.util.mask.bio_masker.BioCAOnlyMasker method) (ls_mlkit.util.mask.bio_masker.BioSO3Masker method) (ls_mlkit.util.mask.image_masker.ImageMasker 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) disable() (ls_mlkit.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.conditioner.conditioner.Conditioner method)

discrete_time_to_continuous_time() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) DistributedPipeline (class in ls_mlkit.pipeline) (class in ls_mlkit.pipeline.distributed_pipeline) DistributedTrainingConfig (class in ls_mlkit.pipeline) (class in ls_mlkit.pipeline.distributed_pipeline)

E

eigen() (ls_mlkit.util.nma.anm.ANM method) enable() (ls_mlkit.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.conditioner.conditioner.Conditioner 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.diffuser.euclidean_ddim_diffuser) EuclideanDDIMDiffuser (class in ls_mlkit.diffuser.euclidean_ddim_diffuser) EuclideanDDPMConfig (class in ls_mlkit.diffuser.euclidean_ddpm_diffuser) EuclideanDDPMDiffuser (class in ls_mlkit.diffuser.euclidean_ddpm_diffuser) EuclideanDiffuser (class in ls_mlkit.diffuser.euclidean_diffuser)

EuclideanDiffuserConfig (class in ls_mlkit.diffuser.euclidean_diffuser) eval() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) eval_a_step() (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.MyDistributedPipeline method) exp() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) 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) 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.diffuser.base_diffuser.BaseDiffuser method) (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)

forward_process() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser method) forward_process_n_step() (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.euclidean_diffuser.EuclideanDiffuser method) forward_process_one_step() (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.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.ObjectAttrScheduler method) (ls_mlkit.util.scheduler.ObjectAttrsScheduler method) (ls_mlkit.util.scheduler.Scheduler method) get_accumulated_conditional_score() (ls_mlkit.diffuser.euclidean_diffuser.EuclideanDiffuser method) 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_conditional_score() (ls_mlkit.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.conditioner.conditioner.Conditioner method) (ls_mlkit.diffuser.conditioner.conditioner.LGDConditioner method) (ls_mlkit.diffuser.conditioner.LGDConditioner method) get_continuous_timesteps_schedule() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) get_correct_n_training_steps() (in module ls_mlkit.util.utils_for_main) 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_diffusion_coefficient_with_proper_shape() (ls_mlkit.diffuser.sde.base_sde.SDE method) (ls_mlkit.diffuser.sde.SDE method) get_discrete_timesteps_schedule() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) get_discretization_steps() (ls_mlkit.diffuser.sde.base_sde.SDE method) (ls_mlkit.diffuser.sde.SDE method) get_discretized_drift_and_diffusion() (ls_mlkit.diffuser.sde.base_sde.SDE method) (ls_mlkit.diffuser.sde.SDE method) (ls_mlkit.diffuser.sde.sde_lib.VESDE method) (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.VESDE method) (ls_mlkit.diffuser.sde.VPSDE method) get_displacements_from_normal_modes() (ls_mlkit.util.nma.anm.ANM method) get_drift_and_diffusion() (ls_mlkit.diffuser.sde.base_sde.SDE method) (ls_mlkit.diffuser.sde.SDE method) (ls_mlkit.diffuser.sde.sde_lib.SubVPSDE method) (ls_mlkit.diffuser.sde.sde_lib.VESDE method) (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.SubVPSDE method) (ls_mlkit.diffuser.sde.VESDE method) (ls_mlkit.diffuser.sde.VPSDE 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.bio_masker.BioBackboneFrameMasker method) (ls_mlkit.util.mask.bio_masker.BioCAOnlyMasker method) (ls_mlkit.util.mask.bio_masker.BioSO3Masker method) (ls_mlkit.util.mask.image_masker.ImageMasker 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) (ls_mlkit.pipeline.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_ground_truth_score() (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser 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.BasePipeline method) (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_and_micro_shape() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) get_macro_shape() (in module ls_mlkit.util.manifold.so3_utils) (ls_mlkit.diffuser.base_diffuser.BaseDiffuser 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.diffuser.model_interface.Model4DiffuserInterface method) get_model_fn() (in module ls_mlkit.diffuser.sde) (in module ls_mlkit.diffuser.sde.score_fn_utils) 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_nma_displacement_from_protein_ligand_complex() (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_pc_sampler() (in module ls_mlkit.diffuser.sde) (in module ls_mlkit.diffuser.sde.sampler) get_posterior_mean_fn() (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser 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.diffuser.sde.base_sde.SDE method) (ls_mlkit.diffuser.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_score_fn() (in module ls_mlkit.diffuser.sde) (in module ls_mlkit.diffuser.sde.score_fn_utils) get_sigma2() (ls_mlkit.diffuser.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_target_score() (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.VPSDE 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) grad() (ls_mlkit.util.manifold.riemannian_manifold.RiemannianManifold method) (ls_mlkit.util.manifold.so3.SO3 method) gradient_clip() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.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.diffuser.conditioner.Conditioner property) (ls_mlkit.diffuser.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)

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_cdf (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser property) igso3_density() (in module ls_mlkit.util.manifold.so3_utils) igso3_density_angle() (in module ls_mlkit.util.manifold.so3_utils) igso3_discrete_omega (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser property) igso3_discrete_sigma (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser property) igso3_exp_score_norms (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser property) igso3_score() (in module ls_mlkit.util.manifold.so3_utils) igso3_score_norm (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser property) 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.diffuser.time_scheduler.TimeScheduler method) inpainting_x0_unconditionally() (ls_mlkit.diffuser.euclidean_diffuser.EuclideanDiffuser 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.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.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

LangevinCorrector (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.corrector) 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.diffuser.conditioner) (class in ls_mlkit.diffuser.conditioner.conditioner) LieGroup (class in ls_mlkit.util.manifold.lie_group) LieGroupDiffuser (class in ls_mlkit.diffuser.lie_group_diffuser) LieGroupDiffuserConfig (class in ls_mlkit.diffuser.lie_group_diffuser) linear_decay_with_warmup() (in module ls_mlkit.util.scheduler) load() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.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_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) log_distribution() (ls_mlkit.util.observer.Observer method) log_gradient_norm() (ls_mlkit.util.observer.Observer method) log_gradients_distribution() (ls_mlkit.util.observer.Observer method) log_histograms() (ls_mlkit.util.observer.Observer method) log_statistics() (ls_mlkit.util.observer.Observer method) log_weight_norm() (ls_mlkit.util.observer.Observer method) log_weights_distribution() (ls_mlkit.util.observer.Observer method) logarithmic_map() (in module ls_mlkit.util.manifold.so3_utils) LogConfig (class in ls_mlkit.pipeline) (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.diffuser module ls_mlkit.diffuser.base_diffuser module ls_mlkit.diffuser.conditioner module ls_mlkit.diffuser.conditioner.conditioner module ls_mlkit.diffuser.euclidean_ddim_diffuser module ls_mlkit.diffuser.euclidean_ddpm_diffuser module ls_mlkit.diffuser.euclidean_diffuser module ls_mlkit.diffuser.lie_group_diffuser module ls_mlkit.diffuser.manifold_diffuser module ls_mlkit.diffuser.model_interface module ls_mlkit.diffuser.sde module ls_mlkit.diffuser.sde.base_sde module ls_mlkit.diffuser.sde.corrector module ls_mlkit.diffuser.sde.predictor module ls_mlkit.diffuser.sde.sampler module ls_mlkit.diffuser.sde.score_fn_utils module ls_mlkit.diffuser.sde.sde_lib module ls_mlkit.diffuser.so3_diffuser module ls_mlkit.diffuser.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.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_config_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.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.bio_masker module ls_mlkit.util.mask.image_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.proxy module ls_mlkit.util.resource_monitor module ls_mlkit.util.scheduler module ls_mlkit.util.se3 module ls_mlkit.util.seed module ls_mlkit.util.shape 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) marginal_prob() (ls_mlkit.diffuser.sde.sde_lib.SubVPSDE method) (ls_mlkit.diffuser.sde.sde_lib.VESDE method) (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.SubVPSDE method) (ls_mlkit.diffuser.sde.VESDE method) (ls_mlkit.diffuser.sde.VPSDE method) 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) Model4DiffuserInterface (class in ls_mlkit.diffuser.model_interface) model_type (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelConfigForAuto attribute) (ls_mlkit.model.decoder_tf.CausalLanguageModelConfigForAuto attribute) 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.diffuser ls_mlkit.diffuser.base_diffuser ls_mlkit.diffuser.conditioner ls_mlkit.diffuser.conditioner.conditioner ls_mlkit.diffuser.euclidean_ddim_diffuser ls_mlkit.diffuser.euclidean_ddpm_diffuser ls_mlkit.diffuser.euclidean_diffuser ls_mlkit.diffuser.lie_group_diffuser ls_mlkit.diffuser.manifold_diffuser ls_mlkit.diffuser.model_interface ls_mlkit.diffuser.sde ls_mlkit.diffuser.sde.base_sde ls_mlkit.diffuser.sde.corrector ls_mlkit.diffuser.sde.predictor ls_mlkit.diffuser.sde.sampler ls_mlkit.diffuser.sde.score_fn_utils ls_mlkit.diffuser.sde.sde_lib ls_mlkit.diffuser.so3_diffuser ls_mlkit.diffuser.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.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_config_class ls_mlkit.util.cgraph ls_mlkit.util.cuda ls_mlkit.util.decorators ls_mlkit.util.dequantize ls_mlkit.util.hash 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.bio_masker ls_mlkit.util.mask.image_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.proxy ls_mlkit.util.resource_monitor ls_mlkit.util.scheduler ls_mlkit.util.se3 ls_mlkit.util.seed ls_mlkit.util.shape 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) (class in ls_mlkit.pipeline.dist_pipeline_impl) MyTrainingConfig (class in ls_mlkit.pipeline) (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) NoneCorrector (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.corrector)

NonePredictor (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.predictor) numpy_to_pil() (in module ls_mlkit.util.image)

O

ObjectAttrScheduler (class in ls_mlkit.util.scheduler) 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) Predictor (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.predictor) prepare_batch_data_for_input() (ls_mlkit.diffuser.model_interface.Model4DiffuserInterface method) prepare_condition_dict() (ls_mlkit.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.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_logp() (ls_mlkit.diffuser.sde.sde_lib.SubVPSDE method) (ls_mlkit.diffuser.sde.sde_lib.VESDE method) (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.SubVPSDE method) (ls_mlkit.diffuser.sde.VESDE method) (ls_mlkit.diffuser.sde.VPSDE method) prior_sampling() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.sde.sde_lib.SubVPSDE method) (ls_mlkit.diffuser.sde.sde_lib.VESDE method) (ls_mlkit.diffuser.sde.sde_lib.VPSDE method) (ls_mlkit.diffuser.sde.SubVPSDE method) (ls_mlkit.diffuser.sde.VESDE method) (ls_mlkit.diffuser.sde.VPSDE method) (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser method) pt_to_pil() (in module ls_mlkit.util.image)

Q

q_xt_x_0() (ls_mlkit.diffuser.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) 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_by_block() (ls_mlkit.util.offload.forward_backward_offload.ForwardBackwardOffloadHookContext method) register_model() (in module ls_mlkit.model.decoder_tf.causal_transformer) 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_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) ReverseDiffusionPredictor (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.predictor) 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.diffuser.manifold_diffuser) RiemannianManifoldDiffuserConfig (class in ls_mlkit.diffuser.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)

S

SAM (class in ls_mlkit.optimizer.sam) SAMConfig (class in ls_mlkit.optimizer.sam) sample_a_continuous_time_step_uniformly() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) sample_a_discrete_time_step_uniformly() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) sample_noise_in_lie_algebra() (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser method) sample_x0_unconditionally() (ls_mlkit.diffuser.euclidean_diffuser.EuclideanDiffuser method) sample_xtm1_conditional_on_xt() (ls_mlkit.diffuser.base_diffuser.BaseDiffuser method) (ls_mlkit.diffuser.euclidean_ddim_diffuser.EuclideanDDIMDiffuser method) (ls_mlkit.diffuser.euclidean_ddpm_diffuser.EuclideanDDPMDiffuser method) (ls_mlkit.diffuser.so3_diffuser.SO3Diffuser method) save() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.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_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) SDE (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.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.diffuser.conditioner.Conditioner method) (ls_mlkit.diffuser.conditioner.conditioner.Conditioner method) set_continuous_timesteps_schedule() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method) set_discrete_timesteps_schedule() (ls_mlkit.diffuser.time_scheduler.TimeScheduler method)

set_eos_token() (ls_mlkit.model.decoder_tf.tokenizer.Tokenizer method) set_guidance_scale (ls_mlkit.diffuser.conditioner.Conditioner property) (ls_mlkit.diffuser.conditioner.conditioner.Conditioner property) 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 (ls_mlkit.util.se3.T property) shared_corrector_update_fn() (in module ls_mlkit.diffuser.sde.sampler) shared_predictor_update_fn() (in module ls_mlkit.diffuser.sde.sampler) 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) SO3Diffuser (class in ls_mlkit.diffuser.so3_diffuser) SO3DiffuserConfig (class in ls_mlkit.diffuser.so3_diffuser) 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.optimizer.kfa.kfa_optimizer.KFAOptimizer method) (ls_mlkit.optimizer.kfa.KFAOptimizer method) (ls_mlkit.optimizer.sam.SAM method) (ls_mlkit.util.scheduler.ObjectAttrScheduler 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) SubVPSDE (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.sde_lib)

T

T (class in ls_mlkit.util.se3) (ls_mlkit.diffuser.sde.base_sde.SDE property) (ls_mlkit.diffuser.sde.SDE property) (ls_mlkit.diffuser.sde.sde_lib.SubVPSDE property) (ls_mlkit.diffuser.sde.sde_lib.VESDE property) (ls_mlkit.diffuser.sde.sde_lib.VPSDE property) (ls_mlkit.diffuser.sde.SubVPSDE property) (ls_mlkit.diffuser.sde.VESDE property) (ls_mlkit.diffuser.sde.VPSDE 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) TimeScheduler (class in ls_mlkit.diffuser.time_scheduler) to() (ls_mlkit.util.base_config_class.BaseConfigClass 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.BasePipeline method) (ls_mlkit.pipeline.dist_pipeline_impl.MyDistributedPipeline method) (ls_mlkit.pipeline.MyDistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method)

train_a_step() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.pipeline.DistributedPipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) train_an_epoch() (ls_mlkit.pipeline.BasePipeline method) (ls_mlkit.pipeline.pipeline.BasePipeline method) training (ls_mlkit.model.decoder_tf.causal_transformer.CausalLanguageModelForAuto attribute) TRAINING_END (ls_mlkit.pipeline.callback.CallbackEvent attribute) TRAINING_START (ls_mlkit.pipeline.callback.CallbackEvent attribute) TrainingConfig (class in ls_mlkit.pipeline) (class in ls_mlkit.pipeline.pipeline) TrainingState (class in ls_mlkit.pipeline) (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.BasePipeline method) (ls_mlkit.pipeline.distributed_pipeline.DistributedPipeline method) (ls_mlkit.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) unset_hf_mirror() (ls_mlkit.util.huggingface.HF_MIRROR static method) unsqueeze() (ls_mlkit.util.se3.T method) update_fn() (ls_mlkit.diffuser.sde.Corrector method) (ls_mlkit.diffuser.sde.corrector.Corrector method) (ls_mlkit.diffuser.sde.corrector.LangevinCorrector method) (ls_mlkit.diffuser.sde.corrector.NoneCorrector method) (ls_mlkit.diffuser.sde.LangevinCorrector method) (ls_mlkit.diffuser.sde.NoneCorrector method) (ls_mlkit.diffuser.sde.NonePredictor method) (ls_mlkit.diffuser.sde.Predictor method) (ls_mlkit.diffuser.sde.predictor.NonePredictor method) (ls_mlkit.diffuser.sde.predictor.Predictor method) (ls_mlkit.diffuser.sde.predictor.ReverseDiffusionPredictor method) (ls_mlkit.diffuser.sde.ReverseDiffusionPredictor 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) VESDE (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.sde_lib)

VPSDE (class in ls_mlkit.diffuser.sde) (class in ls_mlkit.diffuser.sde.sde_lib)

W

wandb_logger() (in module ls_mlkit.util.decorators)

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)