Index

_

__class_vars__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __private_attributes__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_complete__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_computed_fields__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_core_schema__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_custom_init__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_decorators__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_extra__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_fields__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute)

__pydantic_fields_set__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_generic_metadata__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_parent_namespace__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_post_init__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_private__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_root_model__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_serializer__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __pydantic_validator__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) __signature__ (ab_upt_config.AnchorBranchedUPTConfig attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute)

A

ab_upt module ab_upt_config module ABORT (emmi_data_management.verification.FailAction attribute) abupt module AbuptCollator (class in abupt) (class in emmi_inference.models.pipelines.abupt) accelerator_to_device() (in module ksuit.distributed.run.utils) accumulate_messages() (emmi.modules.encoder.supernode_pooling.SupernodePooling method) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly method) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly static method) act (emmi.modules.mlp.upactdown_mlp.UpActDownMlp attribute) (emmi_inference.models.modules.mlp.Mlp attribute) Activation (class in emmi.modules.activations) activation (mlp.mlp.MLPConfig attribute) ActivationTypes (in module emmi.types) add_global_handlers() (in module ksuit.utils.logging.core) add_nonscalar() (ksuit.writers.log_writer.LogWriter method) add_samples() (ksuit.utils.training.counter.UpdateCounter method) add_scalar() (ksuit.writers.log_writer.LogWriter method) AddBatch (class in emmi_inference.preprocess) additional_outputs (ksuit.trainers.types.TrainerResult attribute) AeroDataSpecs (class in emmi.schemas.data.data_specs) affected_items (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) AFTER (emmi_inference.UploadMode attribute) after_epoch() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) after_training() (ksuit.callbacks.base.callback_base.CallbackBase method) after_update() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) aggregation (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) AhmedMLDataset (class in emmi.datasets.ahmedml) AhmedMLDefaultSplitIDs (class in emmi.datasets.ahmedml) AhmedMLStats (class in emmi.datasets.ahmedml) ALL_CONVS (in module emmi.functional.init) all_ctor_kwarg_names() (in module ksuit.factory.base.reflection_utils) all_gather_grad() (in module ksuit.distributed.gather) all_gather_grad_autograd() (in module ksuit.distributed.gather) all_gather_nograd() (in module ksuit.distributed.gather) all_gather_nograd_clipped() (in module ksuit.distributed.gather) ALL_LAYERS (in module emmi.functional.init) all_parameters (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) all_reduce_mean_grad() (in module ksuit.distributed.gather) all_reduce_mean_nograd() (in module ksuit.distributed.gather) all_reduce_sum_grad() (in module ksuit.distributed.gather) all_reduce_sum_nograd() (in module ksuit.distributed.gather) AllGatherGradAutograd (class in ksuit.distributed.functional.all_gather_grad_autograd) ALLOWED_KEYS (ksuit.configs.static_config.StaticConfig attribute) amp() (emmi_inference.device.DeviceContext method) anchor_suffix (attention.anchor_attention.config.MultiBranchAnchorAttentionConfig attribute) (emmi.modules.attention.anchor_attention.multi_branch_anchor_attention.MultiBranchAnchorAttention attribute) AnchorAttention (class in emmi_inference.models.modules.attention.anchor_attention) AnchorBranchedUPTConfig (class in ab_upt_config) AnchoredBranchedUPT (class in ab_upt) (class in emmi_inference.models.abupt)

AnchorPointSamplingPreCollator (class in emmi_inference.models.pipelines.anchor_point_sampling_precollator) app (in module emmi_data_management.cli) (in module emmi_inference.cli) apply_ema() (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback method) apply_initializers() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) apply_resume_initializer() (ksuit.trainers.base.BaseTrainer method) approximator_blocks (upt.UPT attribute) approximator_config (upt_config.UPTConfig attribute) as_query() (emmi_inference.models.pipelines.shapenet_multistage.DataKeys static method) as_target() (emmi_inference.models.pipelines.shapenet_multistage.DataKeys method) assert_positive (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) (layers.continuous_sincos_embedding_config.ContinuousSincosEmbeddingConfig attribute) (layers.rope_frequenty_config.RopeFrequencyConfig attribute) assert_same_length() (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) AsyncWriter (class in emmi_inference.writer) atomic_write_and_hash() (in module emmi_data_management.checkpoint.iohash) attention module attention.anchor_attention module attention.anchor_attention.config module attention.config module attention_arguments (blocks.transformer_block_config.TransformerBlockConfig attribute) (transolver_config.TransolverConfig attribute) (transolver_config.TransolverPlusPlusConfig attribute) attention_block (emmi.modules.blocks.transformer_block.TransformerBlock attribute) attention_constructor (blocks.transformer_block_config.TransformerBlockConfig attribute) (transolver_config.TransolverConfig attribute) (transolver_config.TransolverPlusPlusConfig attribute) AttentionConfig (class in attention.config) AttentionConfig.Config (class in attention.config) AttentionPattern (class in attention.anchor_attention.config) attn (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.transformer_block.TransformerBlock attribute) AUTO (emmi_inference.device.DeviceEnum attribute), [1] autocast (emmi_inference.device.DeviceContext attribute), [1] average_lap_time (ksuit.utils.common.stopwatch.Stopwatch property) AWS (emmi_data_management.providers.Provider attribute) AWS_ACCESS_KEY_ID (emmi_data_management.interfaces.s3.AWSSecrets attribute) AWS_DEFAULT_REGION (emmi_data_management.interfaces.s3.AWSSecrets attribute) AWS_ENDPOINT_URL (emmi_data_management.interfaces.s3.AWSSecrets attribute) aws_estimate() (in module emmi_data_management.cli_aws) aws_fetch() (in module emmi_data_management.cli_aws) aws_file() (in module emmi_data_management.cli_aws) AWS_REGION (emmi_data_management.interfaces.s3.AWSSecrets attribute) aws_s3_app (in module emmi_data_management.cli_aws) AWS_SECRET_ACCESS_KEY (emmi_data_management.interfaces.s3.AWSSecrets attribute) AWS_SESSION_TOKEN (emmi_data_management.interfaces.s3.AWSSecrets attribute) AWSSecrets (class in emmi_data_management.interfaces.s3) AZURE (emmi_data_management.providers.Provider attribute)

B

backward() (ksuit.distributed.functional.all_gather_grad_autograd.AllGatherGradAutograd static method) barrier() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) BaseModel (class in emmi_inference.models.base_model) BaseTracker (class in ksuit.trackers.base_tracker) basetracker_config_uri (ksuit.providers.path_provider.PathProvider property) basetracker_entries_uri (ksuit.providers.path_provider.PathProvider property) basetracker_summary_uri (ksuit.providers.path_provider.PathProvider property) BaseTrainer (class in ksuit.trainers.base) batch_idx_key (emmi.pipeline.collators.concat_sparse_tensor.ConcatSparseTensorCollator attribute) batch_keys (ksuit.trainers.base.BaseTrainer attribute) batch_processors (ksuit.data.pipeline.multistage.MultiStagePipeline attribute) batch_sampler (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) batch_size (ksuit.callbacks.base.periodic_callback.PeriodicCallback attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) BatchProcessor (class in ksuit.data.pipeline.batch_processor) BatchProcessorType (in module ksuit.data.pipeline.multistage) before_training() (ksuit.callbacks.base.callback_base.CallbackBase method) benchmark() (in module emmi_data_management.diskcache.cache_benchmark) best_metric (ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper attribute) best_metric_value (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) BestCheckpointCallback (class in ksuit.callbacks.checkpoint_callbacks.best_checkpoint) BestMetricCallback (class in ksuit.callbacks.online_callbacks.best_metric) BFLOAT16_ALIASES (in module ksuit.utils.torch.amp)

bias (attention.config.AttentionConfig attribute) (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) (emmi.modules.layers.transformer_batchnorm.TransformerBatchNorm attribute) (layers.linear_projection_config.LinearProjectionConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) bias_layers (upt.UPT attribute) (upt_config.UPTConfig attribute) Bidict (class in ksuit.utils.bidict) blocks module blocks (emmi.modules.decoder.deep_perceiver_decoder.DeepPerceiverDecoder attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) (emmi_inference.models.transolver.Transolver attribute) (emmi_inference.models.upt.UPT attribute) (transformer.Transformer attribute) blocks.perceiver_block_config module blocks.transformer_block_config module BOLD_RED (ksuit.utils.logging.formatting.CustomFormatter attribute) branches (attention.anchor_attention.config.MultiBranchAnchorAttentionConfig attribute) (emmi.modules.attention.anchor_attention.multi_branch_anchor_attention.MultiBranchAnchorAttention attribute) bucket (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute), [1] buffers (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) build_manifest() (in module emmi_data_management.verification) bytes_done (emmi_data_management.logging_progress.LogProgress attribute)

C

cache_cleanup_high_watermark (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) cache_cleanup_low_watermark (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) cache_storage_mode (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) cache_storage_size (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) calculate_start() (ksuit.data.samplers.interleaved_sampler.InterleavedSampler static method) call_after_training() (ksuit.trainers.base.BaseTrainer method) call_before_training() (ksuit.trainers.base.BaseTrainer method) callback_state_dicts (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) CallbackBase (class in ksuit.callbacks.base.callback_base) callbacks (ksuit.trainers.base.BaseTrainer attribute) can_handle() (emmi_data_management.checkpoint.providers.file_provider.FileProvider method) (emmi_data_management.checkpoint.providers.huggingface_provider.HFProvider method) (emmi_data_management.checkpoint.types.Provider method) cat_ranges() (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem method) center_shift (emmi.utils.point_cloud_utils.PointCloudRegistration attribute) check_all_none() (in module ksuit.utils.common.validation) check_at_least_one() (in module ksuit.utils.common.validation) check_at_most_one() (in module ksuit.utils.common.validation) check_dims() (mlp.upactdown_mlp.UpActDownMLPConfig method) check_exclusive() (in module ksuit.utils.common.validation) check_inclusive() (in module ksuit.utils.common.validation) check_positive_values() (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig class method) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig class method) check_single_device_visible() (in module ksuit.distributed.run.utils) checkpoint_app (in module emmi_data_management.cli_checkpoint) checkpoint_fetch() (in module emmi_data_management.cli_checkpoint) checkpoint_path (ksuit.providers.path_provider.PathProvider property) checkpoint_tag (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) checkpoint_verify() (in module emmi_data_management.cli_checkpoint) checkpoint_writer (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.trainers.base.BaseTrainer attribute) CheckpointCallback (class in ksuit.callbacks.checkpoint_callbacks.checkpoint) CheckpointInitializer (class in ksuit.initializers.base.checkpoint_initializer) CheckpointKeys (class in ksuit.writers.checkpoint_writer) CheckpointMetadata (class in emmi_data_management.checkpoint.types) CheckpointWriter (class in ksuit.writers.checkpoint_writer) class_constructor_from_class_path() (in module ksuit.factory.base.reflection_utils) CLEANUP_LOCK_FILE (in module emmi_data_management.diskcache.lru_cache), [1] CLI_HELP_MARKDOWN (in module emmi_data_management.cli_aws) close() (emmi_data_management.interfaces.s3_uploader.S3FileUploader method) (emmi_data_management.logging_progress.LogProgress method) (emmi_inference.s3_spooling_uploader.S3SpoolingUploader method) (emmi_inference.writer.AsyncWriter method) (ksuit.trackers.base_tracker.BaseTracker method) Collator (class in ksuit.data.pipeline.collator) collator (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) CollatorRegistry (in module emmi_inference.collator_registry) CollatorType (in module ksuit.data.pipeline.collator) Compose (class in emmi_inference.preprocess) ComposePreProcess (class in ksuit.data.preprocessors.compose) CompositeModel (class in ksuit.models.composite) compute_metrics() (in module emmi_inference.runner) compute_model_norm() (in module ksuit.utils.model.helpers) compute_rope_args() (upt.UPT method) compute_src_and_dst_indices() (emmi.modules.encoder.supernode_pooling.SupernodePooling method) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly method) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly method)

concat_tensor_precollator module ConcatSparseTensorCollator (class in emmi.pipeline.collators.concat_sparse_tensor) ConcatTensorPreCollator (class in concat_tensor_precollator) (class in emmi_inference.models.pipelines.concat_tensor_precollator) condition_dim (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) conditioning_dims (emmi.schemas.data.data_specs.AeroDataSpecs attribute) config (ksuit.data.base.dataset.Dataset attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) (ksuit.trainers.base.BaseTrainer attribute) config_kind (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) CONFIG_PATH (in module emmi_data_management.credentials) configure_logging() (in module emmi_data_management.logging_config) (in module emmi_inference.logging_config) ConstantSchedule (class in ksuit.schedules.constant_schedule) contains_string() (ksuit.utils.training.training_iteration.TrainingIteration static method) CONTEXT_SETTINGS (in module emmi_inference.cli) ContinuousSincosEmbed (class in emmi.modules.layers.continuous_sincos_embed) (class in emmi_inference.models.modules.continuous_sincos_embed) ContinuousSincosEmbeddingConfig (class in layers.continuous_sincos_embedding_config) copy() (ksuit.utils.training.training_iteration.TrainingIteration method) copy_params() (in module ksuit.utils.model.model_utils) cosine() (in module ksuit.schedules.functional) CosineDecreasingSchedule (class in ksuit.schedules.cosine.decreasing) CosineIncreasingSchedule (class in ksuit.schedules.cosine.increasing) counts (ksuit.utils.logging.core.MessageCounter attribute) CPU (emmi_inference.device.DeviceEnum attribute), [1] create() (ksuit.factory.base.factory_base.Factory method) (ksuit.factory.schedule_factory.ScheduleFactory method) create_batch_idx (emmi.pipeline.collators.concat_sparse_tensor.ConcatSparseTensorCollator attribute) create_dict() (ksuit.factory.base.factory_base.Factory method) create_list() (ksuit.factory.base.factory_base.Factory method) create_messages() (emmi.modules.encoder.supernode_pooling.SupernodePooling method) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly method) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly method) create_path_from_str() (in module ksuit.utils.common.path) create_rope_frequencies() (ab_upt.AnchoredBranchedUPT method) create_slices() (emmi.modules.attention.transolver_attention.TransolverAttention method) create_split() (emmi.datasets.ahmedml.AhmedMLDefaultSplitIDs static method) (emmi.datasets.drivaerml.DrivAerMLDefaultSplitIDs static method) create_writer() (in module emmi_inference.api) CrossAchorAttentionConfig (class in attention.anchor_attention.config) CrossAnchorAttention (class in emmi.modules.attention.anchor_attention.cross_anchor_attention) CTX (in module emmi_data_management.cli) (in module emmi_data_management.cli_aws) (in module emmi_data_management.cli_checkpoint) (in module emmi_data_management.cli_huggingface) (in module emmi_data_management.cli_verification) CUDA (emmi_inference.device.DeviceEnum attribute), [1] cur_iteration (ksuit.utils.training.counter.UpdateCounter attribute) CustomFormatter (class in ksuit.utils.logging.formatting) CustomSchedule (class in ksuit.schedules.custom_schedule)

D

data_container (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.models.model_base.ModelBase attribute) (ksuit.trainers.base.BaseTrainer attribute) data_source_modes (ksuit.configs.static_config.StaticConfig property) data_specs (ab_upt.AnchoredBranchedUPT attribute) DataContainer (class in ksuit.utils.data.data_container) DataFieldKeyMap (class in emmi.preprocessing.process_drivaer_and_ahmed_volumetric) DataKeys (class in emmi_inference.models.pipelines.shapenet_multistage) Dataset (class in ksuit.data.base.dataset) DATASET (emmi_data_management.cli_huggingface.HuggingFaceRepoType attribute) dataset (ksuit.data.base.wrapper.DatasetWrapper attribute) (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) dataset_key (ksuit.callbacks.offline_callbacks.offline_loss.OfflineLossCallback attribute) dataset_keymaps (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) dataset_statistics (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) dataset_wrapper_factory (ksuit.factory.dataset_factory.DatasetFactory attribute) DatasetConfigProvider (class in ksuit.providers.dataset_config_provider) DatasetFactory (class in ksuit.factory.dataset_factory) DatasetRegistry (in module emmi_inference.dataset_registry) datasets (ksuit.utils.data.data_container.DataContainer attribute) DatasetStatsCallback (class in ksuit.callbacks.default_callbacks.dataset_stats) DatasetWrapper (class in ksuit.data.base.wrapper) datefmt (ksuit.utils.logging.formatting.CustomFormatter attribute) debug (emmi_inference.runner.InferencePayload attribute) (ksuit.providers.path_provider.PathProvider attribute) decoder module decoder (upt.UPT attribute) decoder.deep_perceiver_decoder module decoder_blocks_forward() (ab_upt.AnchoredBranchedUPT method) decoder_config (upt_config.UPTConfig attribute) decreases_interval (ksuit.schedules.step.decreasing.StepDecreasingSchedule attribute) DecreasingProgressSchedule (class in ksuit.schedules.base.decreasing_progress_schedule) DeepPerceiverDecoder (class in emmi.modules.decoder.deep_perceiver_decoder) DeepPerceiverDecoderConfig (class in decoder.deep_perceiver_decoder) DEFAULT_CHUNK (in module emmi_data_management.checkpoint.iohash) default_cudnn_benchmark (ksuit.configs.static_config.StaticConfig property) default_cudnn_deterministic (ksuit.configs.static_config.StaticConfig property) default_split_ids (emmi.datasets.ahmedml.AhmedMLDataset property) (emmi.datasets.drivaerml.DrivAerMLDataset property) default_tensor_precollator module default_value (default_tensor_precollator.DefaultTensorPreCollator attribute) (emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator attribute) DefaultCollator (class in emmi.pipeline.collators.default) DefaultTensorPreCollator (class in default_tensor_precollator) (class in emmi_inference.models.pipelines.default_tensor_precollator) DELETE (emmi_data_management.verification.FailAction attribute) delta (ksuit.schedules.base.progress_schedule.ProgressSchedule attribute) denormalize() (emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer.ShiftAndScaleNormalizer method) (emmi.modules.preprocessors.numpy_to_torch_tensor.NumpyToTorchTensorPreProcessor method) (emmi.pipeline.batch_processors.moment_normalization.MomentNormalizationBatchProcessor method) (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor method) (emmi.pipeline.batch_processors.rename_keys.RenameKeysBatchProcessor method) (ksuit.data.pipeline.batch_processor.BatchProcessor method) (ksuit.data.preprocessors.preprocessor.PreProcessor method)

depth (decoder.deep_perceiver_decoder.DeepPerceiverDecoderConfig attribute) (emmi_inference.models.transolver.Transolver attribute) (transformer_config.TransformerConfig attribute) derive_run_name() (ksuit.runners.hydra_runner.HydraRunner static method) describe() (emmi_data_management.verification.FailAction method) device (emmi_inference.device.DeviceContext attribute), [1] (ksuit.models.composite.CompositeModel property) (ksuit.models.model_base.ModelBase property) (ksuit.models.single.Model property) (ksuit.trainers.base.BaseTrainer attribute) device_context (emmi_inference.runner.InferenceRunner attribute) DeviceContext (class in emmi_inference.device) DeviceEnum (class in emmi_inference.device) DeviceType (in module emmi_inference.device) dict_mode (emmi_inference.writer.AsyncWriter attribute) dict_to_string() (in module ksuit.utils.logging.formatting) DictMode (class in emmi_inference.common) dim (concat_tensor_precollator.ConcatTensorPreCollator attribute) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) (emmi_inference.models.pipelines.concat_tensor_precollator.ConcatTensorPreCollator attribute) (emmi_inference.models.transolver.Transolver attribute) (emmi_inference.preprocess.AddBatch attribute), [1] (emmi_inference.preprocess.NormalizeAny attribute), [1] dim_head (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) disable() (in module ksuit.utils.torch.amp) DISABLED (ksuit.configs.wandb_config.WandBMode attribute) DistributedConfig (class in ksuit.distributed.config.distributed_config) docs() (in module emmi_data_management.cli_aws) DotProductAttention (class in emmi.modules.attention.dot_product_attention) (class in emmi_inference.models.modules.attention.dot_product_attention) DotProductAttentionConfig (class in attention.config) DotProductAttentionConfig.Config (class in attention.config) DrivAerMLDataset (class in emmi.datasets.drivaerml) DrivAerMLDefaultSplitIDs (class in emmi.datasets.drivaerml) DrivAerMLStats (class in emmi.datasets.drivaerml) drop_last (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) drop_path (blocks.transformer_block_config.TransformerBlockConfig attribute) drop_path1 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) drop_path2 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) drop_path_rate (ab_upt_config.AnchorBranchedUPTConfig attribute) drop_prob (emmi.modules.layers.drop_path.UnquantizedDropPath attribute) (layers.drop_path_config.UnquantizedDropPathConfig attribute) dropout (attention.config.AttentionConfig attribute) (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) DropOutliersSampleProcessor (class in emmi.pipeline.sample_processors.drop_outliers) dropped (emmi_inference.runner.InferencePayload attribute) dst_idx_hook (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) dtype (emmi_inference.device.DeviceContext attribute), [1] DuplicateKeysSampleProcessor (class in emmi.pipeline.sample_processors.duplicate_keys)

E

EarlyStopIteration EarlyStopperBase (class in ksuit.callbacks.early_stoppers.early_stopper_base) effective_batch_size (ksuit.utils.training.counter.UpdateCounter attribute) elapsed_milliseconds (ksuit.utils.common.stopwatch.Stopwatch property) elapsed_seconds (ksuit.utils.common.stopwatch.Stopwatch property) elementwise_affine (emmi.modules.layers.transformer_batchnorm.TransformerBatchNorm attribute) ELU (emmi.modules.activations.Activation attribute) EmaCallback (class in ksuit.callbacks.checkpoint_callbacks.ema) embed (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) (emmi_inference.models.transolver.Transolver attribute) emit() (ksuit.callbacks.default_callbacks.eta.EtaCallback.LoggerWasCalledHandler method) (ksuit.utils.logging.core.MessageCounter method) emmi module emmi.callbacks module emmi.callbacks.supernode_pooling_coverage module emmi.datasets module emmi.datasets.ahmedml module emmi.datasets.drivaerml module emmi.functional module emmi.functional.init module emmi.functional.logscale module emmi.functional.modulation module emmi.functional.rope module emmi.models module emmi.modules module emmi.modules.activations module emmi.modules.attention module emmi.modules.attention.anchor_attention module emmi.modules.attention.anchor_attention.cross_anchor_attention module emmi.modules.attention.anchor_attention.joint_anchor_attention module emmi.modules.attention.anchor_attention.mixed_attention module emmi.modules.attention.anchor_attention.multi_branch_anchor_attention module emmi.modules.attention.anchor_attention.self_anchor_attention module emmi.modules.attention.dot_product_attention module emmi.modules.attention.irregular_nat_attention module emmi.modules.attention.perceiver_attention module emmi.modules.attention.transolver_attention module emmi.modules.attention.transolver_plusplus_attention module emmi.modules.blocks module emmi.modules.blocks.irregular_nat_block module emmi.modules.blocks.perceiver_block module emmi.modules.blocks.perceiver_transformer_blockpair module emmi.modules.blocks.transformer_block module emmi.modules.decoder module emmi.modules.decoder.deep_perceiver_decoder module emmi.modules.encoder module emmi.modules.encoder.mlp_encoder module emmi.modules.encoder.supernode_pooling module emmi.modules.encoder.supernode_pooling_posonly module emmi.modules.layers module emmi.modules.layers.continuous_sincos_embed module emmi.modules.layers.drop_path module emmi.modules.layers.layer_scale module emmi.modules.layers.linear_projection module emmi.modules.layers.rope_frequency module emmi.modules.layers.scalar_conditioner module emmi.modules.layers.transformer_batchnorm module emmi.modules.mlp module emmi.modules.mlp.mlp module emmi.modules.mlp.upactdown_mlp module emmi.modules.preprocessors module emmi.modules.preprocessors.normalizers module emmi.modules.preprocessors.normalizers.mean_std_normalization module emmi.modules.preprocessors.normalizers.position_normalizer module emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer module emmi.modules.preprocessors.normalizers.types module emmi.modules.preprocessors.numpy_to_torch_tensor module emmi.optim module emmi.optim.lion module emmi.pipeline module emmi.pipeline.batch_processors module emmi.pipeline.batch_processors.moment_normalization module emmi.pipeline.batch_processors.position_normalization module emmi.pipeline.batch_processors.rename_keys module emmi.pipeline.collators module emmi.pipeline.collators.concat_sparse_tensor module emmi.pipeline.collators.default module emmi.pipeline.collators.field_decoder module emmi.pipeline.sample_processors module emmi.pipeline.sample_processors.drop_outliers module emmi.pipeline.sample_processors.duplicate_keys module emmi.pipeline.sample_processors.moment_normalization module emmi.pipeline.sample_processors.point_sampling module emmi.pipeline.sample_processors.position_normalization module emmi.pipeline.sample_processors.rename_keys module emmi.pipeline.sample_processors.replace_key module emmi.pipeline.sample_processors.samplewise_normalization module emmi.pipeline.sample_processors.supernode_sampling module emmi.preprocessing module emmi.preprocessing.calculate_statistics_all module emmi.preprocessing.calculate_statistics_train module emmi.preprocessing.process_drivaer_and_ahmed_volumetric module emmi.preprocessing.shapenet_car module emmi.preprocessing.shapenet_car.preprocess_shapenet_car module emmi.schemas module emmi.schemas.data module emmi.schemas.data.data_specs module emmi.schemas.models module emmi.setup module emmi.types module emmi.utils module emmi.utils.aistore_utils module emmi.utils.point_cloud_utils module emmi.utils.running_moments module emmi_data_management module emmi_data_management.checkpoint module emmi_data_management.checkpoint.iohash module emmi_data_management.checkpoint.providers module emmi_data_management.checkpoint.providers.file_provider module

emmi_data_management.checkpoint.providers.huggingface_provider module emmi_data_management.checkpoint.resolver module emmi_data_management.checkpoint.types module emmi_data_management.cli module emmi_data_management.cli_aws module emmi_data_management.cli_checkpoint module emmi_data_management.cli_huggingface module emmi_data_management.cli_utils module emmi_data_management.cli_verification module emmi_data_management.credentials module emmi_data_management.diskcache module emmi_data_management.diskcache.cache_benchmark module emmi_data_management.diskcache.lru_cache module emmi_data_management.interfaces module emmi_data_management.interfaces.azure module emmi_data_management.interfaces.gcp module emmi_data_management.interfaces.huggingface module emmi_data_management.interfaces.s3 module emmi_data_management.interfaces.s3_uploader module emmi_data_management.logging_config module emmi_data_management.logging_progress module emmi_data_management.providers module emmi_data_management.verification module emmi_inference module emmi_inference.api module emmi_inference.cli module emmi_inference.cli_utils module emmi_inference.collator_registry module emmi_inference.common module emmi_inference.dataset_registry module emmi_inference.datasets module emmi_inference.datasets.shapenet module emmi_inference.device module emmi_inference.examples module emmi_inference.examples.run_abupt module emmi_inference.examples.run_transolver module emmi_inference.examples.run_upt module emmi_inference.logging_config module emmi_inference.models module emmi_inference.models.abupt module emmi_inference.models.base_model module emmi_inference.models.loader module emmi_inference.models.modules module emmi_inference.models.modules.attention module emmi_inference.models.modules.attention.anchor_attention module emmi_inference.models.modules.attention.dot_product_attention module emmi_inference.models.modules.attention.perceiver_attention module emmi_inference.models.modules.attention.sharedweights_crossattn_attention module emmi_inference.models.modules.attention.sharedweights_splitattn_attention module emmi_inference.models.modules.blocks module emmi_inference.models.modules.blocks.perceiver_block module emmi_inference.models.modules.blocks.transformer_block module emmi_inference.models.modules.continuous_sincos_embed module emmi_inference.models.modules.mlp module emmi_inference.models.modules.rope module emmi_inference.models.modules.rope_frequency module emmi_inference.models.modules.supernode_pooling_posonly module emmi_inference.models.pipelines module emmi_inference.models.pipelines.abupt module emmi_inference.models.pipelines.anchor_point_sampling_precollator module emmi_inference.models.pipelines.concat_tensor_precollator module emmi_inference.models.pipelines.default_tensor_precollator module emmi_inference.models.pipelines.field_decoder module emmi_inference.models.pipelines.mesh_to_grid_edges_precollator module emmi_inference.models.pipelines.query_to_grid_edges_precollator module emmi_inference.models.pipelines.shapenet_multistage module emmi_inference.models.pipelines.sparse_tensor_offset module emmi_inference.models.pipelines.surface_mask_precollator module emmi_inference.models.registry module emmi_inference.models.transolver module emmi_inference.models.upt module emmi_inference.preprocess module emmi_inference.runner module emmi_inference.s3_spooling_uploader module emmi_inference.writer module enabled (ksuit.extractors.forward_hook.ForwardHook attribute) encoder module encoder (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) (emmi_inference.models.upt.UPT attribute) (upt.UPT attribute) encoder.supernode_pooling_config module encoder_depth (upt_config.UPTConfig attribute) end_checkpoint (ksuit.trainers.base.BaseTrainer attribute) end_iteration (ksuit.utils.training.counter.UpdateCounter attribute) end_percent (ksuit.schedules.base.sequential_percent_schedule.SequentialPercentScheduleConfig attribute) end_step (ksuit.schedules.base.sequential_step_schedule.SequentialStepScheduleConfig attribute) enforce_size_every_seconds (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) entity (ksuit.configs.wandb_config.WandBConfig attribute), [1] (ksuit.trackers.wandb_tracker.WandbTracker attribute) epoch (ksuit.utils.training.counter.UpdateCounter property) (ksuit.utils.training.training_iteration.TrainingIteration attribute) epoch_as_float (ksuit.utils.training.counter.UpdateCounter property) epoch_format (ksuit.callbacks.default_callbacks.eta.EtaCallback attribute) eps (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.layers.transformer_batchnorm.TransformerBatchNorm attribute) EPSILON (emmi.modules.preprocessors.normalizers.mean_std_normalization.MeanStdNormalization attribute) error (emmi_inference.runner.InferencePayload attribute) errors (emmi_data_management.verification.ParallelErrors attribute) (emmi_inference.cli_utils.ParallelErrors attribute), [1] estimate_hf_repo_size() (in module emmi_data_management.interfaces.huggingface) estimate_s3_size() (in module emmi_data_management.interfaces.s3) estimate_size() (emmi_data_management.checkpoint.providers.file_provider.FileProvider method) (emmi_data_management.checkpoint.providers.huggingface_provider.HFProvider method) (emmi_data_management.checkpoint.types.Provider method) estimate_transforms() (emmi.utils.point_cloud_utils.PointCloudRegistration method) EtaCallback (class in ksuit.callbacks.default_callbacks.eta) EtaCallback.LoggerWasCalledHandler (class in ksuit.callbacks.default_callbacks.eta) etag (emmi_data_management.interfaces.s3.S3Object attribute) (emmi_data_management.verification.FileRecord attribute) every_n_epochs (ksuit.callbacks.base.periodic_callback.PeriodicCallback attribute) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) every_n_samples (ksuit.callbacks.base.periodic_callback.PeriodicCallback attribute) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) every_n_updates (ksuit.callbacks.base.periodic_callback.PeriodicCallback attribute) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) exclude_first (ksuit.schedules.base.progress_schedule.ProgressSchedule attribute) exclude_last (ksuit.schedules.base.progress_schedule.ProgressSchedule attribute) extra (attention.config.AttentionConfig.Config attribute) (attention.config.DotProductAttentionConfig.Config attribute) (attention.config.IrregularNatAttentionConfig.Config attribute) (attention.config.PerceiverAttentionConfig.Config attribute) (attention.config.TransolverAttentionConfig.Config attribute) (attention.config.TransolverPlusPlusAttentionConfig.Config attribute) (emmi_data_management.verification.VerificationResult attribute) extra_repr() (emmi.modules.layers.drop_path.UnquantizedDropPath method) extra_samplers (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute)

F

factor (ksuit.schedules.step.decreasing.StepDecreasingSchedule attribute) (ksuit.schedules.step.fixed.StepFixedSchedule attribute) (ksuit.schedules.step.interval.StepIntervalSchedule attribute) Factory (class in ksuit.factory.base.factory_base) FailAction (class in emmi_data_management.verification) fc1 (emmi.modules.mlp.upactdown_mlp.UpActDownMlp attribute) (emmi_inference.models.modules.mlp.Mlp attribute) fc2 (emmi.modules.mlp.upactdown_mlp.UpActDownMlp attribute) (emmi_inference.models.modules.mlp.Mlp attribute) feature_dim (default_tensor_precollator.DefaultTensorPreCollator attribute) (emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator attribute) fetch() (emmi_data_management.checkpoint.providers.file_provider.FileProvider method) (emmi_data_management.checkpoint.providers.huggingface_provider.HFProvider method) (emmi_data_management.checkpoint.types.Provider method) fetch_huggingface_by_extension() (in module emmi_data_management.interfaces.huggingface) fetch_huggingface_file() (in module emmi_data_management.interfaces.huggingface) fetch_huggingface_repo_snapshot() (in module emmi_data_management.interfaces.huggingface) fetch_s3_file() (in module emmi_data_management.interfaces.s3) fetch_s3_prefix() (in module emmi_data_management.interfaces.s3) field_slices (emmi.schemas.data.data_specs.FieldDimSpec property) FieldDecoderCollator (class in emmi.pipeline.collators.field_decoder) (class in emmi_inference.models.pipelines.field_decoder) FieldDimSpec (class in emmi.schemas.data.data_specs) FileProvider (class in emmi_data_management.checkpoint.providers.file_provider) FileRecord (class in emmi_data_management.verification) find_string() (ksuit.utils.training.training_iteration.TrainingIteration static method) finish() (ksuit.writers.log_writer.LogWriter method) FixedEarlyStopper (class in ksuit.callbacks.early_stoppers.fixed_early_stopper) FLOAT16_ALIASES (in module ksuit.utils.torch.amp) FLOAT32_ALIASES (in module ksuit.utils.torch.amp) float_to_integer_exact() (in module ksuit.utils.common.validation) float_to_scientific_notation() (in module ksuit.utils.logging.formatting) FloatOrArray (in module emmi.modules.preprocessors.normalizers.types) flush() (ksuit.writers.log_writer.LogWriter method) fmt (ksuit.utils.logging.formatting.CustomFormatter attribute) fmt_bytes() (in module emmi_data_management.cli_utils) (in module emmi_inference.cli_utils) format() (ksuit.utils.logging.formatting.CustomFormatter method) FORMATS (ksuit.utils.logging.formatting.CustomFormatter attribute) forward() (ab_upt.AnchoredBranchedUPT method) (emmi.modules.attention.anchor_attention.mixed_attention.MixedAttention method) (emmi.modules.attention.anchor_attention.multi_branch_anchor_attention.MultiBranchAnchorAttention method) (emmi.modules.attention.dot_product_attention.DotProductAttention method) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention method) (emmi.modules.attention.perceiver_attention.PerceiverAttention method) (emmi.modules.attention.transolver_attention.TransolverAttention method) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention method) (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock method) (emmi.modules.blocks.perceiver_block.PerceiverBlock method) (emmi.modules.blocks.perceiver_transformer_blockpair.PerceiverTransformerBlock method) (emmi.modules.blocks.transformer_block.TransformerBlock method) (emmi.modules.decoder.deep_perceiver_decoder.DeepPerceiverDecoder method) (emmi.modules.encoder.mlp_encoder.MlpEncoder method) (emmi.modules.encoder.supernode_pooling.SupernodePooling method) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly method) (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed method) (emmi.modules.layers.drop_path.UnquantizedDropPath method) (emmi.modules.layers.layer_scale.LayerScale method) (emmi.modules.layers.linear_projection.LinearProjection method) (emmi.modules.layers.rope_frequency.RopeFrequency method) (emmi.modules.layers.scalar_conditioner.ScalarsConditioner method) (emmi.modules.layers.transformer_batchnorm.TransformerBatchNorm method) (emmi.modules.mlp.mlp.MLP method) (emmi.modules.mlp.upactdown_mlp.UpActDownMlp method) (emmi_inference.models.abupt.AnchoredBranchedUPT method) (emmi_inference.models.modules.attention.anchor_attention.AnchorAttention method) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention method) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention method) (emmi_inference.models.modules.attention.sharedweights_crossattn_attention.SharedweightsCrossattnAttention method) (emmi_inference.models.modules.attention.sharedweights_splitattn_attention.SharedweightsSplitattnAttention method) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock method) (emmi_inference.models.modules.blocks.transformer_block.TransformerBlock method) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed method) (emmi_inference.models.modules.mlp.Mlp method) (emmi_inference.models.modules.rope_frequency.RopeFrequency method) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly method) (emmi_inference.models.transolver.Transolver method) (emmi_inference.models.upt.UPT method) (ksuit.distributed.functional.all_gather_grad_autograd.AllGatherGradAutograd static method) (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel method) (transformer.Transformer method) (upt.UPT method)

forward_properties (ksuit.trainers.base.BaseTrainer attribute) ForwardHook (class in ksuit.extractors.forward_hook) from_dict() (attention.anchor_attention.config.TokenSpec class method) from_filename() (ksuit.utils.training.training_iteration.TrainingIteration static method) from_included_excluded() (ksuit.data.wrappers.property_subset_wrapper.PropertySubsetWrapper class method) from_logscale() (in module emmi.functional.logscale) from_string() (ksuit.utils.training.training_iteration.TrainingIteration static method) from_uri() (ksuit.configs.static_config.StaticConfig static method) frozen_param_count (ksuit.models.model_base.ModelBase property) fs (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute)

G

gather_outputs() (emmi_inference.models.base_model.BaseModel method) GCP (emmi_data_management.providers.Provider attribute) GCP_URI_PREFIX (in module emmi_inference) GELU (emmi.modules.activations.Activation attribute) generate_fake_data() (in module ksuit.scripts.validate_pytorch_compatibilities) generate_run_id() (ksuit.providers.path_provider.PathProvider static method) GEOMETRY_BATCH_IDX (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) geometry_blocks (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) geometry_branch_forward() (ab_upt.AnchoredBranchedUPT method) geometry_depth (ab_upt_config.AnchorBranchedUPTConfig attribute) GEOMETRY_POS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) GEOMETRY_SUPERNODE_IDX (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) get() (emmi_inference.runner.InferencePayload method) get_all_callbacks() (ksuit.trainers.base.BaseTrainer method) get_all_metric_values() (ksuit.writers.log_writer.LogWriter method) get_backend() (in module ksuit.distributed.run.utils) get_bool_gather_supported() (in module ksuit.distributed.gather) get_case_reluris() (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) get_cli_command() (in module ksuit.utils.platform.system) get_credentials() (in module emmi_data_management.credentials) get_data_loader() (ksuit.data.samplers.interleaved_sampler.InterleavedSampler method) (ksuit.trainers.base.BaseTrainer method) (ksuit.utils.data.data_container.DataContainer method) get_dataset() (ksuit.utils.data.data_container.DataContainer method) get_default_callback_intervals() (ksuit.trainers.base.BaseTrainer method) get_default_callback_kwargs() (ksuit.trainers.base.BaseTrainer method) get_default_callbacks() (ksuit.trainers.base.BaseTrainer method) get_device_and_bfloat16supported() (in module ksuit.distributed.gather) get_fair_cpu_count() (in module ksuit.utils.platform.worker) get_grad_scaler_and_autocast_context() (in module ksuit.utils.torch.amp) get_installed_cuda_version() (in module ksuit.utils.platform.system) get_interval_string_verbose() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) get_key_by_value() (ksuit.utils.bidict.Bidict method) get_local_rank() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) get_main_sampler() (ksuit.utils.data.data_container.DataContainer method) get_managed_rank() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) get_managed_world_size() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) get_n_equal_properties() (ksuit.utils.training.training_iteration.TrainingIteration method) get_named_models() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) get_normal() (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) get_normalization_stats() (emmi.datasets.ahmedml.AhmedMLDataset static method) (emmi.datasets.drivaerml.DrivAerMLDataset static method) (emmi_inference.datasets.shapenet.ShapeNetCarDataset static method) get_num_nodes() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) get_ordinality() (ksuit.providers.metric_property_provider.MetricPropertyProvider static method)

get_properties() (ksuit.optimizer.param_group_modifiers.base.param_group_modifier_base.ParamGroupModifierBase method) (ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier.LrScaleByNameModifier method) (ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier.WeightDecayByNameModifier method) get_rank() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) get_s3_client() (in module emmi_data_management.interfaces.s3) get_sample_processor() (ksuit.data.pipeline.multistage.MultiStagePipeline method) get_sdf() (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) get_sequential_schedule_config() (ksuit.schedules.base.sequential_percent_schedule.SequentialPercentSchedule method) (ksuit.schedules.base.sequential_step_schedule.SequentialStepSchedule method) get_start_checkpoint() (ksuit.initializers.resume_initializer.ResumeInitializer method) get_supported_precision() (in module ksuit.utils.torch.amp) get_total_cpu_count() (in module ksuit.utils.platform.worker) get_trainer_callbacks() (ksuit.trainers.base.BaseTrainer method) get_value() (ksuit.schedules.base.schedule_base.ScheduleBase method) (ksuit.utils.training.schedule_wrapper.ScheduleWrapper method) get_value_by_key() (ksuit.utils.bidict.Bidict method) get_world_size() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) getitem_surface_normals() (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_surface_position() (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_surface_position_stl() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_surface_position_stl_resampled100k() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_surface_position_vtp() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_surface_pressure() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_surface_wallshearstress() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_volume_distance_to_surface() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_volume_normals() (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_volume_position() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_volume_sdf() (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_volume_totalpcoeff() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) getitem_volume_velocity() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) (emmi_inference.datasets.shapenet.ShapeNetCarDataset method) getitem_volume_vorticity() (emmi.datasets.ahmedml.AhmedMLDataset method) (emmi.datasets.drivaerml.DrivAerMLDataset method) global_dataset_paths (ksuit.configs.static_config.StaticConfig property) grad_scaler (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) GREY (ksuit.utils.logging.formatting.CustomFormatter attribute) grid_positions (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) grid_resolution (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) group_data_by_paths() (in module emmi.utils.aistore_utils)

H

handler (ksuit.callbacks.default_callbacks.eta.EtaCallback attribute) has_query_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator property) has_same_specified_properties() (ksuit.utils.training.training_iteration.TrainingIteration method) hash (emmi_data_management.verification.FileRecord attribute) HASH (emmi_data_management.verification.VerificationType attribute) hash_file() (in module emmi_data_management.verification) hash_mismatch (emmi_data_management.verification.VerificationResult attribute) HashType (in module emmi_data_management.verification) head_dim (attention.config.AttentionConfig attribute) (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention attribute) head_s3_object() (in module emmi_data_management.interfaces.s3) hf_app (in module emmi_data_management.cli_huggingface) hf_docs() (in module emmi_data_management.cli_huggingface) hf_estimate() (in module emmi_data_management.cli_huggingface) hf_ext() (in module emmi_data_management.cli_huggingface) hf_file() (in module emmi_data_management.cli_huggingface) hf_snapshot() (in module emmi_data_management.cli_huggingface) HFProvider (class in emmi_data_management.checkpoint.providers.huggingface_provider) HFRepoType (in module emmi_data_management.interfaces.huggingface) hidden_dim (ab_upt_config.AnchorBranchedUPTConfig attribute) (attention.config.AttentionConfig attribute) (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (layers.continuous_sincos_embedding_config.ContinuousSincosEmbeddingConfig attribute) (layers.layer_scale_config.LayerScaleConfig attribute) (layers.rope_frequenty_config.RopeFrequencyConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) (upt_config.UPTConfig attribute)

hidden_test (emmi.datasets.drivaerml.DrivAerMLDefaultSplitIDs attribute) high_tensor (emmi.pipeline.sample_processors.samplewise_normalization.SamplewiseNormalizationSampleProcessor attribute) higher_is_better (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) (ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper attribute) (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) HIGHER_IS_BETTER (ksuit.providers.metric_property_provider.Ordinality attribute) higher_is_better() (ksuit.providers.metric_property_provider.MetricPropertyProvider method) hook_outputs (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) host (ksuit.configs.wandb_config.WandBConfig attribute), [1] HUGGINGFACE (emmi_data_management.providers.Provider attribute) HuggingFaceRepoType (class in emmi_data_management.cli_huggingface) HydraRunner (class in ksuit.runners.hydra_runner) Hyperparameters (class in ksuit.configs.hyperparameters)

I

i (in module ksuit.cli.main_calculate_stats) (in module ksuit.cli.main_train) implementation (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (layers.rope_frequenty_config.RopeFrequencyConfig attribute) impls (in module emmi_data_management.diskcache.cache_benchmark) in_project_fx (emmi.modules.attention.transolver_attention.TransolverAttention attribute) in_project_slice (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) in_project_x (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) IncreasingProgressSchedule (class in ksuit.schedules.base.increasing_progress_schedule) index_offsets (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) indices (ksuit.data.base.subset.Subset attribute) INFERENCE_ENGINE_VERSION (in module emmi_inference.runner) InferencePayload (class in emmi_inference.runner) InferenceRunner (class in emmi_inference.runner) InferenceStatus (in module emmi_inference.runner) init() (ksuit.trackers.base_tracker.BaseTracker method) init_bias_to_zero() (in module emmi.functional.init) init_callbacks() (ksuit.initializers.resume_initializer.ResumeInitializer method) init_optimizer() (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer method) (ksuit.initializers.base.initializer_base.InitializerBase method) (ksuit.initializers.resume_initializer.ResumeInitializer method) init_run_path_provider (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) init_trainer() (ksuit.initializers.resume_initializer.ResumeInitializer method) init_trunc_normal_zero_bias() (in module emmi.functional.init) init_values (layers.layer_scale_config.LayerScaleConfig attribute) init_weights (ab_upt_config.AnchorBranchedUPTConfig attribute) (attention.config.AttentionConfig attribute) (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi.modules.layers.linear_projection.LinearProjection attribute) (emmi.modules.mlp.mlp.MLP attribute) (emmi.modules.mlp.upactdown_mlp.UpActDownMlp attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (layers.linear_projection_config.LinearProjectionConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute) init_weights() (ksuit.initializers.base.initializer_base.InitializerBase method) (ksuit.initializers.previous_run_initializer.PreviousRunInitializer method) (ksuit.initializers.resume_initializer.ResumeInitializer method) initialize() (ksuit.models.model_base.ModelBase method) initialize_optimizer() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) initialize_weights() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) initialize_worker() (in module emmi_data_management.diskcache.cache_benchmark) initializer (ksuit.trainers.base.BaseTrainer attribute) InitializerBase (class in ksuit.initializers.base.initializer_base) initializers (ksuit.models.model_base.ModelBase attribute) InitWeightsMode (in module emmi.types) input_dim (attention.config.IrregularNatAttentionConfig attribute) (decoder.deep_perceiver_decoder.DeepPerceiverDecoderConfig attribute) (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.base_model.BaseModel attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (layers.continuous_sincos_embedding_config.ContinuousSincosEmbeddingConfig attribute) (layers.linear_projection_config.LinearProjectionConfig attribute) (layers.rope_frequenty_config.RopeFrequencyConfig attribute) (mlp.mlp.MLPConfig attribute) (mlp.upactdown_mlp.UpActDownMLPConfig attribute)

INPUT_POS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) inputs (emmi_inference.runner.InferencePayload attribute) instantiate() (ksuit.factory.base.factory_base.Factory method) (ksuit.factory.dataset_factory.DatasetFactory method) (ksuit.factory.optim_factory.OptimizerFactory method) InterleavedSampler (class in ksuit.data.samplers.interleaved_sampler) InterleavedSamplerConfig (class in ksuit.data.samplers.interleaved_sampler) interval (ksuit.utils.training.schedule_wrapper.ScheduleWrapper attribute) inverse() (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor method) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor method) (emmi.pipeline.sample_processors.samplewise_normalization.SamplewiseNormalizationSampleProcessor method) (ksuit.data.pipeline.sample_processor.SampleProcessor method) (ksuit.data.preprocessors.compose.ComposePreProcess method) IrregularNatAttention (class in emmi.modules.attention.irregular_nat_attention) IrregularNatAttentionConfig (class in attention.config) IrregularNatAttentionConfig.Config (class in attention.config) IrregularNatBlock (class in emmi.modules.blocks.irregular_nat_block) is_bfloat16_compatible() (in module ksuit.utils.torch.amp) is_compatible() (in module ksuit.utils.torch.amp) is_data_rank0() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) is_disabled (ksuit.configs.wandb_config.WandBConfig property) is_distributed() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig static method) is_finished (ksuit.utils.training.counter.UpdateCounter property) is_float16_compatible() (in module ksuit.utils.torch.amp) is_frozen (ksuit.models.composite.CompositeModel property) (ksuit.models.model_base.ModelBase property) (ksuit.models.single.Model property) is_full_epoch (ksuit.utils.training.counter.UpdateCounter property) is_fully_specified (ksuit.utils.training.training_iteration.TrainingIteration property) is_initialized (ksuit.models.model_base.ModelBase attribute) is_local_rank0() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) is_managed() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig static method) is_minimally_specified (ksuit.utils.training.training_iteration.TrainingIteration property) is_neutral_key() (ksuit.providers.metric_property_provider.MetricPropertyProvider method) is_offline (ksuit.configs.wandb_config.WandBConfig property) is_online (ksuit.configs.wandb_config.WandBConfig property) is_rank0() (in module ksuit.distributed.config) (ksuit.distributed.config.distributed_config.DistributedConfig method) is_using_ais (emmi.datasets.ahmedml.AhmedMLDataset attribute) (emmi.datasets.drivaerml.DrivAerMLDataset attribute) item (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) (emmi.pipeline.sample_processors.samplewise_normalization.SamplewiseNormalizationSampleProcessor attribute) (emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor attribute) (emmi_inference.models.pipelines.sparse_tensor_offset.SparseTensorOffsetCollator attribute) (emmi_inference.models.pipelines.surface_mask_precollator.SurfaceMaskPreCollator attribute) (surface_mask_precollator.SurfaceMaskPreCollator attribute) item_key_name (default_tensor_precollator.DefaultTensorPreCollator attribute) (emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator attribute) items (concat_tensor_precollator.ConcatTensorPreCollator attribute) (emmi.pipeline.batch_processors.moment_normalization.MomentNormalizationBatchProcessor attribute) (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor attribute) (emmi.pipeline.collators.concat_sparse_tensor.ConcatSparseTensorCollator attribute) (emmi.pipeline.collators.default.DefaultCollator attribute) (emmi.pipeline.sample_processors.point_sampling.PointSamplingSampleProcessor attribute) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor attribute) (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) (emmi_inference.models.pipelines.concat_tensor_precollator.ConcatTensorPreCollator attribute) items() (emmi.schemas.data.data_specs.FieldDimSpec method) items_at_supernodes (emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor attribute) iter_s3_object_chunks() (in module emmi_data_management.interfaces.s3) iterable (ksuit.utils.logging.progress.NoopTqdm attribute)

J

JointAnchorAttention (class in emmi.modules.attention.anchor_attention.joint_anchor_attention)

JointAnchorAttentionConfig (class in attention.anchor_attention.config)

K

k (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) keep_prob (emmi.modules.layers.drop_path.UnquantizedDropPath property) keep_queries (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) kept (emmi_inference.runner.InferencePayload attribute) key (emmi_data_management.interfaces.s3.S3Object attribute) (ksuit.extractors.forward_hook.ForwardHook attribute) key_map (emmi.pipeline.batch_processors.rename_keys.RenameKeysBatchProcessor attribute) (emmi.pipeline.sample_processors.duplicate_keys.DuplicateKeysSampleProcessor attribute) (emmi.pipeline.sample_processors.rename_keys.RenameKeysSampleProcessor attribute) key_value_tokens (attention.anchor_attention.config.AttentionPattern attribute) keys (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) keys() (emmi.schemas.data.data_specs.FieldDimSpec method) keys_to_remove (ksuit.initializers.previous_run_initializer.PreviousRunInitializer attribute) ksuit module ksuit.callbacks module ksuit.callbacks.base module ksuit.callbacks.base.callback_base module ksuit.callbacks.base.periodic_callback module ksuit.callbacks.base.periodic_iterator_callback module ksuit.callbacks.checkpoint_callbacks module ksuit.callbacks.checkpoint_callbacks.best_checkpoint module ksuit.callbacks.checkpoint_callbacks.checkpoint module ksuit.callbacks.checkpoint_callbacks.ema module ksuit.callbacks.default_callbacks module ksuit.callbacks.default_callbacks.dataset_stats module ksuit.callbacks.default_callbacks.eta module ksuit.callbacks.default_callbacks.lr module ksuit.callbacks.default_callbacks.online_loss module ksuit.callbacks.default_callbacks.param_count module ksuit.callbacks.default_callbacks.peak_memory module ksuit.callbacks.default_callbacks.progress module ksuit.callbacks.default_callbacks.train_time module ksuit.callbacks.early_stoppers module ksuit.callbacks.early_stoppers.early_stopper_base module ksuit.callbacks.early_stoppers.fixed_early_stopper module ksuit.callbacks.early_stoppers.metric_early_stopper module ksuit.callbacks.offline_callbacks module ksuit.callbacks.offline_callbacks.offline_loss module ksuit.callbacks.online_callbacks module ksuit.callbacks.online_callbacks.best_metric module ksuit.callbacks.online_callbacks.update_output module ksuit.cli module ksuit.cli.main_calculate_stats module ksuit.cli.main_train module ksuit.configs module ksuit.configs.hyperparameters module ksuit.configs.static_config module ksuit.configs.wandb_config module ksuit.data module ksuit.data.base module ksuit.data.base.dataset module ksuit.data.base.subset module ksuit.data.base.wrapper module ksuit.data.pipeline module ksuit.data.pipeline.batch_processor module ksuit.data.pipeline.collator module ksuit.data.pipeline.multistage module ksuit.data.pipeline.sample_processor module ksuit.data.preprocessors module ksuit.data.preprocessors.compose module ksuit.data.preprocessors.preprocessor module ksuit.data.preprocessors.types module ksuit.data.preprocessors.utils module ksuit.data.samplers module ksuit.data.samplers.interleaved_sampler module ksuit.data.samplers.internals module ksuit.data.wrappers module ksuit.data.wrappers.dataset_wrappers module ksuit.data.wrappers.dataset_wrappers.repeat_wrapper module ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper module ksuit.data.wrappers.dataset_wrappers.subset_wrapper module ksuit.data.wrappers.property_subset_wrapper module ksuit.distributed module ksuit.distributed.config module ksuit.distributed.config.distributed_config module ksuit.distributed.functional module ksuit.distributed.functional.all_gather_grad_autograd module ksuit.distributed.gather module ksuit.distributed.run module ksuit.distributed.run.managed module ksuit.distributed.run.unmanaged module ksuit.distributed.run.utils module ksuit.extractors module ksuit.extractors.forward_hook module ksuit.factory module ksuit.factory.base module ksuit.factory.base.factory_base module ksuit.factory.base.reflection_utils module ksuit.factory.dataset_factory module ksuit.factory.optim_factory module ksuit.factory.schedule_factory module ksuit.initializers module ksuit.initializers.base module ksuit.initializers.base.checkpoint_initializer module ksuit.initializers.base.initializer_base module

ksuit.initializers.previous_run_initializer module ksuit.initializers.resume_initializer module ksuit.models module ksuit.models.composite module ksuit.models.model_base module ksuit.models.single module ksuit.optimizer module ksuit.optimizer.optimizer_wrapper module ksuit.optimizer.param_group_modifiers module ksuit.optimizer.param_group_modifiers.base module ksuit.optimizer.param_group_modifiers.base.param_group_modifier_base module ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier module ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier module ksuit.providers module ksuit.providers.dataset_config_provider module ksuit.providers.metric_property_provider module ksuit.providers.path_provider module ksuit.runners module ksuit.runners.hydra_runner module ksuit.runners.stats_runner module ksuit.schedules module ksuit.schedules.base module ksuit.schedules.base.decreasing_progress_schedule module ksuit.schedules.base.increasing_progress_schedule module ksuit.schedules.base.progress_schedule module ksuit.schedules.base.schedule_base module ksuit.schedules.base.sequential_percent_schedule module ksuit.schedules.base.sequential_step_schedule module ksuit.schedules.constant_schedule module ksuit.schedules.cosine module ksuit.schedules.cosine.decreasing module ksuit.schedules.cosine.increasing module ksuit.schedules.custom_schedule module ksuit.schedules.functional module ksuit.schedules.linear module ksuit.schedules.linear.decreasing module ksuit.schedules.linear.increasing module ksuit.schedules.linear_warmup_cosine_decay module ksuit.schedules.polynomial module ksuit.schedules.polynomial.decreasing module ksuit.schedules.polynomial.increasing module ksuit.schedules.step module ksuit.schedules.step.decreasing module ksuit.schedules.step.fixed module ksuit.schedules.step.interval module ksuit.scripts module ksuit.scripts.validate_pytorch_compatibilities module ksuit.setup module ksuit.trackers module ksuit.trackers.base_tracker module ksuit.trackers.noop_tracker module ksuit.trackers.wandb_tracker module ksuit.trainers module ksuit.trainers.base module ksuit.trainers.types module ksuit.utils module ksuit.utils.bidict module ksuit.utils.code module ksuit.utils.common module ksuit.utils.common.naming module ksuit.utils.common.path module ksuit.utils.common.stopwatch module ksuit.utils.common.typing module ksuit.utils.common.validation module ksuit.utils.data module ksuit.utils.data.data_container module ksuit.utils.logging module ksuit.utils.logging.core module ksuit.utils.logging.formatting module ksuit.utils.logging.progress module ksuit.utils.model module ksuit.utils.model.helpers module ksuit.utils.model.model_utils module ksuit.utils.platform module ksuit.utils.platform.system module ksuit.utils.platform.worker module ksuit.utils.torch module ksuit.utils.torch.amp module ksuit.utils.torch.device module ksuit.utils.training module ksuit.utils.training.counter module ksuit.utils.training.running_stats module ksuit.utils.training.schedule_wrapper module ksuit.utils.training.seed module ksuit.utils.training.training_iteration module ksuit.writers module ksuit.writers.checkpoint_writer module ksuit.writers.log_writer module KT (in module ksuit.utils.bidict) kv (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention attribute) kv_dim (attention.config.PerceiverAttentionConfig attribute) (blocks.perceiver_block_config.PerceiverBlockConfig attribute) kwarg_forward() (emmi_inference.models.transolver.Transolver method)

L

label (emmi_data_management.logging_progress.LogProgress attribute) lap() (ksuit.utils.common.stopwatch.Stopwatch method) lap_count (ksuit.utils.common.stopwatch.Stopwatch property) last_lap_time (ksuit.utils.common.stopwatch.Stopwatch property) layer (emmi.modules.encoder.mlp_encoder.MlpEncoder attribute) layers module layers.continuous_sincos_embedding_config module layers.drop_path_config module layers.layer_scale_config module layers.linear_projection_config module layers.rope_frequenty_config module layerscale (blocks.transformer_block_config.TransformerBlockConfig attribute) LayerScale (class in emmi.modules.layers.layer_scale) LayerScaleConfig (class in layers.layer_scale_config) LEAKY_RELU (emmi.modules.activations.Activation attribute) lin1 (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) lin2 (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) lin_out (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) linear() (in module ksuit.schedules.functional) LinearDecreasingSchedule (class in ksuit.schedules.linear.decreasing) LinearIncreasingSchedule (class in ksuit.schedules.linear.increasing) LinearProjection (class in emmi.modules.layers.linear_projection) LinearProjectionConfig (class in layers.linear_projection_config) LinearWarmupCosineDecaySchedule (class in ksuit.schedules.linear_warmup_cosine_decay) link() (ksuit.providers.path_provider.PathProvider method) Lion (class in emmi.optim.lion) lion() (in module emmi.optim.lion) list_s3_objects() (in module emmi_data_management.interfaces.s3) load_components() (in module emmi_inference.api) load_manifest() (in module emmi_data_management.verification) load_model() (in module emmi_inference.models.loader) load_optim (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) load_state_dict() (ksuit.callbacks.base.callback_base.CallbackBase method) (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback method) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper method) (ksuit.trainers.base.BaseTrainer method) load_unstructured_grid_data() (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) load_yaml() (in module emmi_inference.api) local (in module emmi_data_management.diskcache.cache_benchmark) local_dataset_path (ksuit.configs.static_config.StaticConfig property) (ksuit.providers.dataset_config_provider.DatasetConfigProvider property) local_file (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) local_fs() (in module emmi_data_management.diskcache.cache_benchmark) local_path (emmi_data_management.checkpoint.types.CheckpointMetadata attribute) local_root (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute), [1] local_spool_dir (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) local_writer (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) LOCK_FILE_MODE (in module emmi_data_management.diskcache.lru_cache), [1] LOCK_FILE_SUFFIX (in module emmi_data_management.diskcache.lru_cache), [1] log() (ksuit.configs.hyperparameters.Hyperparameters static method) (ksuit.configs.static_config.StaticConfig method) (ksuit.trackers.base_tracker.BaseTracker method) (ksuit.utils.logging.core.MessageCounter method)

log_cache (ksuit.writers.log_writer.LogWriter attribute) log_device_info() (in module ksuit.distributed.run.utils) log_entries (ksuit.writers.log_writer.LogWriter attribute) log_from_all_ranks() (in module ksuit.utils.logging.core) log_output (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) log_system_info() (in module ksuit.utils.platform.system) log_writer (ksuit.trainers.base.BaseTrainer attribute) logfile_uri (ksuit.providers.path_provider.PathProvider property) logger (in module emmi_data_management.diskcache.lru_cache) (in module ksuit.distributed.run.managed) (in module ksuit.distributed.run.unmanaged) (in module ksuit.distributed.run.utils) (in module ksuit.factory.base.reflection_utils) (in module ksuit.runners.hydra_runner) (in module ksuit.runners.stats_runner) (in module ksuit.utils.platform.system) (in module ksuit.utils.platform.worker) (in module ksuit.utils.torch.amp) (in module ksuit.utils.training.seed) (ksuit.callbacks.base.callback_base.CallbackBase property) (ksuit.data.base.dataset.Dataset attribute) (ksuit.factory.base.factory_base.Factory attribute) (ksuit.initializers.base.initializer_base.InitializerBase attribute) (ksuit.models.model_base.ModelBase attribute) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) (ksuit.providers.dataset_config_provider.DatasetConfigProvider attribute) (ksuit.providers.metric_property_provider.MetricPropertyProvider attribute) (ksuit.trackers.base_tracker.BaseTracker attribute) (ksuit.trainers.base.BaseTrainer attribute) (ksuit.utils.data.data_container.DataContainer attribute) (ksuit.writers.checkpoint_writer.CheckpointWriter attribute) (ksuit.writers.log_writer.LogWriter attribute) logmean (ksuit.utils.training.running_stats.RunningStats property) logmean_tensor (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) LogProgress (class in emmi_data_management.logging_progress) logscale (emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer.ShiftAndScaleNormalizer attribute) (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) logstd (ksuit.utils.training.running_stats.RunningStats property) logstd_tensor (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) logvar (ksuit.utils.training.running_stats.RunningStats property) LogWriter (class in ksuit.writers.log_writer) loss_compute() (ksuit.trainers.base.BaseTrainer method) losses_to_log (ksuit.trainers.types.TrainerResult attribute) LossResult (in module ksuit.trainers.types) LOWER_IS_BETTER (ksuit.providers.metric_property_provider.Ordinality attribute) lower_type_name() (in module ksuit.utils.common.naming) LrCallback (class in ksuit.callbacks.default_callbacks.lr) LrScaleByNameModifier (class in ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier) LRUCacheFileSystem (class in emmi_data_management.diskcache.lru_cache) ls1 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) ls2 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute)

M

main() (in module emmi.preprocessing.calculate_statistics_all) (in module emmi.preprocessing.calculate_statistics_train) (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) (in module emmi_data_management.cli) (in module emmi_inference.cli) (in module emmi_inference.models.abupt) (in module ksuit.cli.main_calculate_stats) (in module ksuit.cli.main_train) (in module ksuit.scripts.validate_pytorch_compatibilities) (ksuit.runners.hydra_runner.HydraRunner static method) main_sampler (ksuit.data.samplers.interleaved_sampler.InterleavedSampler attribute) main_sampler_kwargs (ksuit.trainers.base.BaseTrainer attribute) Manifest (in module emmi_data_management.verification) MASTER_PORT (in module ksuit.configs.static_config) master_port (ksuit.configs.static_config.StaticConfig property) matching_item_key (default_tensor_precollator.DefaultTensorPreCollator attribute) (emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator attribute) max (ksuit.utils.training.running_stats.RunningStats property) max_degree (attention.config.IrregularNatAttentionConfig attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) max_epochs (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) max_inflight (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute) (emmi_inference.writer.AsyncWriter attribute) max_quantile (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) max_samples (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) max_updates (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) max_value (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) max_wavelength (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) (layers.continuous_sincos_embedding_config.ContinuousSincosEmbeddingConfig attribute) (layers.rope_frequenty_config.RopeFrequencyConfig attribute) maybe_autocast() (in module emmi_inference.models.modules.rope_frequency) mean (emmi.modules.preprocessors.normalizers.mean_std_normalization.MeanStdNormalization attribute) (emmi_inference.preprocess.NormalizeAny attribute), [1] (ksuit.utils.training.running_stats.RunningStats property) mean() (emmi.utils.running_moments.RunningMoments method) mean_tensor (emmi.pipeline.batch_processors.moment_normalization.MomentNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) MeanStdNormalization (class in emmi.modules.preprocessors.normalizers.mean_std_normalization) merge_vtu_parts() (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) mesh_item (emmi_inference.models.pipelines.mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) mesh_to_grid_edges_precollator module MeshToGridEdgesPreCollator (class in emmi_inference.models.pipelines.mesh_to_grid_edges_precollator) (class in mesh_to_grid_edges_precollator) message (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) message_mode (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) MessageCounter (class in ksuit.utils.logging.core) meta (emmi_inference.runner.InferencePayload attribute) metric_at_exceeded_tolerance (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) metric_key (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) (ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper attribute) metric_property_provider (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.trackers.base_tracker.BaseTracker attribute) (ksuit.trainers.base.BaseTrainer attribute) MetricEarlyStopper (class in ksuit.callbacks.early_stoppers.metric_early_stopper) MetricPropertyProvider (class in ksuit.providers.metric_property_provider) min (ksuit.utils.training.running_stats.RunningStats property) min_level (ksuit.utils.logging.core.MessageCounter attribute) min_quantile (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) min_value (emmi.pipeline.sample_processors.drop_outliers.DropOutliersSampleProcessor attribute) missing (emmi_data_management.verification.VerificationResult attribute) MissingAnchorTokenError MissingBranchTokensError mixed_attention (emmi.modules.attention.anchor_attention.multi_branch_anchor_attention.MultiBranchAnchorAttention attribute) MixedAttention (class in emmi.modules.attention.anchor_attention.mixed_attention) MixedAttentionConfig (class in attention.anchor_attention.config) mlp module MLP (class in emmi.modules.mlp.mlp) Mlp (class in emmi_inference.models.modules.mlp) mlp (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) (emmi.modules.mlp.mlp.MLP attribute) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.transformer_block.TransformerBlock attribute) mlp.mlp module mlp.upactdown_mlp module mlp_expansion_factor (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi_inference.models.transolver.Transolver attribute) (transformer_config.TransformerConfig attribute) (upt_config.UPTConfig attribute) mlp_hidden_dim (blocks.transformer_block_config.TransformerBlockConfig attribute) MLPConfig (class in mlp.mlp) MlpEncoder (class in emmi.modules.encoder.mlp_encoder) mlps (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) mode (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (ksuit.configs.wandb_config.WandBConfig attribute), [1] (ksuit.trackers.wandb_tracker.WandbTracker attribute) Model (class in ksuit.models.single) MODEL (emmi_data_management.cli_huggingface.HuggingFaceRepoType attribute) model (emmi_inference.runner.InferenceRunner attribute) (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) model_config (ksuit.models.model_base.ModelBase attribute) (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) model_info (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) model_name (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) model_names (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) (ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback attribute) model_paths (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) ModelBase (class in ksuit.models.model_base) ModelFactory (in module emmi_inference.models.registry) ModelRegistry (in module emmi_inference.models.registry) MODES (ksuit.trackers.wandb_tracker.WandbTracker attribute) modulate_gate() (in module emmi.functional.modulation) modulate_scale_shift() (in module emmi.functional.modulation) module ab_upt ab_upt_config abupt attention attention.anchor_attention attention.anchor_attention.config attention.config blocks blocks.perceiver_block_config blocks.transformer_block_config concat_tensor_precollator decoder decoder.deep_perceiver_decoder default_tensor_precollator emmi emmi.callbacks emmi.callbacks.supernode_pooling_coverage emmi.datasets emmi.datasets.ahmedml emmi.datasets.drivaerml emmi.functional emmi.functional.init emmi.functional.logscale emmi.functional.modulation emmi.functional.rope emmi.models emmi.modules emmi.modules.activations emmi.modules.attention emmi.modules.attention.anchor_attention emmi.modules.attention.anchor_attention.cross_anchor_attention emmi.modules.attention.anchor_attention.joint_anchor_attention emmi.modules.attention.anchor_attention.mixed_attention emmi.modules.attention.anchor_attention.multi_branch_anchor_attention emmi.modules.attention.anchor_attention.self_anchor_attention emmi.modules.attention.dot_product_attention emmi.modules.attention.irregular_nat_attention emmi.modules.attention.perceiver_attention emmi.modules.attention.transolver_attention emmi.modules.attention.transolver_plusplus_attention emmi.modules.blocks emmi.modules.blocks.irregular_nat_block emmi.modules.blocks.perceiver_block emmi.modules.blocks.perceiver_transformer_blockpair emmi.modules.blocks.transformer_block emmi.modules.decoder emmi.modules.decoder.deep_perceiver_decoder emmi.modules.encoder emmi.modules.encoder.mlp_encoder emmi.modules.encoder.supernode_pooling emmi.modules.encoder.supernode_pooling_posonly emmi.modules.layers emmi.modules.layers.continuous_sincos_embed emmi.modules.layers.drop_path emmi.modules.layers.layer_scale emmi.modules.layers.linear_projection emmi.modules.layers.rope_frequency emmi.modules.layers.scalar_conditioner emmi.modules.layers.transformer_batchnorm emmi.modules.mlp emmi.modules.mlp.mlp emmi.modules.mlp.upactdown_mlp emmi.modules.preprocessors emmi.modules.preprocessors.normalizers emmi.modules.preprocessors.normalizers.mean_std_normalization emmi.modules.preprocessors.normalizers.position_normalizer emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer emmi.modules.preprocessors.normalizers.types emmi.modules.preprocessors.numpy_to_torch_tensor emmi.optim emmi.optim.lion emmi.pipeline emmi.pipeline.batch_processors emmi.pipeline.batch_processors.moment_normalization emmi.pipeline.batch_processors.position_normalization emmi.pipeline.batch_processors.rename_keys emmi.pipeline.collators emmi.pipeline.collators.concat_sparse_tensor emmi.pipeline.collators.default emmi.pipeline.collators.field_decoder emmi.pipeline.sample_processors emmi.pipeline.sample_processors.drop_outliers emmi.pipeline.sample_processors.duplicate_keys emmi.pipeline.sample_processors.moment_normalization emmi.pipeline.sample_processors.point_sampling emmi.pipeline.sample_processors.position_normalization emmi.pipeline.sample_processors.rename_keys emmi.pipeline.sample_processors.replace_key emmi.pipeline.sample_processors.samplewise_normalization emmi.pipeline.sample_processors.supernode_sampling emmi.preprocessing emmi.preprocessing.calculate_statistics_all emmi.preprocessing.calculate_statistics_train emmi.preprocessing.process_drivaer_and_ahmed_volumetric emmi.preprocessing.shapenet_car emmi.preprocessing.shapenet_car.preprocess_shapenet_car emmi.schemas emmi.schemas.data emmi.schemas.data.data_specs emmi.schemas.models emmi.setup emmi.types emmi.utils emmi.utils.aistore_utils emmi.utils.point_cloud_utils emmi.utils.running_moments emmi_data_management emmi_data_management.checkpoint emmi_data_management.checkpoint.iohash emmi_data_management.checkpoint.providers emmi_data_management.checkpoint.providers.file_provider emmi_data_management.checkpoint.providers.huggingface_provider emmi_data_management.checkpoint.resolver emmi_data_management.checkpoint.types emmi_data_management.cli emmi_data_management.cli_aws emmi_data_management.cli_checkpoint emmi_data_management.cli_huggingface emmi_data_management.cli_utils emmi_data_management.cli_verification emmi_data_management.credentials emmi_data_management.diskcache emmi_data_management.diskcache.cache_benchmark emmi_data_management.diskcache.lru_cache emmi_data_management.interfaces emmi_data_management.interfaces.azure emmi_data_management.interfaces.gcp emmi_data_management.interfaces.huggingface emmi_data_management.interfaces.s3 emmi_data_management.interfaces.s3_uploader emmi_data_management.logging_config emmi_data_management.logging_progress emmi_data_management.providers emmi_data_management.verification emmi_inference emmi_inference.api emmi_inference.cli emmi_inference.cli_utils emmi_inference.collator_registry emmi_inference.common emmi_inference.dataset_registry emmi_inference.datasets emmi_inference.datasets.shapenet emmi_inference.device emmi_inference.examples emmi_inference.examples.run_abupt emmi_inference.examples.run_transolver emmi_inference.examples.run_upt emmi_inference.logging_config emmi_inference.models emmi_inference.models.abupt emmi_inference.models.base_model emmi_inference.models.loader emmi_inference.models.modules emmi_inference.models.modules.attention emmi_inference.models.modules.attention.anchor_attention emmi_inference.models.modules.attention.dot_product_attention emmi_inference.models.modules.attention.perceiver_attention emmi_inference.models.modules.attention.sharedweights_crossattn_attention emmi_inference.models.modules.attention.sharedweights_splitattn_attention emmi_inference.models.modules.blocks emmi_inference.models.modules.blocks.perceiver_block emmi_inference.models.modules.blocks.transformer_block emmi_inference.models.modules.continuous_sincos_embed emmi_inference.models.modules.mlp emmi_inference.models.modules.rope emmi_inference.models.modules.rope_frequency emmi_inference.models.modules.supernode_pooling_posonly emmi_inference.models.pipelines emmi_inference.models.pipelines.abupt emmi_inference.models.pipelines.anchor_point_sampling_precollator emmi_inference.models.pipelines.concat_tensor_precollator emmi_inference.models.pipelines.default_tensor_precollator emmi_inference.models.pipelines.field_decoder emmi_inference.models.pipelines.mesh_to_grid_edges_precollator emmi_inference.models.pipelines.query_to_grid_edges_precollator emmi_inference.models.pipelines.shapenet_multistage emmi_inference.models.pipelines.sparse_tensor_offset emmi_inference.models.pipelines.surface_mask_precollator emmi_inference.models.registry emmi_inference.models.transolver emmi_inference.models.upt emmi_inference.preprocess emmi_inference.runner emmi_inference.s3_spooling_uploader emmi_inference.writer encoder encoder.supernode_pooling_config ksuit ksuit.callbacks ksuit.callbacks.base ksuit.callbacks.base.callback_base ksuit.callbacks.base.periodic_callback ksuit.callbacks.base.periodic_iterator_callback ksuit.callbacks.checkpoint_callbacks ksuit.callbacks.checkpoint_callbacks.best_checkpoint ksuit.callbacks.checkpoint_callbacks.checkpoint ksuit.callbacks.checkpoint_callbacks.ema ksuit.callbacks.default_callbacks ksuit.callbacks.default_callbacks.dataset_stats ksuit.callbacks.default_callbacks.eta ksuit.callbacks.default_callbacks.lr ksuit.callbacks.default_callbacks.online_loss ksuit.callbacks.default_callbacks.param_count ksuit.callbacks.default_callbacks.peak_memory ksuit.callbacks.default_callbacks.progress ksuit.callbacks.default_callbacks.train_time ksuit.callbacks.early_stoppers ksuit.callbacks.early_stoppers.early_stopper_base ksuit.callbacks.early_stoppers.fixed_early_stopper ksuit.callbacks.early_stoppers.metric_early_stopper ksuit.callbacks.offline_callbacks ksuit.callbacks.offline_callbacks.offline_loss ksuit.callbacks.online_callbacks ksuit.callbacks.online_callbacks.best_metric ksuit.callbacks.online_callbacks.update_output ksuit.cli ksuit.cli.main_calculate_stats ksuit.cli.main_train ksuit.configs ksuit.configs.hyperparameters ksuit.configs.static_config ksuit.configs.wandb_config ksuit.data ksuit.data.base ksuit.data.base.dataset ksuit.data.base.subset ksuit.data.base.wrapper ksuit.data.pipeline ksuit.data.pipeline.batch_processor ksuit.data.pipeline.collator ksuit.data.pipeline.multistage ksuit.data.pipeline.sample_processor ksuit.data.preprocessors ksuit.data.preprocessors.compose ksuit.data.preprocessors.preprocessor ksuit.data.preprocessors.types ksuit.data.preprocessors.utils ksuit.data.samplers ksuit.data.samplers.interleaved_sampler ksuit.data.samplers.internals ksuit.data.wrappers ksuit.data.wrappers.dataset_wrappers ksuit.data.wrappers.dataset_wrappers.repeat_wrapper ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper ksuit.data.wrappers.dataset_wrappers.subset_wrapper ksuit.data.wrappers.property_subset_wrapper ksuit.distributed ksuit.distributed.config ksuit.distributed.config.distributed_config ksuit.distributed.functional ksuit.distributed.functional.all_gather_grad_autograd ksuit.distributed.gather ksuit.distributed.run ksuit.distributed.run.managed ksuit.distributed.run.unmanaged ksuit.distributed.run.utils ksuit.extractors ksuit.extractors.forward_hook ksuit.factory ksuit.factory.base ksuit.factory.base.factory_base ksuit.factory.base.reflection_utils ksuit.factory.dataset_factory ksuit.factory.optim_factory ksuit.factory.schedule_factory ksuit.initializers ksuit.initializers.base ksuit.initializers.base.checkpoint_initializer ksuit.initializers.base.initializer_base ksuit.initializers.previous_run_initializer ksuit.initializers.resume_initializer ksuit.models ksuit.models.composite ksuit.models.model_base ksuit.models.single ksuit.optimizer ksuit.optimizer.optimizer_wrapper ksuit.optimizer.param_group_modifiers ksuit.optimizer.param_group_modifiers.base ksuit.optimizer.param_group_modifiers.base.param_group_modifier_base ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier ksuit.providers ksuit.providers.dataset_config_provider ksuit.providers.metric_property_provider ksuit.providers.path_provider ksuit.runners ksuit.runners.hydra_runner ksuit.runners.stats_runner ksuit.schedules ksuit.schedules.base ksuit.schedules.base.decreasing_progress_schedule ksuit.schedules.base.increasing_progress_schedule ksuit.schedules.base.progress_schedule ksuit.schedules.base.schedule_base ksuit.schedules.base.sequential_percent_schedule ksuit.schedules.base.sequential_step_schedule ksuit.schedules.constant_schedule ksuit.schedules.cosine ksuit.schedules.cosine.decreasing ksuit.schedules.cosine.increasing ksuit.schedules.custom_schedule ksuit.schedules.functional ksuit.schedules.linear ksuit.schedules.linear.decreasing ksuit.schedules.linear.increasing ksuit.schedules.linear_warmup_cosine_decay ksuit.schedules.polynomial ksuit.schedules.polynomial.decreasing ksuit.schedules.polynomial.increasing ksuit.schedules.step ksuit.schedules.step.decreasing ksuit.schedules.step.fixed ksuit.schedules.step.interval ksuit.scripts ksuit.scripts.validate_pytorch_compatibilities ksuit.setup ksuit.trackers ksuit.trackers.base_tracker ksuit.trackers.noop_tracker ksuit.trackers.wandb_tracker ksuit.trainers ksuit.trainers.base ksuit.trainers.types ksuit.utils ksuit.utils.bidict ksuit.utils.code ksuit.utils.common ksuit.utils.common.naming ksuit.utils.common.path ksuit.utils.common.stopwatch ksuit.utils.common.typing ksuit.utils.common.validation ksuit.utils.data ksuit.utils.data.data_container ksuit.utils.logging ksuit.utils.logging.core ksuit.utils.logging.formatting ksuit.utils.logging.progress ksuit.utils.model ksuit.utils.model.helpers ksuit.utils.model.model_utils ksuit.utils.platform ksuit.utils.platform.system ksuit.utils.platform.worker ksuit.utils.torch ksuit.utils.torch.amp ksuit.utils.torch.device ksuit.utils.training ksuit.utils.training.counter ksuit.utils.training.running_stats ksuit.utils.training.schedule_wrapper ksuit.utils.training.seed ksuit.utils.training.training_iteration ksuit.writers ksuit.writers.checkpoint_writer ksuit.writers.log_writer layers layers.continuous_sincos_embedding_config layers.drop_path_config layers.layer_scale_config layers.linear_projection_config layers.rope_frequenty_config mesh_to_grid_edges_precollator mlp mlp.mlp mlp.upactdown_mlp query_to_grid_edges_precollator surface_mask_precollator transformer transformer_config transolver transolver_config upt upt_config

MomentNormalizationBatchProcessor (class in emmi.pipeline.batch_processors.moment_normalization) MomentNormalizationSampleProcessor (class in emmi.pipeline.sample_processors.moment_normalization) move_items_to_device() (in module ksuit.utils.torch.device) MPS (emmi_inference.device.DeviceEnum attribute), [1] MultiBranchAnchorAttention (class in emmi.modules.attention.anchor_attention.multi_branch_anchor_attention) MultiBranchAnchorAttentionConfig (class in attention.anchor_attention.config) MultiStagePipeline (class in ksuit.data.pipeline.multistage)

N

naive_fs() (in module emmi_data_management.diskcache.cache_benchmark) naive_sqlite_fs() (in module emmi_data_management.diskcache.cache_benchmark) name (attention.anchor_attention.config.TokenSpec attribute) (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.models.model_base.ModelBase attribute) (ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier.LrScaleByNameModifier attribute) (ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier.WeightDecayByNameModifier attribute) (ksuit.utils.training.running_stats.RunningStats attribute) ndim (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (layers.linear_projection_config.LinearProjectionConfig attribute) ndim_padding (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) NEUTRAL (ksuit.providers.metric_property_provider.Ordinality attribute) next_epoch() (ksuit.utils.training.counter.UpdateCounter method) next_pct (emmi_data_management.logging_progress.LogProgress attribute) next_update() (ksuit.utils.training.counter.UpdateCounter method) nograd_paramnames (ksuit.models.model_base.ModelBase property) non_scalar_keys (ksuit.writers.log_writer.LogWriter attribute) noop() (ksuit.utils.logging.progress.NoopTqdm method) NoopContext (class in ksuit.utils.torch.amp) NoopGradScaler (class in ksuit.utils.torch.amp) NoopTqdm (class in ksuit.utils.logging.progress) NoopTracker (class in ksuit.trackers.noop_tracker) norm (upt.UPT attribute) norm1 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) (emmi_inference.models.modules.blocks.transformer_block.TransformerBlock attribute) norm1kv (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock attribute) norm1q (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock attribute) norm2 (emmi.modules.blocks.irregular_nat_block.IrregularNatBlock attribute) (emmi.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi.modules.blocks.transformer_block.TransformerBlock attribute) (emmi_inference.models.modules.blocks.perceiver_block.PerceiverBlock attribute) (emmi_inference.models.modules.blocks.transformer_block.TransformerBlock attribute) normalization_constructor (blocks.transformer_block_config.TransformerBlockConfig attribute) normalization_key (ksuit.data.preprocessors.compose.ComposePreProcess attribute) (ksuit.data.preprocessors.preprocessor.PreProcessor attribute)

normalize() (emmi_inference.common.SaveFormat class method) NormalizeAny (class in emmi_inference.preprocess) normalizers (ksuit.data.base.dataset.Dataset attribute) num_features (emmi.modules.layers.transformer_batchnorm.TransformerBatchNorm attribute) num_geometry_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_geometry_supernodes (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_heads (attention.config.AttentionConfig attribute) (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi_inference.models.transolver.Transolver attribute) (upt_config.UPTConfig attribute) num_input_features (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) num_layers (mlp.mlp.MLPConfig attribute) num_perceivers (ab_upt.AnchoredBranchedUPT attribute) num_points (emmi.pipeline.sample_processors.point_sampling.PointSamplingSampleProcessor attribute) (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) num_repeats (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) num_scalars (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) num_slices (attention.config.TransolverAttentionConfig attribute) num_supernodes (emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor attribute) (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_surface_anchor_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_surface_blocks (ab_upt_config.AnchorBranchedUPTConfig attribute) num_surface_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) (emmi_inference.models.pipelines.surface_mask_precollator.SurfaceMaskPreCollator attribute) (surface_mask_precollator.SurfaceMaskPreCollator attribute) num_surface_queries (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_volume_anchor_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_volume_blocks (ab_upt_config.AnchorBranchedUPTConfig attribute) num_volume_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) (emmi_inference.models.pipelines.surface_mask_precollator.SurfaceMaskPreCollator attribute) (surface_mask_precollator.SurfaceMaskPreCollator attribute) num_volume_queries (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) num_workers (ksuit.utils.data.data_container.DataContainer attribute) NUMPY_NPY (emmi_inference.common.SaveFormat attribute) NUMPY_NPZ (emmi_inference.common.SaveFormat attribute) NumpyToTorchTensorPreProcessor (class in emmi.modules.preprocessors.numpy_to_torch_tensor)

O

OFFLINE (ksuit.configs.wandb_config.WandBMode attribute) OfflineLossCallback (class in ksuit.callbacks.offline_callbacks.offline_loss) offset_key (emmi_inference.models.pipelines.sparse_tensor_offset.SparseTensorOffsetCollator attribute) ok (emmi_data_management.verification.VerificationResult attribute) ONLINE (ksuit.configs.wandb_config.WandBMode attribute) OnlineLossCallback (class in ksuit.callbacks.default_callbacks.online_loss) open_checkpoint_file() (in module emmi_inference.models.loader) ops (emmi_inference.preprocess.Compose attribute), [1] optimizer (ksuit.models.model_base.ModelBase property) optimizer_schedule_step() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) optimizer_step() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) optimizer_zero_grad() (ksuit.models.composite.CompositeModel method) (ksuit.models.model_base.ModelBase method) (ksuit.models.single.Model method) OptimizerFactory (class in ksuit.factory.optim_factory) OptimizerWrapper (class in ksuit.optimizer.optimizer_wrapper) optional (emmi_inference.models.pipelines.field_decoder.FieldDecoderCollator attribute) (layers.linear_projection_config.LinearProjectionConfig attribute)

optional_items (emmi.pipeline.collators.default.DefaultCollator attribute) OPTIONAL_PROVIDER_ENV_VARS (in module emmi_data_management.credentials) optional_target_metric_keys (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) Ordinality (class in ksuit.providers.metric_property_provider) output_dim (decoder.deep_perceiver_decoder.DeepPerceiverDecoderConfig attribute) (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (emmi_inference.models.base_model.BaseModel attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (layers.linear_projection_config.LinearProjectionConfig attribute) (mlp.mlp.MLPConfig attribute) output_key (emmi_inference.models.pipelines.mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (emmi_inference.models.pipelines.query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) (mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) output_path (ksuit.configs.static_config.StaticConfig property) output_patterns_to_log (ksuit.callbacks.offline_callbacks.offline_loss.OfflineLossCallback attribute) output_projection() (emmi_inference.models.base_model.BaseModel method) output_root (ksuit.providers.path_provider.PathProvider attribute) outputs (emmi_inference.runner.InferencePayload attribute) (ksuit.extractors.forward_hook.ForwardHook attribute) overhang_percent (ksuit.schedules.base.schedule_base.ScheduleBase attribute) overhang_steps (ksuit.schedules.base.schedule_base.ScheduleBase attribute)

P

padding (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) parallel (attention.anchor_attention.config.MixedAttentionConfig attribute) (emmi.modules.attention.anchor_attention.mixed_attention.MixedAttention attribute) parallel_map_collect_errors() (in module emmi_data_management.verification) ParallelErrors, [1] param_count (ksuit.models.model_base.ModelBase property) param_idx_to_name (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) param_was_found (ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier.LrScaleByNameModifier attribute) (ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier.WeightDecayByNameModifier attribute) ParamCountCallback (class in ksuit.callbacks.default_callbacks.param_count) parameters (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) ParamGroupModifierBase (class in ksuit.optimizer.param_group_modifiers.base.param_group_modifier_base) parse_args() (in module emmi.preprocessing.calculate_statistics_all) (in module emmi.preprocessing.calculate_statistics_train) (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) parse_devices() (in module ksuit.distributed.run.utils) parse_uri() (emmi_data_management.checkpoint.providers.huggingface_provider.HFProvider static method) pascal_to_snake() (in module ksuit.utils.common.naming) path (emmi_data_management.verification.FileRecord attribute) path_provider (ksuit.initializers.base.initializer_base.InitializerBase attribute) (ksuit.models.model_base.ModelBase attribute) (ksuit.trackers.base_tracker.BaseTracker attribute) (ksuit.trainers.base.BaseTrainer attribute) (ksuit.writers.checkpoint_writer.CheckpointWriter attribute) (ksuit.writers.log_writer.LogWriter attribute) PathProvider (class in ksuit.providers.path_provider) patterns (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) patterns_to_instantiate (ksuit.initializers.previous_run_initializer.PreviousRunInitializer attribute) patterns_to_remove (ksuit.initializers.previous_run_initializer.PreviousRunInitializer attribute) patterns_to_rename (ksuit.initializers.previous_run_initializer.PreviousRunInitializer attribute) PeakMemoryCallback (class in ksuit.callbacks.default_callbacks.peak_memory) per_key (emmi_inference.preprocess.NormalizeAny attribute), [1] perceiver (emmi.modules.blocks.perceiver_transformer_blockpair.PerceiverTransformerBlock attribute) perceiver_block_config (decoder.deep_perceiver_decoder.DeepPerceiverDecoderConfig attribute) PerceiverAttention (class in emmi.modules.attention.perceiver_attention) (class in emmi_inference.models.modules.attention.perceiver_attention) PerceiverAttentionConfig (class in attention.config) PerceiverAttentionConfig.Config (class in attention.config) PerceiverBlock (class in emmi.modules.blocks.perceiver_block) (class in emmi_inference.models.modules.blocks.perceiver_block) PerceiverBlockConfig (class in blocks.perceiver_block_config) PerceiverTransformerBlock (class in emmi.modules.blocks.perceiver_transformer_blockpair) PeriodicCallback (class in ksuit.callbacks.base.periodic_callback) PeriodicIteratorCallback (class in ksuit.callbacks.base.periodic_iterator_callback) physics_blocks (ab_upt.AnchoredBranchedUPT attribute) (ab_upt_config.AnchorBranchedUPTConfig attribute) physics_blocks_forward() (ab_upt.AnchoredBranchedUPT method) physics_dim (emmi.schemas.data.data_specs.AeroDataSpecs property) (emmi_inference.models.base_model.BaseModel attribute) PHYSICS_FEATURES (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) pin_memory (ksuit.utils.data.data_container.DataContainer attribute) pipeline (ksuit.data.base.dataset.Dataset property) (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) plot_3d_comparison() (in module emmi_inference.examples.run_abupt) PointCloudRegistration (class in emmi.utils.point_cloud_utils) PointSamplingSampleProcessor (class in emmi.pipeline.sample_processors.point_sampling) polynomial() (in module ksuit.schedules.functional) PolynomialDecreasingSchedule (class in ksuit.schedules.polynomial.decreasing)

PolynomialIncreasingSchedule (class in ksuit.schedules.polynomial.increasing) pool (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute) (emmi_inference.writer.AsyncWriter attribute) pop_ckpt_kwargs_keys (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) pos_embed (ab_upt.AnchoredBranchedUPT attribute) (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (upt.UPT attribute) pos_embed_hook (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) position_dim (emmi.schemas.data.data_specs.AeroDataSpecs attribute) position_item (emmi.pipeline.collators.field_decoder.FieldDecoderCollator attribute) (emmi_inference.models.pipelines.field_decoder.FieldDecoderCollator attribute) position_projection (emmi_inference.models.base_model.BaseModel attribute) position_scale (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) PositionNormalizationBatchProcessor (class in emmi.pipeline.batch_processors.position_normalization) PositionNormalizationSampleProcessor (class in emmi.pipeline.sample_processors.position_normalization) PositionNormalizer (class in emmi.modules.preprocessors.normalizers.position_normalizer) power (ksuit.schedules.polynomial.decreasing.PolynomialDecreasingSchedule attribute) (ksuit.schedules.polynomial.increasing.PolynomialIncreasingSchedule attribute) precision (ksuit.trainers.base.BaseTrainer attribute) pred (emmi_inference.models.upt.UPT attribute) prediction_layer (upt.UPT attribute) prefix (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute), [1] preprocess_inputs_only() (abupt.AbuptCollator method) (emmi_inference.models.pipelines.abupt.AbuptCollator method) PreProcessor (class in ksuit.data.preprocessors.preprocessor) preprocessor (emmi_inference.runner.InferenceRunner attribute) previous_log_values (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) PreviousRunInitializer (class in ksuit.initializers.previous_run_initializer) process_dataset_sample_by_sample() (in module emmi_inference.api) process_run() (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) ProgressCallback (class in ksuit.callbacks.default_callbacks.progress) ProgressSchedule (class in ksuit.schedules.base.progress_schedule) proj (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) proj_dropout (emmi.modules.attention.transolver_attention.TransolverAttention attribute) proj_temperature (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) project (emmi.modules.layers.linear_projection.LinearProjection attribute) (ksuit.configs.wandb_config.WandBConfig attribute), [1] (ksuit.trackers.wandb_tracker.WandbTracker attribute) properties (ksuit.data.wrappers.property_subset_wrapper.PropertySubsetWrapper attribute) PropertySubsetWrapper (class in ksuit.data.wrappers.property_subset_wrapper) protocol (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) (emmi_data_management.diskcache.lru_cache.SqliteLRUCacheFileSystem attribute) Provider (class in emmi_data_management.checkpoint.types) (class in emmi_data_management.providers) PROVIDER_ENV_VARS (in module emmi_data_management.credentials) PROVIDERS (in module emmi_data_management.checkpoint.resolver) push_scalar() (emmi.utils.running_moments.RunningMoments method) push_tensor() (emmi.utils.running_moments.RunningMoments method) (ksuit.utils.training.running_stats.RunningStats method) PYTORCH_PT (emmi_inference.common.SaveFormat attribute) PYTORCH_PTH (emmi_inference.common.SaveFormat attribute) PYTORCH_TH (emmi_inference.common.SaveFormat attribute)

Q

q (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (emmi_inference.models.modules.attention.perceiver_attention.PerceiverAttention attribute) qkv (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.attention.transolver_attention.TransolverAttention attribute) (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) (emmi_inference.models.modules.attention.dot_product_attention.DotProductAttention attribute)

query_item (emmi_inference.models.pipelines.query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) (query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) query_to_grid_edges_precollator module query_tokens (attention.anchor_attention.config.AttentionPattern attribute) QueryToGridEdgesPreCollator (class in emmi_inference.models.pipelines.query_to_grid_edges_precollator) (class in query_to_grid_edges_precollator)

R

R (in module emmi_data_management.verification) radius (attention.config.IrregularNatAttentionConfig attribute) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (emmi_inference.models.modules.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) (ksuit.scripts.validate_pytorch_compatibilities.StabilityTestModel attribute) radius_graph_max_num_neighbors (emmi_inference.models.pipelines.mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) (mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) radius_graph_r (emmi_inference.models.pipelines.mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (emmi_inference.models.pipelines.query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) (mesh_to_grid_edges_precollator.MeshToGridEdgesPreCollator attribute) (query_to_grid_edges_precollator.QueryToGridEdgesPreCollator attribute) raise_exception (ksuit.extractors.forward_hook.ForwardHook attribute) raw_pos_max (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi.modules.preprocessors.normalizers.position_normalizer.PositionNormalizer attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) raw_pos_max_tensor (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor attribute) raw_pos_min (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi.modules.preprocessors.normalizers.position_normalizer.PositionNormalizer attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) raw_pos_min_tensor (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor attribute) raw_pos_surface_max (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) raw_pos_surface_min (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) raw_pos_z_min (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) raw_size (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor attribute) read_file() (in module emmi_data_management.diskcache.cache_benchmark) readd_supernode_pos (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) RED (ksuit.utils.logging.formatting.CustomFormatter attribute) REDOWNLOAD (emmi_data_management.verification.FailAction attribute) reduce (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) register_pattern() (ksuit.providers.metric_property_provider.MetricPropertyProvider static method) register_sampler_config() (ksuit.callbacks.base.periodic_iterator_callback.PeriodicIteratorCallback method) relative_position_bias_mlp (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention attribute) relpath (in module ksuit.cli.main_calculate_stats) (in module ksuit.cli.main_train) relpos_mlp_dropout (attention.config.IrregularNatAttentionConfig attribute) relpos_mlp_hidden_dim (attention.config.IrregularNatAttentionConfig attribute) RELU (emmi.modules.activations.Activation attribute) remove_on_upload (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) RenameKeysBatchProcessor (class in emmi.pipeline.batch_processors.rename_keys) RenameKeysSampleProcessor (class in emmi.pipeline.sample_processors.rename_keys) RepeatWrapper (class in ksuit.data.wrappers.dataset_wrappers.repeat_wrapper) repetitions (ksuit.data.wrappers.dataset_wrappers.repeat_wrapper.RepeatWrapper attribute)

ReplaceKeySampleProcessor (class in emmi.pipeline.sample_processors.replace_key) require_online() (ksuit.configs.wandb_config.WandBConfig method) required_batch_modes (emmi_inference.models.base_model.BaseModel attribute) RESET (ksuit.utils.logging.formatting.CustomFormatter attribute) reset_parameters() (emmi.modules.attention.dot_product_attention.DotProductAttention method) (emmi.modules.attention.irregular_nat_attention.IrregularNatAttention method) (emmi.modules.attention.perceiver_attention.PerceiverAttention method) (emmi.modules.layers.linear_projection.LinearProjection method) (emmi.modules.mlp.mlp.MLP method) (emmi.modules.mlp.upactdown_mlp.UpActDownMlp method) resizing_scale (emmi.modules.preprocessors.normalizers.position_normalizer.PositionNormalizer attribute) resolve_autocast() (in module emmi_inference.device) resolve_checkpoint() (in module emmi_data_management.checkpoint.resolver) resolve_device() (in module emmi_inference.device) resolve_dir() (in module emmi_data_management.cli_utils) (in module emmi_inference.cli_utils) resume_from_checkpoint() (ksuit.callbacks.base.callback_base.CallbackBase method) (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback method) ResumeInitializer (class in ksuit.initializers.resume_initializer) returns_partials (ksuit.factory.base.factory_base.Factory attribute) RICH_MARKUP_MODE (in module emmi_data_management.cli) (in module emmi_data_management.cli_aws) (in module emmi_data_management.cli_checkpoint) (in module emmi_data_management.cli_huggingface) (in module emmi_data_management.cli_verification) (in module emmi_inference.cli) root (emmi_inference.writer.AsyncWriter attribute) rope (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) rope() (in module emmi.functional.rope) (in module emmi_inference.models.modules.rope) RopeFrequency (class in emmi.modules.layers.rope_frequency) (class in emmi_inference.models.modules.rope_frequency) RopeFrequencyConfig (class in layers.rope_frequenty_config) run() (emmi_inference.runner.InferenceRunner method) (in module emmi_inference.cli) (in module ksuit.distributed.run) (ksuit.runners.hydra_runner.HydraRunner method) (ksuit.runners.stats_runner.StatsRunner static method) run_abupt() (in module emmi_inference.examples.run_abupt) run_cli() (in module emmi_data_management.cli_utils) (in module emmi_inference.cli_utils) run_cuda_assertions() (in module ksuit.scripts.validate_pytorch_compatibilities) run_id (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) (ksuit.providers.path_provider.PathProvider attribute) (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) run_inference_from_config() (in module emmi_inference.api) run_managed() (in module ksuit.distributed.run.managed) run_output_path (ksuit.providers.path_provider.PathProvider property) run_post_upload() (in module emmi_inference.api) run_transolver() (in module emmi_inference.examples.run_transolver) run_unmanaged() (in module ksuit.distributed.run.unmanaged) run_upt() (in module emmi_inference.examples.run_upt) RunningMoments (class in emmi.utils.running_moments) RunningStats (class in ksuit.utils.training.running_stats)

S

s3_bucket (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) s3_client (emmi_data_management.interfaces.s3_uploader.S3FileUploader attribute) (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) s3_pool (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) s3_prefix (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) S3_URI_PREFIX (in module emmi_inference) S3FileUploader (class in emmi_data_management.interfaces.s3_uploader) S3Object (class in emmi_data_management.interfaces.s3) S3SpoolingUploader (class in emmi_inference.s3_spooling_uploader) sample (ksuit.utils.training.counter.UpdateCounter property) (ksuit.utils.training.training_iteration.TrainingIteration attribute) sample_processors (ksuit.data.pipeline.multistage.MultiStagePipeline attribute) sample_query_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) SampleProcessor (class in ksuit.data.pipeline.sample_processor) SampleProcessorType (in module ksuit.data.pipeline.multistage) sampler (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig attribute) SamplerIntervalConfig (class in ksuit.data.samplers.interleaved_sampler) SamplewiseNormalizationSampleProcessor (class in emmi.pipeline.sample_processors.samplewise_normalization) sanitize() (in module emmi_data_management.cli_utils) sanitize_suffix() (in module emmi_inference.common) save() (ksuit.writers.checkpoint_writer.CheckpointWriter method) save_copy() (ksuit.data.pipeline.sample_processor.SampleProcessor static method) save_format (emmi_inference.writer.AsyncWriter attribute) save_frozen_weights (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) save_last_weights (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) save_latest_optim (ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback attribute) save_latest_weights (ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback attribute) (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) save_manifest() (in module emmi_data_management.verification) save_model_checkpoint() (ksuit.writers.checkpoint_writer.CheckpointWriter method) save_optim (ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback attribute) save_output (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) save_resolved() (ksuit.configs.hyperparameters.Hyperparameters static method) save_weights (ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback attribute) (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) SaveFormat (class in emmi_inference.common) ScalarOrSequence (in module ksuit.data.preprocessors.types) ScalarsConditioner (class in emmi.modules.layers.scalar_conditioner) scale (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) (emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer.ShiftAndScaleNormalizer attribute) (emmi.pipeline.batch_processors.position_normalization.PositionNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor attribute) (ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier.LrScaleByNameModifier attribute) scale() (ksuit.utils.torch.amp.NoopGradScaler method) scale_by_keep (emmi.modules.layers.drop_path.UnquantizedDropPath attribute) (layers.drop_path_config.UnquantizedDropPathConfig attribute) schedule (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) (ksuit.schedules.base.sequential_percent_schedule.SequentialPercentScheduleConfig attribute) (ksuit.schedules.base.sequential_step_schedule.SequentialStepScheduleConfig attribute) (ksuit.utils.training.schedule_wrapper.ScheduleWrapper attribute) schedule_configs (ksuit.schedules.base.sequential_percent_schedule.SequentialPercentSchedule attribute) (ksuit.schedules.base.sequential_step_schedule.SequentialStepSchedule attribute) schedule_step() (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper method) ScheduleBase (class in ksuit.schedules.base.schedule_base) ScheduleFactory (class in ksuit.factory.schedule_factory) ScheduleWrapper (class in ksuit.utils.training.schedule_wrapper) scheme (emmi_data_management.checkpoint.providers.file_provider.FileProvider attribute) (emmi_data_management.checkpoint.providers.huggingface_provider.HFProvider attribute) (emmi_data_management.checkpoint.types.Provider attribute) seconds_to_duration_str() (in module ksuit.utils.logging.formatting) seed (emmi.pipeline.sample_processors.point_sampling.PointSamplingSampleProcessor attribute) (emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor attribute) (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) (ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper.ShuffleWrapper attribute) seed() (in module ksuit.cli.main_train) select_with_path() (in module ksuit.utils.common.path) SelfAnchorAttention (class in emmi.modules.attention.anchor_attention.self_anchor_attention) SequenceOrTensor (in module emmi.modules.preprocessors.normalizers.types) SequentialPercentSchedule (class in ksuit.schedules.base.sequential_percent_schedule) SequentialPercentScheduleConfig (class in ksuit.schedules.base.sequential_percent_schedule) SequentialStepSchedule (class in ksuit.schedules.base.sequential_step_schedule) SequentialStepScheduleConfig (class in ksuit.schedules.base.sequential_step_schedule) set() (ksuit.utils.bidict.Bidict method) set_config() (in module ksuit.distributed.config) set_deterministic_mode() (in module ksuit.scripts.validate_pytorch_compatibilities) set_kv_dim() (attention.config.PerceiverAttentionConfig method) (blocks.perceiver_block_config.PerceiverBlockConfig method) set_mlp_hidden_dim() (blocks.transformer_block_config.TransformerBlockConfig method) (transformer_config.TransformerConfig method) set_seed() (in module ksuit.utils.training.seed) set_summary() (ksuit.trackers.base_tracker.BaseTracker method) setup_environment() (in module emmi_inference.api) setup_source_root() (ksuit.providers.dataset_config_provider.DatasetConfigProvider method) sha256 (emmi_data_management.checkpoint.types.CheckpointMetadata attribute) ShapeNetCarDataset (class in emmi_inference.datasets.shapenet) ShapeNetCarStats (class in emmi_inference.datasets.shapenet) ShapenetMultistageCollator (class in emmi_inference.models.pipelines.shapenet_multistage) shared_mlp (emmi.modules.layers.scalar_conditioner.ScalarsConditioner attribute) SharedweightsCrossattnAttention (class in emmi_inference.models.modules.attention.sharedweights_crossattn_attention) SharedweightsSplitattnAttention (class in emmi_inference.models.modules.attention.sharedweights_splitattn_attention) shift (emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer.ShiftAndScaleNormalizer attribute) ShiftAndScaleNormalizer (class in emmi.modules.preprocessors.normalizers.shift_and_scale_normalizer) short_number_str() (in module ksuit.utils.logging.formatting) should_log_after_epoch() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) should_log_after_sample() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) should_log_after_update() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) (ksuit.callbacks.default_callbacks.lr.LrCallback method) ShuffleWrapper (class in ksuit.data.wrappers.dataset_wrappers.shuffle_wrapper) SIGMOID (emmi.modules.activations.Activation attribute) SILU (emmi.modules.activations.Activation attribute) sincos_padding (emmi.modules.layers.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi.modules.layers.rope_frequency.RopeFrequency attribute) (emmi_inference.models.modules.continuous_sincos_embed.ContinuousSincosEmbed attribute) (emmi_inference.models.modules.rope_frequency.RopeFrequency attribute) SINGLE (emmi_inference.common.DictMode attribute) size (attention.anchor_attention.config.TokenSpec attribute) (default_tensor_precollator.DefaultTensorPreCollator attribute) (emmi_data_management.checkpoint.types.CheckpointMetadata attribute) (emmi_data_management.interfaces.s3.S3Object attribute) (emmi_data_management.verification.FileRecord attribute) SIZE (emmi_data_management.verification.VerificationType attribute) size (emmi_inference.models.pipelines.default_tensor_precollator.DefaultTensorPreCollator attribute) size_mismatch (emmi_data_management.verification.VerificationResult attribute) SizedIterable (class in ksuit.utils.common.typing) skip_nan_loss_counter (ksuit.trainers.base.BaseTrainer attribute)

small_sync_bytes (emmi_inference.writer.AsyncWriter attribute) snake_type_name() (in module ksuit.utils.common.naming) softmax (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) SOFTPLUS (emmi.modules.activations.Activation attribute) source (emmi_data_management.verification.FileRecord attribute) source_key (emmi.pipeline.sample_processors.replace_key.ReplaceKeySampleProcessor attribute) source_metric_key (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) source_root (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) SparseTensorOffsetCollator (class in emmi_inference.models.pipelines.sparse_tensor_offset) specified_properties_count (ksuit.utils.training.training_iteration.TrainingIteration property) split (emmi.datasets.ahmedml.AhmedMLDataset attribute) (emmi.datasets.drivaerml.DrivAerMLDataset attribute) SPLIT (emmi_inference.common.DictMode attribute) split (emmi_inference.datasets.shapenet.ShapeNetCarDataset attribute) SPOOL (emmi_inference.UploadMode attribute) spool_pos_mode (emmi.modules.encoder.supernode_pooling.SupernodePooling attribute) (emmi.modules.encoder.supernode_pooling_posonly.SupernodePoolingPosonly attribute) (encoder.supernode_pooling_config.SupernodePoolingConfig attribute) SqliteLRUCacheFileSystem (class in emmi_data_management.diskcache.lru_cache) src_idx_hook (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) StabilityTestModel (class in ksuit.scripts.validate_pytorch_compatibilities) stage_name (ksuit.initializers.base.checkpoint_initializer.CheckpointInitializer attribute) (ksuit.providers.path_provider.PathProvider attribute) start() (ksuit.utils.common.stopwatch.Stopwatch method) start_epoch (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) start_iteration (ksuit.utils.training.counter.UpdateCounter attribute) start_percent (ksuit.schedules.base.sequential_percent_schedule.SequentialPercentScheduleConfig attribute) start_sample (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) start_step (ksuit.schedules.base.sequential_step_schedule.SequentialStepScheduleConfig attribute) start_update (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig attribute) start_value (ksuit.schedules.base.progress_schedule.ProgressSchedule attribute) (ksuit.schedules.step.fixed.StepFixedSchedule attribute) (ksuit.schedules.step.interval.StepIntervalSchedule attribute) state_dict (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) state_dict() (ksuit.callbacks.base.callback_base.CallbackBase method) (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback method) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper method) (ksuit.trainers.base.BaseTrainer method) static_context (ksuit.models.model_base.ModelBase attribute) StaticConfig (class in ksuit.configs.static_config) StatsRunner (class in ksuit.runners.stats_runner) status (emmi_inference.runner.InferencePayload attribute) std (emmi.modules.preprocessors.normalizers.mean_std_normalization.MeanStdNormalization attribute) (emmi_inference.preprocess.NormalizeAny attribute), [1] (ksuit.utils.training.running_stats.RunningStats property) std() (emmi.utils.running_moments.RunningMoments method) std_tensor (emmi.pipeline.batch_processors.moment_normalization.MomentNormalizationBatchProcessor attribute) (emmi.pipeline.sample_processors.moment_normalization.MomentNormalizationSampleProcessor attribute) step (emmi_data_management.logging_progress.LogProgress attribute) step() (emmi.optim.lion.Lion method) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper method) (ksuit.utils.torch.amp.NoopGradScaler static method) StepDecreasingSchedule (class in ksuit.schedules.step.decreasing) StepFixedSchedule (class in ksuit.schedules.step.fixed) StepIntervalSchedule (class in ksuit.schedules.step.interval) steps (ksuit.schedules.step.fixed.StepFixedSchedule attribute) stl_prefix (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) stop() (ksuit.utils.common.stopwatch.Stopwatch method) stop_at_epoch (ksuit.callbacks.early_stoppers.fixed_early_stopper.FixedEarlyStopper attribute) stop_at_sample (ksuit.callbacks.early_stoppers.fixed_early_stopper.FixedEarlyStopper attribute) stop_at_update (ksuit.callbacks.early_stoppers.fixed_early_stopper.FixedEarlyStopper attribute) StopForwardException Stopwatch (class in ksuit.utils.common.stopwatch) storage (emmi_data_management.diskcache.lru_cache.LRUCacheFileSystem attribute) store_code_archive() (in module ksuit.utils.code) submodels (ksuit.models.composite.CompositeModel property) Subset (class in ksuit.data.base.subset) SubsetWrapper (class in ksuit.data.wrappers.dataset_wrappers.subset_wrapper) summarize_indices_list() (in module ksuit.utils.logging.formatting) summarize_logvalues() (ksuit.trackers.base_tracker.BaseTracker method) summary (ksuit.trackers.base_tracker.BaseTracker attribute) summary() (emmi_inference.runner.InferencePayload method) supernode_idx_key (emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor attribute) supernode_pooling_config (ab_upt_config.AnchorBranchedUPTConfig attribute) (upt_config.UPTConfig attribute) supernode_pooling_path (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) SupernodePooling (class in emmi.modules.encoder.supernode_pooling) SupernodePoolingConfig (class in encoder.supernode_pooling_config) SupernodePoolingCoverageCallback (class in emmi.callbacks.supernode_pooling_coverage) SupernodePoolingPosonly (class in emmi.modules.encoder.supernode_pooling_posonly) (class in emmi_inference.models.modules.supernode_pooling_posonly) SupernodeSamplingSampleProcessor (class in emmi.pipeline.sample_processors.supernode_sampling) surface_and_volume_bias() (emmi_inference.models.base_model.BaseModel method) (upt.UPT method) surface_bias (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) surface_blocks (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) surface_decoder (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) surface_decoder_blocks (ab_upt.AnchoredBranchedUPT attribute) surface_features (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) SURFACE_MASK_INPUT (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) surface_mask_precollator module SURFACE_NORMALS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) surface_output_dims (emmi.schemas.data.data_specs.AeroDataSpecs attribute) SURFACE_POS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) surface_pos_mean (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) surface_pos_std (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) SURFACE_PRESSURE (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) surface_pressure_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) surface_pressure_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) SURFACE_PRESSURE_TARGET (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) SURFACE_SDF (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) surface_targets (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) surface_wallshearstress_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) surface_wallshearstress_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) SurfaceMaskPreCollator (class in emmi_inference.models.pipelines.surface_mask_precollator) (class in surface_mask_precollator) sync_device() (in module emmi_inference.device)

T

T (in module emmi_data_management.verification) (in module ksuit.data.pipeline.sample_processor) TANH (emmi.modules.activations.Activation attribute) target_factors (ksuit.callbacks.checkpoint_callbacks.ema.EmaCallback attribute) target_items (emmi.pipeline.collators.field_decoder.FieldDecoderCollator attribute) (emmi_inference.models.pipelines.field_decoder.FieldDecoderCollator attribute) target_key (concat_tensor_precollator.ConcatTensorPreCollator attribute) (emmi_inference.models.pipelines.concat_tensor_precollator.ConcatTensorPreCollator attribute) target_keys (emmi.pipeline.sample_processors.replace_key.ReplaceKeySampleProcessor attribute) target_metric_keys (ksuit.callbacks.online_callbacks.best_metric.BestMetricCallback attribute) target_properties (ksuit.trainers.base.BaseTrainer attribute) temperature (emmi.modules.attention.transolver_attention.TransolverAttention attribute) temperature_activation (attention.config.TransolverPlusPlusAttentionConfig attribute) tensor_like_to_string() (in module ksuit.utils.logging.formatting) test (emmi.datasets.ahmedml.AhmedMLDefaultSplitIDs attribute) (emmi.datasets.drivaerml.DrivAerMLDefaultSplitIDs attribute) time_block() (in module emmi.preprocessing.process_drivaer_and_ahmed_volumetric) time_since_last_log (ksuit.callbacks.default_callbacks.eta.EtaCallback attribute) to() (ksuit.models.composite.CompositeModel method) (ksuit.models.single.Model method) to_backward() (ksuit.utils.bidict.Bidict method) to_cpu() (emmi_inference.runner.InferencePayload method) to_dict() (attention.anchor_attention.config.TokenSpec method) to_forward() (ksuit.utils.bidict.Bidict method) to_fully_specified() (ksuit.utils.training.training_iteration.TrainingIteration method) to_fully_specified_from_filenames() (ksuit.utils.training.training_iteration.TrainingIteration static method) to_jsonable() (in module emmi_data_management.cli_checkpoint) to_logscale() (in module emmi.functional.logscale) to_out (emmi.modules.attention.transolver_plusplus_attention.TransolverPlusPlusAttention attribute) to_prefix_and_postfix (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) to_prefix_midfix_postfix (emmi_inference.models.pipelines.anchor_point_sampling_precollator.AnchorPointSamplingPreCollator attribute) to_short_interval_string() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) (ksuit.callbacks.early_stoppers.early_stopper_base.EarlyStopperBase method) to_target_specification() (ksuit.utils.training.training_iteration.TrainingIteration method) to_tensor() (in module ksuit.data.preprocessors.utils) TokenSpec (class in attention.anchor_attention.config) tolerance (ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper attribute) tolerance_counter (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) (ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper attribute) tolerances_is_exceeded (ksuit.callbacks.checkpoint_callbacks.best_checkpoint.BestCheckpointCallback attribute) torch_optim (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) total (emmi_data_management.logging_progress.LogProgress attribute) total_data_time (ksuit.callbacks.base.periodic_iterator_callback.PeriodicIteratorCallback attribute) total_dim (emmi.schemas.data.data_specs.FieldDimSpec property) total_loss (ksuit.trainers.types.TrainerResult attribute) total_output_dim (emmi.schemas.data.data_specs.AeroDataSpecs property) total_time (ksuit.callbacks.default_callbacks.eta.EtaCallback attribute) total_train_data_time (ksuit.callbacks.default_callbacks.train_time.TrainTimeCallback attribute) track_after_accumulation_step() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method)

track_after_update_step() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) tracked_coverages (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) tracked_degrees (emmi.callbacks.supernode_pooling_coverage.SupernodePoolingCoverageCallback attribute) tracked_losses (ksuit.callbacks.default_callbacks.online_loss.OnlineLossCallback attribute) tracked_values (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) tracker (ksuit.callbacks.base.callback_base.CallbackBase attribute) (ksuit.trainers.base.BaseTrainer attribute) (ksuit.writers.log_writer.LogWriter attribute) train (emmi.datasets.ahmedml.AhmedMLDefaultSplitIDs attribute) (emmi.datasets.drivaerml.DrivAerMLDefaultSplitIDs attribute) train() (ksuit.models.composite.CompositeModel method) (ksuit.models.single.Model method) (ksuit.trainers.base.BaseTrainer method) train_data_times (ksuit.callbacks.default_callbacks.train_time.TrainTimeCallback attribute) train_step() (ksuit.trainers.base.BaseTrainer method) trainable_param_count (ksuit.models.model_base.ModelBase property) trainer (ksuit.callbacks.base.callback_base.CallbackBase attribute) TrainerResult (class in ksuit.trainers.types) training_iteration (ksuit.writers.checkpoint_writer.CheckpointKeys attribute) TrainingIteration (class in ksuit.utils.training.training_iteration) TrainTimeCallback (class in ksuit.callbacks.default_callbacks.train_time) transform() (emmi.utils.point_cloud_utils.PointCloudRegistration method) transformer module Transformer (class in transformer) transformer (emmi.modules.blocks.perceiver_transformer_blockpair.PerceiverTransformerBlock attribute) transformer_block_config (ab_upt_config.AnchorBranchedUPTConfig attribute) transformer_config module TransformerBatchNorm (class in emmi.modules.layers.transformer_batchnorm) TransformerBlock (class in emmi.modules.blocks.transformer_block) (class in emmi_inference.models.modules.blocks.transformer_block) TransformerBlockConfig (class in blocks.transformer_block_config) TransformerConfig (class in transformer_config) transforms (ksuit.data.preprocessors.compose.ComposePreProcess attribute) transolver module Transolver (class in emmi_inference.models.transolver) (class in transolver) transolver_config module TransolverAttention (class in emmi.modules.attention.transolver_attention) TransolverAttentionConfig (class in attention.config) TransolverAttentionConfig.Config (class in attention.config) TransolverConfig (class in transolver_config) TransolverPlusPlusAttention (class in emmi.modules.attention.transolver_plusplus_attention) TransolverPlusPlusAttentionConfig (class in attention.config) TransolverPlusPlusAttentionConfig.Config (class in attention.config) TransolverPlusPlusConfig (class in transolver_config)

U

underlying_fs() (in module emmi_data_management.diskcache.cache_benchmark) UnexpectedTokenError UnquantizedDropPath (class in emmi.modules.layers.drop_path) UnquantizedDropPathConfig (class in layers.drop_path_config) unscale_() (ksuit.utils.torch.amp.NoopGradScaler method) unstructured_grid_data_to_poly_data() (in module emmi.preprocessing.shapenet_car.preprocess_shapenet_car) UpActDownMlp (class in emmi.modules.mlp.upactdown_mlp) UpActDownMLPConfig (class in mlp.upactdown_mlp) update (ksuit.utils.training.counter.UpdateCounter property) (ksuit.utils.training.training_iteration.TrainingIteration attribute) update() (emmi_data_management.logging_progress.LogProgress method) (ksuit.trainers.base.BaseTrainer method) (ksuit.utils.torch.amp.NoopGradScaler method) update_counter (ksuit.models.model_base.ModelBase attribute) (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) (ksuit.trainers.base.BaseTrainer attribute) (ksuit.utils.training.schedule_wrapper.ScheduleWrapper attribute) (ksuit.writers.checkpoint_writer.CheckpointWriter attribute) (ksuit.writers.log_writer.LogWriter attribute) update_ema() (in module ksuit.utils.model.model_utils) update_format (ksuit.callbacks.default_callbacks.eta.EtaCallback attribute) update_interval (ksuit.schedules.step.interval.StepIntervalSchedule attribute) update_summary() (ksuit.trackers.base_tracker.BaseTracker method) UpdateCounter (class in ksuit.utils.training.counter) UpdateOutputCallback (class in ksuit.callbacks.online_callbacks.update_output) updates_per_epoch (ksuit.trainers.base.BaseTrainer attribute) (ksuit.utils.training.counter.UpdateCounter attribute) updates_per_log_interval() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method)

updates_till_next_log() (ksuit.callbacks.base.periodic_callback.PeriodicCallback method) upload_all() (emmi_data_management.interfaces.s3_uploader.S3FileUploader method) upload_batch_size (emmi_inference.s3_spooling_uploader.S3SpoolingUploader attribute) upload_queue (emmi_inference.writer.AsyncWriter attribute) UploadMode (class in emmi_inference) upt module UPT (class in emmi_inference.models.upt) (class in upt) upt_config module UPTConfig (class in upt_config) uri (emmi_data_management.checkpoint.types.CheckpointMetadata attribute) use_adaptive_temperature (attention.config.TransolverPlusPlusAttentionConfig attribute) use_anchor_points (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator property) use_gino_precollators (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) use_gumbel_softmax (attention.config.TransolverPlusPlusAttentionConfig attribute) USE_OCI (in module emmi_data_management.diskcache.cache_benchmark) use_overparameterization (attention.config.TransolverPlusPlusAttentionConfig attribute) use_physics_features (emmi.schemas.data.data_specs.AeroDataSpecs property) (emmi_inference.models.base_model.BaseModel attribute) (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) use_query_positions (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) use_rope (attention.config.AttentionConfig attribute) (blocks.transformer_block_config.TransformerBlockConfig attribute) (emmi.modules.attention.dot_product_attention.DotProductAttention attribute) (emmi.modules.attention.perceiver_attention.PerceiverAttention attribute) (upt.UPT attribute) (upt_config.UPTConfig attribute)

V

val (emmi.datasets.ahmedml.AhmedMLDefaultSplitIDs attribute) (emmi.datasets.drivaerml.DrivAerMLDefaultSplitIDs attribute) VALID_PRECISIONS (in module ksuit.utils.torch.amp) validate_frequency() (ksuit.data.samplers.interleaved_sampler.SamplerIntervalConfig method) validate_hidden_dim_and_num_heads() (attention.config.AttentionConfig method) validate_ndim() (layers.linear_projection_config.LinearProjectionConfig method) validate_path() (in module ksuit.utils.common.path) validate_radius_and_k() (encoder.supernode_pooling_config.SupernodePoolingConfig method) validate_required_batch_input() (in module emmi_inference.models.base_model) validate_start() (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig method) validate_stop() (ksuit.data.samplers.interleaved_sampler.InterleavedSamplerConfig method) value (ksuit.configs.wandb_config.WandBMode property) (ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier.WeightDecayByNameModifier attribute) (ksuit.schedules.constant_schedule.ConstantSchedule attribute) values (ksuit.schedules.custom_schedule.CustomSchedule attribute) values() (emmi.schemas.data.data_specs.FieldDimSpec method) var (ksuit.utils.training.running_stats.RunningStats property) var() (emmi.utils.running_moments.RunningMoments method) verbose (ksuit.callbacks.default_callbacks.online_loss.OnlineLossCallback attribute) (ksuit.callbacks.online_callbacks.update_output.UpdateOutputCallback attribute) verification_app (in module emmi_data_management.cli_verification) VerificationResult (class in emmi_data_management.verification) VerificationType (class in emmi_data_management.verification) verify_build() (in module emmi_data_management.cli_verification) verify_check() (in module emmi_data_management.cli_verification) verify_tree() (in module emmi_data_management.verification) volume_bias (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) volume_blocks (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) volume_decoder (ab_upt.AnchoredBranchedUPT attribute) (emmi_inference.models.abupt.AnchoredBranchedUPT attribute) volume_decoder_blocks (ab_upt.AnchoredBranchedUPT attribute) volume_features (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) VOLUME_NORMALS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) volume_output_dims (emmi.schemas.data.data_specs.AeroDataSpecs attribute) VOLUME_POS (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute)

volume_pos_mean (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) volume_pos_std (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) VOLUME_SDF (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) volume_sdf_mean (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) volume_sdf_std (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) volume_targets (emmi_inference.models.pipelines.shapenet_multistage.ShapenetMultistageCollator attribute) volume_totalpcoeff_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_totalpcoeff_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) VOLUME_VELOCITY (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) volume_velocity_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) volume_velocity_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) (emmi_inference.datasets.shapenet.ShapeNetCarStats attribute) VOLUME_VELOCITY_TARGET (emmi_inference.models.pipelines.shapenet_multistage.DataKeys attribute) volume_vorticity_logscale_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_vorticity_logscale_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_vorticity_magnitude_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_vorticity_magnitude_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_vorticity_mean (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) volume_vorticity_std (emmi.datasets.ahmedml.AhmedMLStats attribute) (emmi.datasets.drivaerml.DrivAerMLStats attribute) VT (in module ksuit.utils.bidict) vtp_pressure (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) vtp_wallshearstress (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) vtu_pressure (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) vtu_totalp (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) vtu_velocity (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute) vtu_vorticity (emmi.preprocessing.process_drivaer_and_ahmed_volumetric.DataFieldKeyMap attribute)

W

WandBConfig (class in ksuit.configs.wandb_config) WandBMode (class in ksuit.configs.wandb_config) WandbTracker (class in ksuit.trackers.wandb_tracker) warmup_percent (ksuit.schedules.linear_warmup_cosine_decay.LinearWarmupCosineDecaySchedule attribute) warmup_steps (ksuit.schedules.linear_warmup_cosine_decay.LinearWarmupCosineDecaySchedule attribute) WARN (emmi_data_management.verification.FailAction attribute) was_applied_successfully() (ksuit.optimizer.param_group_modifiers.base.param_group_modifier_base.ParamGroupModifierBase method) (ksuit.optimizer.param_group_modifiers.lr_scale_by_name_modifier.LrScaleByNameModifier method) (ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier.WeightDecayByNameModifier method) was_called (ksuit.callbacks.default_callbacks.eta.EtaCallback.LoggerWasCalledHandler attribute)

weight_decay_schedule (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper attribute) WeightDecayByNameModifier (class in ksuit.optimizer.param_group_modifiers.weight_decay_by_name_modifier) with_normalizers() (in module ksuit.data.base.dataset) with_run() (ksuit.providers.path_provider.PathProvider method) wrap_compile() (ksuit.trainers.base.BaseTrainer method) wrap_ddp() (ksuit.trainers.base.BaseTrainer method) wrap_model() (ksuit.trainers.base.BaseTrainer method) write() (emmi_data_management.interfaces.s3_uploader.S3FileUploader method) (emmi_inference.s3_spooling_uploader.S3SpoolingUploader method) (emmi_inference.writer.AsyncWriter method) writer (ksuit.callbacks.base.callback_base.CallbackBase attribute)

X

xyz_scale (emmi.utils.point_cloud_utils.PointCloudRegistration attribute)

Y

YELLOW (ksuit.utils.logging.formatting.CustomFormatter attribute)

Z

zero_grad() (ksuit.optimizer.optimizer_wrapper.OptimizerWrapper method)