Index

_

_async_copies (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _available_hbm_pages (easydel.layers.caching.__init__.HBMPageManager attribute) (easydel.layers.caching.paged_attention.__init__.HBMPageManager attribute) _base (easydel.modules.auto.auto_modeling.BaseAutoEasyState attribute) _compile_config (easydel.inference.vinference.__init__.SampleState attribute) (easydel.inference.vinference.utilities.SampleState attribute) _current_page_index (easydel.layers.caching.__init__.HBMPageManager attribute) (easydel.layers.caching.paged_attention.__init__.HBMPageManager attribute) _loop_rows (easydel.__init__.vInferenceConfig attribute) (easydel.inference.__init__.vInferenceConfig attribute) (easydel.inference.vinference.__init__.vInferenceConfig attribute) (easydel.inference.vinference.utilities.vInferenceConfig attribute)

_metadata (easydel.layers.caching.__init__.HBMPageManager attribute) (easydel.layers.caching.paged_attention.__init__.HBMPageManager attribute) _num_pages_to_load (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _page_indices (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _page_indices_start_offset (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _pages_hbm_ref (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _sem (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute) _time_spent_computing (easydel.inference.vinference.__init__.SampleState attribute) (easydel.inference.vinference.utilities.SampleState attribute) _vmem_buffer (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor attribute)

A

A8BIT (easydel.__init__.EasyDeLQuantizationMethods attribute), [1] (easydel.infra.etils.EasyDeLQuantizationMethods attribute), [1] accuracy (easydel.infra.loss_utils.LossMetrics attribute), [1] act (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) act_fn (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) activation_fn (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) active_slot_requests_map (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] ActiveRequest (class in easydel.inference.vsurge.utils) ActiveRequestMetadata (class in easydel.inference.vsurge.utils) ActiveSequenceBatch (class in easydel.layers.caching.paged_attention.__init__) ADAFACTOR (easydel.__init__.EasyDeLOptimizers attribute), [1] (easydel.infra.etils.EasyDeLOptimizers attribute), [1] ADAMW (easydel.__init__.EasyDeLOptimizers attribute), [1] (easydel.infra.etils.EasyDeLOptimizers attribute), [1] add_basic_configurations() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) add_bos_token_if_needed() (in module easydel.trainers.utils) add_eos_token_if_needed() (in module easydel.trainers.utils) add_result() (easydel.inference.vsurge.utils.AsyncMultifuture method) add_special_tokens (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) add_start_docstrings() (in module easydel.inference.logits_process) (in module easydel.infra.utils) adjust_kwargs() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) alibi (easydel.__init__.MptModel attribute) (easydel.__init__.MptModel property) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.__init__.MptModel property) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel property) ALL_GATHER (easydel.kernels.collective_matmul.MatrixMultiplyMethod attribute), [1] all_router_losses (easydel.infra.modeling_outputs.MoeCausalLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) alloc_hbm_pages() (easydel.layers.caching.__init__.HBMPageManager method) (easydel.layers.caching.paged_attention.__init__.HBMPageManager method) alloc_prefill_hbm_pages() (easydel.layers.caching.__init__.HBMPageManager method) (easydel.layers.caching.paged_attention.__init__.HBMPageManager method) AllocatedPrefillPages (class in easydel.layers.caching.paged_attention.__init__) allocation_efficiency (easydel.utils.analyze_memory.DeviceStats attribute), [1] analyze_device() (easydel.utils.analyze_memory.SMPMemoryMonitor method) analyze_memory() (easydel.utils.analyze_memory.MemoryMonitorClient method) apply_assignment() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) apply_basic_rope() (in module easydel.layers.rotary_embedding) apply_chat_template() (in module easydel.trainers.prompt_utils) apply_complex_rotary() (easydel.__init__.AttentionModule static method) (easydel.layers.attention.AttentionModule static method) apply_fn (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_state.EasyDeLState attribute), [1] apply_gradients() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) apply_lora_to_layers() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (in module easydel.infra.utils) apply_out_shardings() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) apply_phi3_rope() (in module easydel.layers.rotary_embedding) apply_qk_shardings() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) apply_qkv_shardings() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) apply_rotary_pos_emb() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) (in module easydel.modules.pixtral.modeling_pixtral_flax) apply_rotary_pos_emb_vision() (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) apply_sparsity_to_params() (in module easydel.infra.utils) apply_training_hooks() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) ArcticAttention (class in easydel.modules.arctic.modeling_arctic_flax) ArcticConfig (class in easydel.__init__) (class in easydel.modules.arctic.__init__) (class in easydel.modules.arctic.arctic_configuration) ArcticDecoderLayer (class in easydel.modules.arctic.modeling_arctic_flax) ArcticForCausalLM (class in easydel.__init__) (class in easydel.modules.arctic.__init__) (class in easydel.modules.arctic.modeling_arctic_flax) ArcticForSequenceClassification (class in easydel.modules.arctic.__init__) (class in easydel.modules.arctic.modeling_arctic_flax) ArcticMLP (class in easydel.modules.arctic.modeling_arctic_flax) ArcticModel (class in easydel.__init__) (class in easydel.modules.arctic.__init__) (class in easydel.modules.arctic.modeling_arctic_flax) ArcticMoeBlock (class in easydel.modules.arctic.modeling_arctic_flax) Argu() (in module easydel.utils.cli_helpers) arguments (easydel.__init__.DPOTrainer attribute) (easydel.__init__.GRPOTrainer attribute) (easydel.__init__.ORPOTrainer attribute) (easydel.trainers.__init__.DPOTrainer attribute) (easydel.trainers.__init__.GRPOTrainer attribute) (easydel.trainers.__init__.ORPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) AsyncMultifuture (class in easydel.inference.vsurge.utils) attach_custom_arguments() (easydel.__init__.CohereConfig method) (easydel.__init__.EasyDeLBaseConfig method) (easydel.__init__.ExaoneConfig method) (easydel.__init__.FalconConfig method) (easydel.__init__.Gemma2Config method) (easydel.__init__.Gemma3TextConfig method) (easydel.__init__.GemmaConfig method) (easydel.__init__.GPT2Config method) (easydel.__init__.GPTJConfig method) (easydel.__init__.GPTNeoXConfig method) (easydel.__init__.Grok1Config method) (easydel.__init__.InternLM2Config method) (easydel.__init__.LlamaConfig method) (easydel.__init__.MistralConfig method) (easydel.__init__.MixtralConfig method) (easydel.__init__.MptConfig method) (easydel.__init__.Olmo2Config method) (easydel.__init__.OlmoConfig method) (easydel.__init__.OPTConfig method) (easydel.__init__.PhiMoeConfig method) (easydel.__init__.Qwen2Config method) (easydel.__init__.RobertaConfig method) (easydel.__init__.Xerxes2Config method) (easydel.__init__.XerxesConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) (easydel.modules.cohere.__init__.CohereConfig method) (easydel.modules.cohere.cohere_configuration.CohereConfig method) (easydel.modules.exaone.__init__.ExaoneConfig method) (easydel.modules.exaone.exaone_configuration.ExaoneConfig method) (easydel.modules.falcon.__init__.FalconConfig method) (easydel.modules.falcon.falcon_configuration.FalconConfig method) (easydel.modules.gemma.__init__.GemmaConfig method) (easydel.modules.gemma.gemma_configuration.GemmaConfig method) (easydel.modules.gemma2.__init__.Gemma2Config method) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config method) (easydel.modules.gemma3.__init__.Gemma3TextConfig method) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig method) (easydel.modules.gpt2.__init__.GPT2Config method) (easydel.modules.gpt2.gpt2_configuration.GPT2Config method) (easydel.modules.gpt_j.__init__.GPTJConfig method) (easydel.modules.gpt_j.gpt_j_configuration.GPTJConfig method) (easydel.modules.gpt_neox.__init__.GPTNeoXConfig method) (easydel.modules.gpt_neox.gpt_neox_configuration.GPTNeoXConfig method) (easydel.modules.grok_1.__init__.Grok1Config method) (easydel.modules.grok_1.grok_1_configuration.Grok1Config method) (easydel.modules.internlm2.__init__.InternLM2Config method) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config method) (easydel.modules.llama.__init__.LlamaConfig method) (easydel.modules.llama.llama_configuration.LlamaConfig method) (easydel.modules.mistral.__init__.MistralConfig method) (easydel.modules.mistral.mistral_configuration.MistralConfig method) (easydel.modules.mixtral.__init__.MixtralConfig method) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig method) (easydel.modules.mosaic_mpt.__init__.MptConfig method) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig method) (easydel.modules.olmo.__init__.OlmoConfig method) (easydel.modules.olmo.olmo_configuration.OlmoConfig method) (easydel.modules.olmo2.__init__.Olmo2Config method) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config method) (easydel.modules.opt.__init__.OPTConfig method) (easydel.modules.opt.opt_configuration.OPTConfig method) (easydel.modules.phimoe.__init__.PhiMoeConfig method) (easydel.modules.phimoe.phimoe_configuration.PhiMoeConfig method) (easydel.modules.qwen2.__init__.Qwen2Config method) (easydel.modules.qwen2.qwen_configuration.Qwen2Config method) (easydel.modules.roberta.__init__.RobertaConfig method) (easydel.modules.roberta.roberta_configuration.RobertaConfig method) (easydel.modules.xerxes.__init__.XerxesConfig method) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig method) (easydel.modules.xerxes2.__init__.Xerxes2Config method) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config method) attention (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) attention3d_sharding() (in module easydel.infra.utils) attention4d_sharding() (in module easydel.infra.utils) attention_dim_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute)

attention_dropout (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) attention_kv_dim_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) attention_module (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) attention_norm (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) attention_output (easydel.infra.modeling_outputs.AttentionLayerOutput attribute) attention_outputs (easydel.layers.attention_operator.__init__.AttentionOutput attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionOutput attribute), [1] attention_pack_with_static_shape() (in module easydel.kernels.gpu_ops.flash_attention_triton._utils) attention_performer (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) attention_unpack_with_static_shape() (in module easydel.kernels.gpu_ops.flash_attention_triton._utils) attention_weight (easydel.infra.modeling_outputs.AttentionLayerOutput attribute) (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) (easydel.infra.modeling_outputs.EncoderLayerOutput attribute) attention_weights (easydel.layers.attention_operator.__init__.AttentionOutput attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionOutput attribute), [1] AttentionConfig (class in easydel.kernels.flash_attention) AttentionImpl (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator._attention_impl) AttentionLayerOutput (class in easydel.infra.modeling_outputs) AttentionMechanisms (class in easydel.__init__) (class in easydel.layers.attention) AttentionMetadata (class in easydel.__init__) (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator._attention_impl) AttentionModule (class in easydel.__init__) (class in easydel.layers.attention) AttentionOutput (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator._attention_impl) AttentionRegistry (class in easydel.__init__) (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator._attention_impl) attentions (easydel.infra.modeling_outputs.BaseModelOutput attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPast attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.CLIPTextModelOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutput attribute) (easydel.infra.modeling_outputs.MaskedLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput attribute) (easydel.infra.modeling_outputs.NextSentencePredictorOutput attribute) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.TokenClassifierOutput attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput attribute) attn (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) attn_factor (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) (easydel.layers.rotary_embedding.YaRNScalingRotaryEmbedding attribute) attn_mechanism (easydel.utils.readme_generator.ModelInfo attribute) attn_meta (easydel.layers.caching.paged_attention.__init__.ModelInputBatch attribute), [1] attn_norm (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) attribute_map (easydel.__init__.DbrxConfig attribute) (easydel.__init__.FalconConfig attribute) (easydel.__init__.GPT2Config attribute) (easydel.__init__.GPTJConfig attribute) (easydel.__init__.MptConfig attribute) (easydel.__init__.OpenELMConfig attribute) (easydel.__init__.PhiConfig attribute) (easydel.__init__.WhisperConfig attribute) (easydel.modules.dbrx.__init__.DbrxConfig attribute) (easydel.modules.dbrx.dbrx_configuration.DbrxConfig attribute) (easydel.modules.falcon.__init__.FalconConfig attribute) (easydel.modules.falcon.falcon_configuration.FalconConfig attribute) (easydel.modules.gpt2.__init__.GPT2Config attribute) (easydel.modules.gpt2.gpt2_configuration.GPT2Config attribute) (easydel.modules.gpt_j.__init__.GPTJConfig attribute) (easydel.modules.gpt_j.gpt_j_configuration.GPTJConfig attribute) (easydel.modules.mosaic_mpt.__init__.MptConfig attribute) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig attribute) (easydel.modules.openelm.__init__.OpenELMConfig attribute) (easydel.modules.openelm.openelm_configuration.OpenELMConfig attribute) (easydel.modules.phi.__init__.PhiConfig attribute) (easydel.modules.phi.phi_configuration.PhiConfig attribute) (easydel.modules.rwkv.__init__.RwkvConfig attribute) (easydel.modules.rwkv.rwkv_configuration.RwkvConfig attribute) (easydel.modules.whisper.__init__.WhisperConfig attribute) (easydel.modules.whisper.whisper_configuration.WhisperConfig attribute) AUDIO_CLASSIFICATION (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] AUTO (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] AUTO_BIND (easydel.__init__.TaskType attribute) (easydel.infra.factory.TaskType attribute) auto_pytree() (in module easydel.__init__) (in module easydel.infra.__init__) auto_remat() (in module easydel.infra.utils) auto_shard_states (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) AutoEasyDeLConfig (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_configuration) AutoEasyDeLModel (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForCausalLM (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForImageTextToText (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForSeq2SeqLM (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForSequenceClassification (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForSpeechSeq2Seq (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLModelForZeroShotImageClassification (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoEasyDeLVisionModel (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoRegressiveDecodeAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) AutoShardAndGatherFunctions (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_configuration) AutoState (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForCausalLM (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForImageSequenceClassification (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForImageTextToText (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForSeq2SeqLM (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForSpeechSeq2Seq (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateForZeroShotImageClassification (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) AutoStateVisionModel (class in easydel.__init__) (class in easydel.modules.auto.__init__) (class in easydel.modules.auto.auto_modeling) aux_loss (easydel.infra.modeling_outputs.MoeCausalLMOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) aux_loss_enabled (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) auxiliary_load_balancing_loss_func() (in module easydel.infra.loss_utils) available_inference() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) AVAILABLE_ROPE_TYPES (in module easydel.layers.rotary_embedding) available_slots (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] AVERAGE_PER_SEQUENCE (easydel.infra.loss_utils.SpecialLossNormalizingFactor attribute), [1] axis_name (easydel.layers.linear.TensorParallelConfig attribute), [1] AyaVisionCausalLMOutputWithPast (class in easydel.modules.aya_vision.modeling_aya_vision_flax) AyaVisionConfig (class in easydel.__init__) (class in easydel.modules.aya_vision.__init__) (class in easydel.modules.aya_vision.aya_vision_configuration) AyaVisionForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.aya_vision.__init__) (class in easydel.modules.aya_vision.modeling_aya_vision_flax) AyaVisionMultiModalProjector (class in easydel.modules.aya_vision.modeling_aya_vision_flax)

B

Backend (class in easydel.kernels.flash_attention) backend (easydel.__init__.AttentionMetadata attribute), [1] (easydel.__init__.TrainingArguments attribute) (easydel.kernels.flash_attention.AttentionConfig attribute) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) base (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) base_config (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] base_config_key (easydel.__init__.CLIPTextConfig attribute) (easydel.__init__.CLIPVisionConfig attribute) (easydel.__init__.Llama4VisionConfig attribute) (easydel.__init__.SiglipTextConfig attribute) (easydel.__init__.SiglipVisionConfig attribute) (easydel.modules.clip.__init__.CLIPTextConfig attribute) (easydel.modules.clip.__init__.CLIPVisionConfig attribute) (easydel.modules.clip.clip_configuration.CLIPTextConfig attribute) (easydel.modules.clip.clip_configuration.CLIPVisionConfig attribute) (easydel.modules.llama4.__init__.Llama4VisionConfig attribute) (easydel.modules.llama4.llama4_configuration.Llama4VisionConfig attribute) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLVisionConfig attribute) (easydel.modules.siglip.__init__.SiglipTextConfig attribute) (easydel.modules.siglip.__init__.SiglipVisionConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipTextConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipVisionConfig attribute) base_model_prefix (easydel.infra.mixins.__init__.BaseModuleProtocol attribute) (easydel.infra.mixins.__init__.EasyBridgeMixin attribute) (easydel.infra.mixins.__init__.EasyGenerationMixin attribute) (easydel.infra.mixins.bridge.EasyBridgeMixin attribute) (easydel.infra.mixins.generation.EasyGenerationMixin attribute) (easydel.infra.mixins.protocol.BaseModuleProtocol attribute) BASE_MODULE (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] BASE_VISION (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] BaseAutoEasyModel (class in easydel.modules.auto.auto_modeling) BaseAutoEasyState (class in easydel.modules.auto.auto_modeling) BaseCache (class in easydel.layers.caching._abstracts) BaseCacheMetadata (class in easydel.layers.caching._abstracts) BaseCacheView (class in easydel.layers.caching._abstracts) BaseModelOutput (class in easydel.infra.modeling_outputs) BaseModelOutputWithNoAttention (class in easydel.infra.modeling_outputs) BaseModelOutputWithPast (class in easydel.infra.modeling_outputs) BaseModelOutputWithPastAndCrossAttentions (class in easydel.infra.modeling_outputs) BaseModelOutputWithPooling (class in easydel.infra.modeling_outputs) BaseModelOutputWithPoolingAndCrossAttentions (class in easydel.infra.modeling_outputs) BaseModelOutputWithPoolingAndNoAttention (class in easydel.infra.modeling_outputs) BaseModuleProtocol (class in easydel.infra.mixins.__init__) (class in easydel.infra.mixins.protocol) BaseOperation (class in easydel.layers.ops.__init__) (class in easydel.layers.ops._base_operation) BaseProgressBar (class in easydel.trainers.trainer_protocol) BaseRunTimeMetadata (class in easydel.layers.caching._abstracts) BaseTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.base_trainer) BaseTrainerProtocol (class in easydel.trainers.trainer_protocol) basic_attention_refrence() (in module easydel.kernels.gpu_ops.flash_attention_triton._utils) basic_layer_norm() (in module easydel.kernels.rms_norm) batch_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) batch_size (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.metrics.ModelMetadata attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) (easydel.layers.caching.paged_attention.__init__.InferenceScheduler attribute) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute)

BeamSearchOutput (class in easydel.infra.modeling_outputs) BeamSearchState (class in easydel.infra.mixins.generation) begin_index (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor attribute) begin_suppress_tokens (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor attribute) below_or_on_diag() (in module easydel.kernels.cpu_ops.ring_attention_jax._utils) (in module easydel.kernels.tpu_ops.ring_attention_pallas._utils) beta (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] beta_fast (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) (easydel.layers.rotary_embedding.YaRNScalingRotaryEmbedding attribute) beta_slow (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) (easydel.layers.rotary_embedding.YaRNScalingRotaryEmbedding attribute) bias (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) bias_head_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) bias_init (easydel.layers.linear.ParallelLinear attribute) bias_key_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) bind_model_task() (easydel.__init__.AutoEasyDeLConfig static method) (easydel.modules.auto.__init__.AutoEasyDeLConfig static method) (easydel.modules.auto.auto_configuration.AutoEasyDeLConfig static method) block_b (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k_dkv (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k_dq (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k_major (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k_major_dkv (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_k_major_dq (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_q (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_q_dkv (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_q_dq (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_q_major_dkv (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes attribute) block_sparse_moe (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) block_wise_ffn() (in module easydel.infra.utils) blocks (easydel.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) blocks_class (easydel.__init__.InternLM2Model attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) blocksize_b (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] blocksize_k (easydel.__init__.AttentionMetadata attribute), [1] (easydel.kernels.flash_attention.AttentionConfig attribute) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] blocksize_q (easydel.__init__.AttentionMetadata attribute), [1] (easydel.kernels.flash_attention.AttentionConfig attribute) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] BlockSizes (class in easydel.kernels.tpu_ops.ring_attention_pallas._utils) BLOCKWISE (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] blockwise_attn() (in module easydel.layers.attention_operator.modules.ring) bmm() (in module easydel.modules.llama4.modeling_llama4_flax) bos_token_id (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor attribute) (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] break_on_nan (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] build_cache_pos() (easydel.__init__.AttentionModule static method) (easydel.layers.attention.AttentionModule static method) build_contiguous_kv_vectorized() (easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention.PagedAttention method) build_input() (easydel.layers.caching.paged_attention.__init__.ModelIOProcessor class method) build_mpt_alibi_tensor() (in module easydel.modules.mosaic_mpt.modeling_mosaic_flax) build_paged_kv() (easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention.PagedAttention method) build_slope_tensor() (in module easydel.layers.ops._lightning_attention) build_tokenized_answer() (easydel.__init__.ORPOTrainer method) (easydel.trainers.__init__.ORPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer method) built_bloom_alibi() (in module easydel.modules.falcon.modeling_falcon_flax) bulk_insert() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method)

C

cache (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput attribute) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput attribute) cache_compiles() (in module easydel.utils.__init__) (in module easydel.utils.compiling_utils) cache_index (easydel.__init__.AttentionModule attribute) (easydel.layers.attention.AttentionModule attribute) cache_params (easydel.infra.modeling_outputs.MambaCausalLMOutput attribute) (easydel.infra.modeling_outputs.MambaOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output attribute) cache_view (easydel.infra.modeling_outputs.AttentionLayerOutput attribute) (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) cached_key (easydel.__init__.AttentionModule attribute) (easydel.layers.attention.AttentionModule attribute) cached_value (easydel.__init__.AttentionModule attribute) (easydel.layers.attention.AttentionModule attribute) calc_bias_strides() (in module easydel.kernels.gpu_ops.flash_attention_triton._utils) calculate() (easydel.trainers.trainer_protocol.StepMetrics method) calculate_mesh_dimension_size() (in module easydel.kernels.collective_matmul) calculate_number_total_flops() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) can_generate() (easydel.infra.mixins.__init__.EasyBridgeMixin class method) (easydel.infra.mixins.bridge.EasyBridgeMixin class method) cancel() (easydel.inference.vsurge.utils.AsyncMultifuture method) cancelled() (easydel.inference.vsurge.utils.AsyncMultifuture method) canonicalize_dtype() (in module easydel.infra.utils) capture_time() (in module easydel.utils.__init__) (in module easydel.utils.helpers) Carry (class in easydel.layers.attention_operator.modules.ring) causal (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) CAUSAL_LM (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] causal_mask (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder property) CausalLMOutput (in module easydel.infra.modeling_outputs) CausalLMOutputWithCrossAttentions (class in easydel.infra.modeling_outputs) ceildiv() (in module easydel.layers.ops._gla) center_rewards_coefficient (easydel.__init__.RewardConfig attribute) (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) chars_per_token (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) chat_completions() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) ChatCompletionRequest (class in easydel.inference.openai_api_modules) ChatCompletionResponse (class in easydel.inference.openai_api_modules) ChatCompletionResponseChoice (class in easydel.inference.openai_api_modules) ChatCompletionStreamResponse (class in easydel.inference.openai_api_modules) ChatCompletionStreamResponseChoice (class in easydel.inference.openai_api_modules) ChatMessage (class in easydel.inference.openai_api_modules) check_all_devices() (easydel.utils.analyze_memory.SMPMemoryMonitor method) check_bool_flag() (in module easydel.utils.__init__) (in module easydel.utils.helpers) CHECKPOINT_DOTS (easydel.__init__.EasyDeLGradientCheckPointers attribute), [1] (easydel.infra.etils.EasyDeLGradientCheckPointers attribute), [1] CHECKPOINT_DOTS_WITH_NO_BATCH_DMIS (easydel.__init__.EasyDeLGradientCheckPointers attribute), [1] (easydel.infra.etils.EasyDeLGradientCheckPointers attribute), [1] checkpoint_manager (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput attribute) checkpoint_path (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerOutput attribute) CheckpointManager (class in easydel.utils.checkpoint_managers.__init__) (class in easydel.utils.checkpoint_managers.streamer) choices (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) chosen_rewards (easydel.infra.loss_utils.LossMetrics attribute), [1] chunk_idx (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] chunk_iter_with_batch() (in module easydel.inference.vwhisper.__init__) (in module easydel.inference.vwhisper.utils) chunk_size (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] cjit() (in module easydel.utils.__init__) (in module easydel.utils.compiling_utils) classification_problem_type (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] clip_grad (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) clip_loss() (in module easydel.modules.clip.modeling_clip_flax) CLIPAttention (class in easydel.modules.clip.modeling_clip_flax) CLIPConfig (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.clip_configuration) CLIPEncoder (class in easydel.modules.clip.modeling_clip_flax) CLIPEncoderLayer (class in easydel.modules.clip.modeling_clip_flax) CLIPForImageClassification (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.modeling_clip_flax) CLIPMLP (class in easydel.modules.clip.modeling_clip_flax) CLIPModel (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.modeling_clip_flax) CLIPOutput (class in easydel.infra.modeling_outputs) CLIPTextConfig (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.clip_configuration) CLIPTextEmbeddings (class in easydel.modules.clip.modeling_clip_flax) CLIPTextModel (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.modeling_clip_flax) CLIPTextModelOutput (class in easydel.infra.modeling_outputs) CLIPTextModelWithProjection (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.modeling_clip_flax) CLIPTextTransformer (class in easydel.modules.clip.modeling_clip_flax) CLIPVisionConfig (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.clip_configuration) CLIPVisionEmbeddings (class in easydel.modules.clip.modeling_clip_flax) CLIPVisionModel (class in easydel.__init__) (class in easydel.modules.clip.__init__) (class in easydel.modules.clip.modeling_clip_flax) CLIPVisionTransformer (class in easydel.modules.clip.modeling_clip_flax) close() (easydel.inference.vsurge.utils.AsyncMultifuture method) (easydel.trainers.trainer_protocol.BaseProgressBar method) (easydel.trainers.trainer_protocol.JSONProgressBar method) (easydel.trainers.trainer_protocol.NullProgressBar method) (easydel.trainers.trainer_protocol.RichProgressBar method) (easydel.trainers.trainer_protocol.TqdmProgressBar method) Cohere2Attention (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2Block (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2Config (class in easydel.__init__) (class in easydel.modules.cohere2.__init__) (class in easydel.modules.cohere2.cohere2_configuration) Cohere2ForCausalLM (class in easydel.__init__) (class in easydel.modules.cohere2.__init__) (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.cohere2.__init__) (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2LayerNorm (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2MLP (class in easydel.modules.cohere2.modeling_cohere2_flax) Cohere2Model (class in easydel.__init__) (class in easydel.modules.cohere2.__init__) (class in easydel.modules.cohere2.modeling_cohere2_flax) CohereAttention (class in easydel.modules.cohere.modeling_cohere_flax) CohereBlock (class in easydel.modules.cohere.modeling_cohere_flax) CohereConfig (class in easydel.__init__) (class in easydel.modules.cohere.__init__) (class in easydel.modules.cohere.cohere_configuration) CohereForCausalLM (class in easydel.__init__) (class in easydel.modules.cohere.__init__) (class in easydel.modules.cohere.modeling_cohere_flax) CohereForSequenceClassification (class in easydel.__init__) (class in easydel.modules.cohere.__init__) (class in easydel.modules.cohere.modeling_cohere_flax) CohereMLP (class in easydel.modules.cohere.modeling_cohere_flax) CohereModel (class in easydel.__init__) (class in easydel.modules.cohere.__init__) (class in easydel.modules.cohere.modeling_cohere_flax) collective_forward() (easydel.layers.linear.ParallelLinear method) colocated_cpus (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) ColorFormatter (class in easydel.utils.helpers) compilation_time (easydel.inference.vinference.metrics.vInferenceMetrics attribute) CompilationTracker (class in easydel.infra.utils) compile() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.__init__.vSurge method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) compile_aot() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) compile_function() (in module easydel.inference.utilities) (in module easydel.utils.__init__) (in module easydel.utils.compiling_utils) complete (easydel.inference.vsurge.utils.ActiveRequest attribute) complete() (easydel.__init__.vSurge method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) complete_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) completion_tokens (easydel.inference.openai_api_modules.UsageInfo attribute) CompletionLogprobs (class in easydel.inference.openai_api_modules) CompletionRequest (class in easydel.inference.openai_api_modules) CompletionResponse (class in easydel.inference.openai_api_modules) CompletionResponseChoice (class in easydel.inference.openai_api_modules) completions() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) CompletionStreamResponse (class in easydel.inference.openai_api_modules) CompletionStreamResponseChoice (class in easydel.inference.openai_api_modules) compute_basic_frequencies() (in module easydel.layers.rotary_embedding) compute_basic_inv_frequencies() (in module easydel.layers.rotary_embedding) compute_complex_rotary() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) compute_deepseek_frequencies() (in module easydel.layers.rotary_embedding) compute_device_linear_index() (in module easydel.kernels.collective_matmul) compute_dynamic_frequencies() (in module easydel.layers.rotary_embedding) compute_frequencies() (in module easydel.modules.pixtral.modeling_pixtral_flax) compute_heads() (in module easydel.modules.openelm.openelm_configuration) compute_linear_frequencies() (in module easydel.layers.rotary_embedding) compute_llama3_frequencies() (in module easydel.layers.rotary_embedding) compute_llama3_inv_frequencies() (in module easydel.layers.rotary_embedding) compute_loss() (easydel.__init__.CLIPModel method) (easydel.__init__.EasyDeLBaseModule method) (easydel.__init__.WhisperForConditionalGeneration method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (easydel.modules.clip.__init__.CLIPModel method) (easydel.modules.clip.modeling_clip_flax.CLIPModel method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method) compute_per_token_logps() (in module easydel.trainers.group_relative_policy_optimization._fn) compute_phi3_frequencies() (in module easydel.layers.rotary_embedding) compute_prefill_length() (easydel.infra.mixins.__init__.EasyGenerationMixin static method) (easydel.infra.mixins.generation.EasyGenerationMixin static method) compute_reference_log_probs() (easydel.__init__.DPOTrainer method) (easydel.trainers.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer method) compute_slops() (in module easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) compute_weight_stats() (in module easydel.trainers.utils) compute_weighted_cross_entropy() (in module easydel.infra.loss_utils) compute_weighted_cross_entropy_and_accuracy() (in module easydel.infra.loss_utils) compute_yarn_frequencies() (in module easydel.layers.rotary_embedding) compute_yarn_inv_frequencies() (in module easydel.layers.rotary_embedding) concatenate() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) concatenate_to_cache() (easydel.layers.caching.__init__.LightningCacheView method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.__init__.TransformerCacheView method) (easydel.layers.caching._abstracts.BaseCacheView method) (easydel.layers.caching.lightning.__init__.LightningCacheView method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) (easydel.layers.caching.transformer.__init__.TransformerCacheView method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView method) concatenated_forward() (in module easydel.trainers.direct_preference_optimization_trainer._fn) (in module easydel.trainers.odds_ratio_preference_optimization_trainer._fn) concatenated_inputs() (in module easydel.trainers.direct_preference_optimization_trainer._fn) (in module easydel.trainers.odds_ratio_preference_optimization_trainer._fn) config (easydel.__init__.ArcticForCausalLM attribute) (easydel.__init__.ArcticModel attribute) (easydel.__init__.AttentionModule attribute) (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.CLIPForImageClassification attribute) (easydel.__init__.CLIPTextModel attribute) (easydel.__init__.CLIPTextModelWithProjection attribute) (easydel.__init__.CLIPVisionModel attribute) (easydel.__init__.CohereForSequenceClassification attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.GPT2Model attribute) (easydel.__init__.GPTJForCausalLM attribute) (easydel.__init__.GPTJModel attribute) (easydel.__init__.GPTNeoXForCausalLM attribute) (easydel.__init__.GPTNeoXModel attribute) (easydel.__init__.Grok1ForCausalLM attribute) (easydel.__init__.Grok1Model attribute) (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.InternLM2Model attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.Llama4ForSequenceClassification attribute) (easydel.__init__.LlamaForCausalLM attribute) (easydel.__init__.LlamaForSequenceClassification attribute) (easydel.__init__.LlamaModel attribute) (easydel.__init__.MistralForCausalLM attribute) (easydel.__init__.MistralForSequenceClassification attribute) (easydel.__init__.MistralModel attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.MptModel attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.OPTModel attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.StableLmModel attribute) (easydel.infra.factory.ModuleRegistration attribute), [1] (easydel.infra.mixins.__init__.BaseModuleProtocol attribute) (easydel.infra.mixins.__init__.EasyBridgeMixin attribute) (easydel.infra.mixins.__init__.EasyGenerationMixin attribute) (easydel.infra.mixins.bridge.EasyBridgeMixin attribute) (easydel.infra.mixins.generation.EasyGenerationMixin attribute) (easydel.infra.mixins.protocol.BaseModuleProtocol attribute) (easydel.layers.attention.AttentionModule attribute) (easydel.modules.arctic.__init__.ArcticForCausalLM attribute) (easydel.modules.arctic.__init__.ArcticForSequenceClassification attribute) (easydel.modules.arctic.__init__.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticAttention attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForCausalLM attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForSequenceClassification attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute), [1] (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector attribute) (easydel.modules.clip.__init__.CLIPForImageClassification attribute) (easydel.modules.clip.__init__.CLIPTextModel attribute) (easydel.modules.clip.__init__.CLIPTextModelWithProjection attribute) (easydel.modules.clip.__init__.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPAttention attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification attribute) (easydel.modules.clip.modeling_clip_flax.CLIPMLP attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer attribute) (easydel.modules.cohere.__init__.CohereForSequenceClassification attribute) (easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.__init__.GPT2Model attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute), [1] (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model attribute) (easydel.modules.gpt_j.__init__.GPTJForCausalLM attribute) (easydel.modules.gpt_j.__init__.GPTJModel attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXModel attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXAttention attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXBlock attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXMlp attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel attribute) (easydel.modules.grok_1.__init__.Grok1ForCausalLM attribute) (easydel.modules.grok_1.__init__.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1BLockSparseMLP attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1DecoderLayer attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1ForCausalLM attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1SparseMoeBlock attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.llama.__init__.LlamaForCausalLM attribute) (easydel.modules.llama.__init__.LlamaForSequenceClassification attribute) (easydel.modules.llama.__init__.LlamaModel attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForCausalLM attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForSequenceClassification attribute) (easydel.modules.llama.modeling_llama_flax.LlamaModel attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.__init__.Llama4ForSequenceClassification attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute), [1] (easydel.modules.llama4.modeling_llama4_flax.Llama4ForSequenceClassification attribute) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.mistral.__init__.MistralForCausalLM attribute) (easydel.modules.mistral.__init__.MistralForSequenceClassification attribute) (easydel.modules.mistral.__init__.MistralModel attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForCausalLM attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForSequenceClassification attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralModel attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.__init__.OPTModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTModel attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput attribute)

config_class (easydel.infra.mixins.__init__.BaseModuleProtocol attribute) (easydel.infra.mixins.__init__.EasyBridgeMixin attribute) (easydel.infra.mixins.__init__.EasyGenerationMixin attribute) (easydel.infra.mixins.bridge.EasyBridgeMixin attribute) (easydel.infra.mixins.generation.EasyGenerationMixin attribute) (easydel.infra.mixins.protocol.BaseModuleProtocol attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VisionTransformerPretrainedModel attribute) config_prune_kernel() (in module easydel.kernels.gpu_ops.flash_attention_triton._backward_triton) (in module easydel.kernels.gpu_ops.flash_attention_triton._forward_triton) config_registry (easydel.infra.factory.Registry property) ConfigType (class in easydel.__init__) (class in easydel.infra.factory) configure_dataloaders() (easydel.__init__.BaseTrainer method) (easydel.__init__.DPOTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.DPOTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) configure_functions() (easydel.__init__.BaseTrainer method) (easydel.__init__.DPOTrainer method) (easydel.__init__.GRPOTrainer method) (easydel.__init__.ORPOTrainer method) (easydel.__init__.RewardTrainer method) (easydel.__init__.Trainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.DPOTrainer method) (easydel.trainers.__init__.GRPOTrainer method) (easydel.trainers.__init__.ORPOTrainer method) (easydel.trainers.__init__.RewardTrainer method) (easydel.trainers.__init__.Trainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer method) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer method) (easydel.trainers.reward_trainer.__init__.RewardTrainer method) (easydel.trainers.reward_trainer.reward_trainer.RewardTrainer method) (easydel.trainers.trainer.__init__.Trainer method) (easydel.trainers.trainer.trainer.Trainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) configure_model() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) content (easydel.inference.openai_api_modules.ChatMessage attribute) (easydel.inference.openai_api_modules.DeltaMessage attribute) context_lock (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] contrastive_loss() (in module easydel.modules.clip.modeling_clip_flax) conv1 (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) Conv1D (class in easydel.modules.gpt2.modeling_gpt2_flax) (class in easydel.modules.mamba2.modeling_mamba2_flax) conv2 (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) conv_kernel_size (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) conv_states (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) conversation (easydel.inference.openai_api_modules.CountTokenRequest attribute) conversations_formatting_function() (in module easydel.trainers.__init__) (in module easydel.trainers.utils) CONVERT (easydel.infra.base_config.EasyMethod attribute) convert_pytorch_tensor_to_jax() (in module easydel.utils.parameters_transformation) convert_special_loss_normalizing_factor_to_enum() (in module easydel.infra.loss_utils) convert_to_openai_format() (in module easydel.trainers.prompt_utils) copy_decode() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) copy_prefill() (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest method) COSINE (easydel.__init__.EasyDeLSchedulers attribute), [1] (easydel.infra.etils.EasyDeLSchedulers attribute), [1] cost_analysis (easydel.infra.utils.TraceResult property) count_flop_jaxpr() (in module easydel.infra.utils) count_model_parameters() (easydel.__init__.BaseTrainer static method) (easydel.trainers.__init__.BaseTrainer static method) (easydel.trainers.base_trainer.BaseTrainer static method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol static method) count_tokens() (easydel.__init__.vInference method) (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurge method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInference method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurge method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vinference.vinference.vInference method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) (easydel.inference.vsurge.vsurge.vSurge method) CountTokenRequest (class in easydel.inference.openai_api_modules) CPU (easydel.__init__.EasyDeLBackends attribute), [1] (easydel.infra.etils.EasyDeLBackends attribute), [1] (easydel.kernels.flash_attention.Backend attribute) create() (easydel.__init__.AttentionRegistry class method) (easydel.__init__.EasyDeLState class method) (easydel.infra.__init__.EasyDeLState class method) (easydel.infra.base_state.EasyDeLState class method) (easydel.layers.attention_operator.__init__.AttentionRegistry class method) (easydel.layers.attention_operator._attention_impl.AttentionRegistry class method) (easydel.layers.caching.__init__.LightningCacheMetaData class method) (easydel.layers.caching.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.__init__.MambaCacheMetaData class method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.__init__.TransformerCacheMetaData class method) (easydel.layers.caching._abstracts.BaseCacheMetadata class method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData class method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData class method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch class method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData class method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData class method) create_attention_bias (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) create_attention_mask() (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) create_cache_metadata() (easydel.__init__.Xerxes2ForCausalLM method) (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) (easydel.modules.xerxes2.__init__.Xerxes2ForCausalLM method) (easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2ForCausalLM method) create_collect_function() (easydel.__init__.BaseTrainer method) (easydel.__init__.DPOTrainer method) (easydel.__init__.ORPOTrainer method) (easydel.__init__.RewardTrainer method) (easydel.__init__.Trainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.DPOTrainer method) (easydel.trainers.__init__.ORPOTrainer method) (easydel.trainers.__init__.RewardTrainer method) (easydel.trainers.__init__.Trainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer method) (easydel.trainers.reward_trainer.__init__.RewardTrainer method) (easydel.trainers.reward_trainer.reward_trainer.RewardTrainer method) (easydel.trainers.trainer.__init__.Trainer method) (easydel.trainers.trainer.trainer.Trainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) create_constant_length_dataset() (in module easydel.trainers.__init__) (in module easydel.trainers.utils) create_distributed_matmul() (in module easydel.kernels.collective_matmul) create_error_response() (in module easydel.inference.vinference.api_server) (in module easydel.inference.vsurge.api_server) create_flash_attention() (in module easydel.kernels.flash_attention) create_graphdef() (in module easydel.utils.traversals) create_mesh() (easydel.__init__.EasyDeLBaseConfig static method) (easydel.infra.__init__.EasyDeLBaseConfig static method) (easydel.infra.base_config.EasyDeLBaseConfig static method) create_odriver() (easydel.__init__.vSurge class method) (easydel.inference.__init__.vSurge class method) (easydel.inference.vsurge.__init__.vSurge class method) (easydel.inference.vsurge.vsurge.vSurge class method) create_optimized_compo() (easydel.__init__.vInferencePreCompileConfig class method) (easydel.inference.__init__.vInferencePreCompileConfig class method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig class method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig class method) create_paged_metadata() (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) create_plan() (easydel.layers.caching.paged_attention.__init__.InferenceScheduler method) create_progress_bar() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) create_prompt_creator() (in module easydel.trainers.__init__) (in module easydel.trainers.utils) create_sampling_step() (in module easydel.inference.vinference.utilities) create_stable_sharding() (easydel.layers.attention_operator.__init__.AttentionImpl method) (easydel.layers.attention_operator._attention_impl.AttentionImpl method) create_tuple_parser() (in module easydel.modules.mamba.modeling_mamba_flax) (in module easydel.modules.mamba2.modeling_mamba2_flax) create_vdriver() (easydel.__init__.vSurge class method) (easydel.inference.__init__.vSurge class method) (easydel.inference.vsurge.__init__.vSurge class method) (easydel.inference.vsurge.vsurge.vSurge class method) create_whisper_app() (in module easydel.inference.vwhisper.__init__) (in module easydel.inference.vwhisper.server) created (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) cross_attention (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) cross_attentions (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) cross_entropy_loss_and_accuracy() (in module easydel.infra.loss_utils) CUDA_FLASH_ATTN2 (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] CUDNN (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] cur_len (easydel.infra.mixins.generation.BeamSearchState attribute), [1] (easydel.infra.mixins.generation.GreedyState attribute), [1] (easydel.infra.mixins.generation.SampleState attribute), [1] current_backend() (easydel.layers.attention_operator.__init__.AttentionImpl method) (easydel.layers.attention_operator._attention_impl.AttentionImpl method) current_length (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] current_page_index (easydel.layers.caching.__init__.HBMPageManager property) (easydel.layers.caching.paged_attention.__init__.HBMPageManager property) custom_dot_general_kernel() (in module easydel.kernels.matmul) custom_scheduler (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute)

D

data_collator (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) data_parallel_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) dataclass_types (easydel.utils.__init__.DataClassArgumentParser attribute) (easydel.utils.cli_helpers.DataClassArgumentParser attribute) DataClassArgumentParser (class in easydel.utils.__init__) (class in easydel.utils.cli_helpers) DataCollatorForCompletionOnlyLM (class in easydel.trainers.utils) DataCollatorForPreference (class in easydel.trainers.utils) dataloader_eval (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput attribute) dataloader_num_workers (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) dataloader_pin_memory (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) dataloader_train (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput attribute) dataset_batch_size (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) dataset_eval (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) dataset_kwargs (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) dataset_num_proc (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.__init__.RewardConfig attribute) (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) dataset_text_field (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) dataset_train (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) DbrxAttention (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxAttentionConfig (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.dbrx_configuration) DbrxBlock (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxConfig (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.dbrx_configuration) DbrxExpertGLU (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxExperts (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxFFN (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxFFNConfig (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.dbrx_configuration) DbrxForCausalLM (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxForSequenceClassification (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxModel (class in easydel.__init__) (class in easydel.modules.dbrx.__init__) (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxNormAttentionNorm (class in easydel.modules.dbrx.modeling_dbrx_flax) DbrxRouter (class in easydel.modules.dbrx.modeling_dbrx_flax) decode() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.__init__.WhisperForConditionalGeneration method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel method) decode_attention_dim_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_attention_kv_dim_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_batch_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_fn() (in module easydel.inference.vinference._fn) decode_head_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_key_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_kv_head_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_query_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) decode_start_time (easydel.inference.vsurge.utils.ActiveRequest attribute) decoder (easydel.__init__.OPTModel attribute) (easydel.modules.opt.__init__.OPTModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) decoder_attentions (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) decoder_hidden_states (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) DecoderLayerOutput (class in easydel.infra.modeling_outputs) decodes_active_request_ids (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] decodes_active_slots (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] decodes_completes (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] decodes_next_position (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] decodes_page_table (easydel.layers.caching.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata attribute), [1] decodes_position (easydel.layers.caching.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata attribute), [1] decodes_queue (easydel.layers.caching.paged_attention.__init__.InferenceScheduler attribute) decodes_state_page_updates (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] decodes_token_ids (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] deepcopy_model() (in module easydel.utils.traversals) DeepseekScalingRotaryEmbedding (class in easydel.layers.rotary_embedding) DeepseekV2Attention (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV2Config (class in easydel.__init__) (class in easydel.modules.deepseek_v2.__init__) (class in easydel.modules.deepseek_v2.deepseek_configuration) DeepseekV2DecoderLayer (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV2ForCausalLM (class in easydel.__init__) (class in easydel.modules.deepseek_v2.__init__) (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV2MLP (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV2Model (class in easydel.__init__) (class in easydel.modules.deepseek_v2.__init__) (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV2MoE (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) DeepseekV3Attention (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) DeepseekV3Config (class in easydel.__init__) (class in easydel.modules.deepseek_v3.__init__) (class in easydel.modules.deepseek_v3.deepseek_configuration) DeepseekV3DecoderLayer (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) DeepseekV3ForCausalLM (class in easydel.__init__) (class in easydel.modules.deepseek_v3.__init__) (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) DeepseekV3MLP (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) DeepseekV3Model (class in easydel.__init__) (class in easydel.modules.deepseek_v3.__init__) (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) DeepseekV3MoE (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) default_frequencies (easydel.__init__.Gemma3TextModel property) (easydel.modules.gemma3.__init__.Gemma3TextModel property) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3TextModel property) default_key_value_sharding (easydel.__init__.AttentionModule property) (easydel.layers.attention.AttentionModule property) define_flags_with_default() (in module easydel.infra.etils) delete_tree() (in module easydel.trainers.group_relative_policy_optimization.grpo_trainer) delta (easydel.inference.openai_api_modules.ChatCompletionStreamResponseChoice attribute) DeltaMessage (class in easydel.inference.openai_api_modules) denominator (easydel.layers.attention_operator.modules.ring.Carry attribute) dense (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) dequantize_8bit() (in module easydel.layers.quantization.linear_8bit) dequantize_nf4() (in module easydel.layers.quantization._nf4_quantizer) (in module easydel.layers.quantization.linear_nf4) dequantize_row_q8_0() (in module easydel.layers.quantization._8bit_quantizer) description (easydel.inference.openai_api_modules.FunctionDefinition attribute) (easydel.utils.readme_generator.ModelInfo attribute) deterministic (easydel.__init__.FlexibleAttentionModule attribute) (easydel.layers.attention.FlexibleAttentionModule attribute) device_id (easydel.utils.analyze_memory.DeviceStats attribute), [1] DeviceStats (class in easydel.utils.analyze_memory) diffrentiate_state() (in module easydel.utils.traversals) dim (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm attribute) disable_dropout (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.ORPOConfig attribute), [1] (easydel.__init__.RewardConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) divide_weight_sum (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] do_eval (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) do_last_save (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) do_layer_norm_before (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) do_train (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) done() (easydel.inference.vsurge.utils.AsyncMultifuture method) dot_general (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) down_proj (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) DPOConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.direct_preference_optimization_trainer.__init__) (class in easydel.trainers.direct_preference_optimization_trainer.dpo_config) DPODataCollatorWithPadding (class in easydel.trainers.utils) DPOTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.direct_preference_optimization_trainer.__init__) (class in easydel.trainers.direct_preference_optimization_trainer.dpo_trainer) driver (easydel.__init__.vSurge property) (easydel.inference.__init__.vSurge property) (easydel.inference.vsurge.__init__.vSurge property) (easydel.inference.vsurge.vsurge.vSurge property) driver_name (easydel.__init__.oDriver property) (easydel.__init__.vDriver property) (easydel.inference.__init__.oDriver property) (easydel.inference.__init__.vDriver property) (easydel.inference.vsurge.__init__.oDriver property) (easydel.inference.vsurge.__init__.vDriver property) dropout (easydel.__init__.Qwen2Model attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) dropout_add() (in module easydel.modules.falcon.modeling_falcon_flax) dropout_layer (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) dropout_prob (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] dropout_rng_key (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) dtype (easydel.__init__.ArcticForCausalLM attribute) (easydel.__init__.ArcticModel attribute) (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.CLIPForImageClassification attribute) (easydel.__init__.CLIPTextModel attribute) (easydel.__init__.CLIPTextModelWithProjection attribute) (easydel.__init__.CLIPVisionModel attribute) (easydel.__init__.CohereForSequenceClassification attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.GPT2Model attribute) (easydel.__init__.GPTJForCausalLM attribute) (easydel.__init__.GPTJModel attribute) (easydel.__init__.GPTNeoXForCausalLM attribute) (easydel.__init__.GPTNeoXModel attribute) (easydel.__init__.Grok1ForCausalLM attribute) (easydel.__init__.Grok1Model attribute) (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.InternLM2Model attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.Llama4ForSequenceClassification attribute) (easydel.__init__.LlamaForCausalLM attribute) (easydel.__init__.LlamaForSequenceClassification attribute) (easydel.__init__.LlamaModel attribute) (easydel.__init__.MistralForCausalLM attribute) (easydel.__init__.MistralForSequenceClassification attribute) (easydel.__init__.MistralModel attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.MptModel attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.OPTModel attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.StableLmModel attribute) (easydel.inference.vinference.metrics.ModelMetadata attribute), [1] (easydel.layers.linear.ParallelLinear attribute) (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) (easydel.modules.arctic.__init__.ArcticForCausalLM attribute) (easydel.modules.arctic.__init__.ArcticForSequenceClassification attribute) (easydel.modules.arctic.__init__.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticAttention attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForCausalLM attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForSequenceClassification attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute), [1] (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector attribute) (easydel.modules.clip.__init__.CLIPForImageClassification attribute) (easydel.modules.clip.__init__.CLIPTextModel attribute) (easydel.modules.clip.__init__.CLIPTextModelWithProjection attribute) (easydel.modules.clip.__init__.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPAttention attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification attribute) (easydel.modules.clip.modeling_clip_flax.CLIPMLP attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer attribute) (easydel.modules.cohere.__init__.CohereForSequenceClassification attribute) (easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.__init__.GPT2Model attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute), [1] (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model attribute) (easydel.modules.gpt_j.__init__.GPTJForCausalLM attribute) (easydel.modules.gpt_j.__init__.GPTJModel attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXModel attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXAttention attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXBlock attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXMlp attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel attribute) (easydel.modules.grok_1.__init__.Grok1ForCausalLM attribute) (easydel.modules.grok_1.__init__.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1BLockSparseMLP attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1DecoderLayer attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1ForCausalLM attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1SparseMoeBlock attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.llama.__init__.LlamaForCausalLM attribute) (easydel.modules.llama.__init__.LlamaForSequenceClassification attribute) (easydel.modules.llama.__init__.LlamaModel attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForCausalLM attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForSequenceClassification attribute) (easydel.modules.llama.modeling_llama_flax.LlamaModel attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.__init__.Llama4ForSequenceClassification attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute), [1] (easydel.modules.llama4.modeling_llama4_flax.Llama4ForSequenceClassification attribute) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.mistral.__init__.MistralForCausalLM attribute) (easydel.modules.mistral.__init__.MistralForSequenceClassification attribute) (easydel.modules.mistral.__init__.MistralModel attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForCausalLM attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForSequenceClassification attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralModel attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.__init__.OPTModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTModel attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute)

dtype_index() (in module easydel.kernels.gpu_ops._utils) DummyObject (class in easydel.utils.lazy_import) DummyStream (class in easydel.utils.helpers) dynamic_cross_entropy_loss() (in module easydel.infra.loss_utils) DynamicNTKScalingRotaryEmbedding (class in easydel.layers.rotary_embedding)

E

EasyBridgeMixin (class in easydel.infra.mixins.__init__) (class in easydel.infra.mixins.bridge) easydel.__init__ module easydel.inference.__init__ module easydel.inference.logits_process module easydel.inference.openai_api_modules module easydel.inference.utilities module easydel.inference.vinference.__init__ module easydel.inference.vinference._fn module easydel.inference.vinference.api_server module easydel.inference.vinference.metrics module easydel.inference.vinference.utilities module easydel.inference.vinference.vinference module easydel.inference.vsurge.__init__ module easydel.inference.vsurge.api_server module easydel.inference.vsurge.utils module easydel.inference.vsurge.vsurge module easydel.inference.vwhisper.__init__ module easydel.inference.vwhisper.cli module easydel.inference.vwhisper.config module easydel.inference.vwhisper.core module easydel.inference.vwhisper.generation module easydel.inference.vwhisper.server module easydel.inference.vwhisper.utils module easydel.infra.__init__ module easydel.infra.base_config module easydel.infra.base_module module easydel.infra.base_state module easydel.infra.errors module easydel.infra.etils module easydel.infra.factory module easydel.infra.loss_utils module easydel.infra.mixins.__init__ module easydel.infra.mixins.bridge module easydel.infra.mixins.generation module easydel.infra.mixins.protocol module easydel.infra.modeling_outputs module easydel.infra.utils module easydel.kernels.__init__ module easydel.kernels.collective_matmul module easydel.kernels.cpu_ops.__init__ module easydel.kernels.cpu_ops.flash_attention_jax.__init__ module easydel.kernels.cpu_ops.flash_attention_jax._backward_jax module easydel.kernels.cpu_ops.flash_attention_jax._flash_attention module easydel.kernels.cpu_ops.flash_attention_jax._forward_jax module easydel.kernels.cpu_ops.ring_attention_jax.__init__ module easydel.kernels.cpu_ops.ring_attention_jax._backward_jax module easydel.kernels.cpu_ops.ring_attention_jax._forward_jax module easydel.kernels.cpu_ops.ring_attention_jax._ring_attention module easydel.kernels.cpu_ops.ring_attention_jax._utils module easydel.kernels.flash_attention module easydel.kernels.gpu_ops.__init__ module easydel.kernels.gpu_ops._deprecated.pallas_gemm module easydel.kernels.gpu_ops._deprecated.pallas_gqa_flash_attention_2 module easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2 module easydel.kernels.gpu_ops._deprecated.triton_gqa_flash_attention_2 module easydel.kernels.gpu_ops._utils module easydel.kernels.gpu_ops.flash_attention_triton.__init__ module easydel.kernels.gpu_ops.flash_attention_triton._backward_triton module easydel.kernels.gpu_ops.flash_attention_triton._flash_attention module easydel.kernels.gpu_ops.flash_attention_triton._forward_triton module easydel.kernels.gpu_ops.flash_attention_triton._utils module easydel.kernels.gpu_ops.matmul_triton.__init__ module easydel.kernels.gpu_ops.matmul_triton._caller module easydel.kernels.matmul module easydel.kernels.ring_attention module easydel.kernels.rms_norm module easydel.kernels.tpu_ops.__init__ module easydel.kernels.tpu_ops.matmul_pallas.__init__ module easydel.kernels.tpu_ops.matmul_pallas._caller module easydel.kernels.tpu_ops.paged_attention_pallas.__init__ module easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas module easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention module easydel.kernels.tpu_ops.ragged_attention_pallas.__init__ module easydel.kernels.tpu_ops.ragged_attention_pallas._caller module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas module easydel.kernels.tpu_ops.ring_attention_pallas.__init__ module easydel.kernels.tpu_ops.ring_attention_pallas._backward_pallas module easydel.kernels.tpu_ops.ring_attention_pallas._forward_pallas module easydel.kernels.tpu_ops.ring_attention_pallas._ring_attention module easydel.kernels.tpu_ops.ring_attention_pallas._utils module easydel.kernels.utils module easydel.layers.__init__ module easydel.layers.attention module easydel.layers.attention_operator.__init__ module easydel.layers.attention_operator._attention_impl module easydel.layers.attention_operator.modules.__init__ module easydel.layers.attention_operator.modules.flash module easydel.layers.attention_operator.modules.paged_attn module easydel.layers.attention_operator.modules.ring module easydel.layers.attention_operator.modules.scaled_dot_product module easydel.layers.attention_operator.modules.splash module easydel.layers.attention_operator.modules.vanilla module easydel.layers.caching.__init__ module easydel.layers.caching._abstracts module easydel.layers.caching.lightning.__init__ module easydel.layers.caching.lightning.lightning_cache module easydel.layers.caching.mamba.__init__ module easydel.layers.caching.mamba.mamba_cache module easydel.layers.caching.mamba2.__init__ module easydel.layers.caching.mamba2.mamba2_cache module easydel.layers.caching.paged_attention.__init__ module easydel.layers.caching.paged_attention.paged_attention_cache module easydel.layers.caching.transformer.__init__ module easydel.layers.caching.transformer.transformer_cache module easydel.layers.linear module easydel.layers.norms module easydel.layers.ops.__init__ module easydel.layers.ops._base_operation module easydel.layers.ops._gla module easydel.layers.ops._lightning_attention module easydel.layers.quantization._8bit_quantizer module easydel.layers.quantization.__init__ module easydel.layers.quantization._nf4_quantizer module easydel.layers.quantization.base_quant module easydel.layers.quantization.linear_8bit module easydel.layers.quantization.linear_nf4 module easydel.layers.quantization.quantizers module easydel.layers.rotary_embedding module easydel.layers.rotary_embedding_test module easydel.modules.__init__ module easydel.modules.arctic.__init__ module easydel.modules.arctic.arctic_configuration module easydel.modules.arctic.modeling_arctic_flax module easydel.modules.auto.__init__ module easydel.modules.auto.auto_configuration module easydel.modules.auto.auto_modeling module easydel.modules.aya_vision.__init__ module easydel.modules.aya_vision.aya_vision_configuration module easydel.modules.aya_vision.modeling_aya_vision_flax module easydel.modules.clip.__init__ module easydel.modules.clip.clip_configuration module easydel.modules.clip.modeling_clip_flax module easydel.modules.cohere.__init__ module easydel.modules.cohere.cohere_configuration module easydel.modules.cohere.modeling_cohere_flax module easydel.modules.cohere2.__init__ module easydel.modules.cohere2.cohere2_configuration module easydel.modules.cohere2.modeling_cohere2_flax module easydel.modules.dbrx.__init__ module easydel.modules.dbrx.dbrx_configuration module easydel.modules.dbrx.modeling_dbrx_flax module easydel.modules.deepseek_v2.__init__ module easydel.modules.deepseek_v2.deepseek_configuration module easydel.modules.deepseek_v2.modeling_deepseek_flax module easydel.modules.deepseek_v3.__init__ module easydel.modules.deepseek_v3.deepseek_configuration module easydel.modules.deepseek_v3.modeling_deepseek_flax module easydel.modules.exaone.__init__ module easydel.modules.exaone.exaone_configuration module easydel.modules.exaone.modeling_exaone_flax module easydel.modules.falcon.__init__ module easydel.modules.falcon.falcon_configuration module easydel.modules.falcon.modeling_falcon_flax module easydel.modules.gemma.__init__ module easydel.modules.gemma.gemma_configuration module easydel.modules.gemma.modeling_gemma_flax module easydel.modules.gemma2.__init__ module easydel.modules.gemma2.gemma2_configuration module easydel.modules.gemma2.modeling_gemma2_flax module easydel.modules.gemma3.__init__ module easydel.modules.gemma3.gemma3_configuration module easydel.modules.gemma3.modeling_gemma3_flax module easydel.modules.gpt2.__init__ module easydel.modules.gpt2.gpt2_configuration module easydel.modules.gpt2.modeling_gpt2_flax module easydel.modules.gpt_j.__init__ module easydel.modules.gpt_j.gpt_j_configuration module easydel.modules.gpt_j.modeling_gpt_j_flax module easydel.modules.gpt_neox.__init__ module easydel.modules.gpt_neox.gpt_neox_configuration module easydel.modules.gpt_neox.modeling_gpt_neox_flax module easydel.modules.grok_1.__init__ module easydel.modules.grok_1.grok_1_configuration module easydel.modules.grok_1.modeling_grok_1_flax module easydel.modules.internlm2.__init__ module easydel.modules.internlm2.internlm2_configuration module easydel.modules.internlm2.modeling_internlm2_flax module easydel.modules.llama.__init__ module easydel.modules.llama.llama_configuration module easydel.modules.llama.modeling_llama_flax module easydel.modules.llama4.__init__ module easydel.modules.llama4.llama4_configuration module easydel.modules.llama4.modeling_llama4_flax module easydel.modules.llava.__init__ module easydel.modules.llava.llava_configuration module easydel.modules.llava.modeling_llava_flax module easydel.modules.mamba.__init__ module easydel.modules.mamba.mamba_configuration module easydel.modules.mamba.modeling_mamba_flax module easydel.modules.mamba2.__init__ module easydel.modules.mamba2.mamba2_configuration module easydel.modules.mamba2.modeling_mamba2_flax module easydel.modules.minimax_text_v1.__init__ module easydel.modules.minimax_text_v1.minimax_text_01_configuration module easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax module easydel.modules.mistral.__init__ module easydel.modules.mistral.mistral_configuration module easydel.modules.mistral.modeling_mistral_flax module easydel.modules.mixtral.__init__ module easydel.modules.mixtral.mixtral_configuration module easydel.modules.mixtral.modeling_mixtral_flax module easydel.modules.mosaic_mpt.__init__ module easydel.modules.mosaic_mpt.modeling_mosaic_flax module easydel.modules.mosaic_mpt.mosaic_configuration module easydel.modules.olmo.__init__ module easydel.modules.olmo.modeling_olmo_flax module easydel.modules.olmo.olmo_configuration module easydel.modules.olmo2.__init__ module easydel.modules.olmo2.modeling_olmo2_flax module easydel.modules.olmo2.olmo2_configuration module easydel.modules.openelm.__init__ module easydel.modules.openelm.modeling_openelm_flax module easydel.modules.openelm.openelm_configuration module easydel.modules.opt.__init__ module easydel.modules.opt.modeling_opt_flax module easydel.modules.opt.opt_configuration module

easydel.modules.phi.__init__ module easydel.modules.phi.modeling_phi_flax module easydel.modules.phi.phi_configuration module easydel.modules.phi3.__init__ module easydel.modules.phi3.modeling_phi3_flax module easydel.modules.phi3.phi3_configuration module easydel.modules.phimoe.__init__ module easydel.modules.phimoe.modeling_phimoe_flax module easydel.modules.phimoe.phimoe_configuration module easydel.modules.pixtral.__init__ module easydel.modules.pixtral.modeling_pixtral_flax module easydel.modules.pixtral.pixtral_configuration module easydel.modules.qwen2.__init__ module easydel.modules.qwen2.modeling_qwen_flax module easydel.modules.qwen2.qwen_configuration module easydel.modules.qwen2_moe.__init__ module easydel.modules.qwen2_moe.configuration_qwen2_moe module easydel.modules.qwen2_moe.modeling_qwen2_moe_flax module easydel.modules.qwen2_vl.__init__ module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax module easydel.modules.qwen2_vl.qwen2_vl_configuration module easydel.modules.qwen3.__init__ module easydel.modules.qwen3.modeling_qwen3_flax module easydel.modules.qwen3.qwen3_configuration module easydel.modules.qwen3_moe.__init__ module easydel.modules.qwen3_moe.modeling_qwen3_moe_flax module easydel.modules.qwen3_moe.qwen3_moe_configuration module easydel.modules.roberta.__init__ module easydel.modules.roberta.modeling_roberta_flax module easydel.modules.roberta.roberta_configuration module easydel.modules.rwkv.__init__ module easydel.modules.rwkv.modeling_rwkv_flax module easydel.modules.rwkv.rwkv_configuration module easydel.modules.siglip.__init__ module easydel.modules.siglip.configuration_siglip module easydel.modules.siglip.modeling_siglip_flax module easydel.modules.stablelm.__init__ module easydel.modules.stablelm.modeling_stablelm_flax module easydel.modules.stablelm.stablelm_configuration module easydel.modules.whisper.__init__ module easydel.modules.whisper.modeling_whisper_flax module easydel.modules.whisper.whisper_configuration module easydel.modules.xerxes.__init__ module easydel.modules.xerxes.modeling_xerxes_flax module easydel.modules.xerxes.xerxes_configuration module easydel.modules.xerxes2.__init__ module easydel.modules.xerxes2.modeling_xerxes2_flax module easydel.modules.xerxes2.xerxes2_configuration module easydel.scripts.__init__ module easydel.scripts.finetune.__init__ module easydel.scripts.install_on_hosts module easydel.trainers.__init__ module easydel.trainers.base_trainer module easydel.trainers.direct_preference_optimization_trainer.__init__ module easydel.trainers.direct_preference_optimization_trainer._fn module easydel.trainers.direct_preference_optimization_trainer.dpo_config module easydel.trainers.direct_preference_optimization_trainer.dpo_trainer module easydel.trainers.group_relative_policy_optimization.__init__ module easydel.trainers.group_relative_policy_optimization._fn module easydel.trainers.group_relative_policy_optimization.grpo_config module easydel.trainers.group_relative_policy_optimization.grpo_trainer module easydel.trainers.odds_ratio_preference_optimization_trainer.__init__ module easydel.trainers.odds_ratio_preference_optimization_trainer._fn module easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config module easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer module easydel.trainers.packer module easydel.trainers.prompt_utils module easydel.trainers.reward_trainer.__init__ module easydel.trainers.reward_trainer._fn module easydel.trainers.reward_trainer.reward_config module easydel.trainers.reward_trainer.reward_trainer module easydel.trainers.supervised_fine_tuning_trainer.__init__ module easydel.trainers.supervised_fine_tuning_trainer.sft_config module easydel.trainers.supervised_fine_tuning_trainer.sft_trainer module easydel.trainers.trainer.__init__ module easydel.trainers.trainer._fn module easydel.trainers.trainer.modeling_output module easydel.trainers.trainer.trainer module easydel.trainers.trainer_protocol module easydel.trainers.training_configurations module easydel.trainers.training_utils module easydel.trainers.utils module easydel.utils.__init__ module easydel.utils.analyze_memory module easydel.utils.checkpoint_managers.__init__ module easydel.utils.checkpoint_managers.streamer module easydel.utils.cli_helpers module easydel.utils.compiling_utils module easydel.utils.graph_utils module easydel.utils.helpers module easydel.utils.lazy_import module easydel.utils.parameters_transformation module easydel.utils.readme_generator module easydel.utils.rngs_utils module easydel.utils.traversals module EasyDeLBackends (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLBaseConfig (class in easydel.__init__) (class in easydel.infra.__init__) (class in easydel.infra.base_config) EasyDeLBaseConfigDict (class in easydel.__init__) (class in easydel.infra.__init__) (class in easydel.infra.base_config) EasyDeLBaseModule (class in easydel.__init__) (class in easydel.infra.__init__) (class in easydel.infra.base_module) EasyDeLBlockWiseFFNError EasyDeLBreakRequest EasyDeLComputeError EasyDeLGradientCheckPointers (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLNotImplementedFeatureError EasyDeLOptimizers (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLPlatforms (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLProcessError EasyDeLQuantizationMethods (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLRuntimeError, [1] EasyDeLSchedulers (class in easydel.__init__) (class in easydel.infra.etils) EasyDeLState (class in easydel.__init__) (class in easydel.infra.__init__) (class in easydel.infra.base_state) EasyDeLSyntaxRuntimeError, [1] EasyDeLTimerError, [1] EasyGenerationMixin (class in easydel.infra.mixins.__init__) (class in easydel.infra.mixins.generation) EasyMethod (class in easydel.infra.base_config) EasyQuantizer (class in easydel.layers.quantization.__init__) (class in easydel.layers.quantization.quantizers) elapsed_time() (easydel.utils.__init__.Timer method) (easydel.utils.helpers.Timer method) emb_drop (easydel.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) embed_dim (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) embed_dropout (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) embed_positions (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) embed_scale (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) embed_tokens (easydel.__init__.InternLM2Model attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmModel attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) embedding_layer_names (easydel.infra.factory.ModuleRegistration attribute), [1] EmptyProcessor (class in easydel.inference.logits_process) encode() (easydel.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel method) encoder (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) encoder_attentions (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) encoder_attn (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) encoder_attn_layer_norm (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) encoder_hidden_states (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) encoder_last_hidden_state (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) EncoderLayerOutput (class in easydel.infra.modeling_outputs) end_logits (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) EndpointConfig (class in easydel.inference.vinference.api_server) (class in easydel.inference.vsurge.api_server) enqueue_decodes_request() (easydel.layers.caching.paged_attention.__init__.InferenceScheduler method) enqueue_prefill_request() (easydel.layers.caching.paged_attention.__init__.InferenceScheduler method) enqueue_samples() (easydel.inference.vsurge.utils.ActiveRequest method) ensure_checkpoint_path() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) ensure_training_time_limit() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) eos_token_id (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor attribute) (easydel.inference.logits_process.MinLengthLogitsProcessor attribute) (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] eos_token_ids (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) eps (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm attribute) error (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.vinference.PromptOutput attribute) EVAL (easydel.infra.base_config.EasyMethod attribute) eval() (easydel.__init__.Trainer method) (easydel.trainers.__init__.Trainer method) (easydel.trainers.trainer.__init__.Trainer method) (easydel.trainers.trainer.trainer.Trainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) eval_batch_size (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) eval_packing (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) evalu_tracker (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) evaluation_batch_size (easydel.__init__.BaseTrainer property) (easydel.trainers.__init__.BaseTrainer property) (easydel.trainers.base_trainer.BaseTrainer property) (easydel.trainers.trainer_protocol.BaseTrainerProtocol property) evaluation_step() (in module easydel.trainers.direct_preference_optimization_trainer._fn) (in module easydel.trainers.reward_trainer._fn) (in module easydel.trainers.trainer._fn) evaluation_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) EVERYTHING_SAVEABLE (easydel.__init__.EasyDeLGradientCheckPointers attribute), [1] (easydel.infra.etils.EasyDeLGradientCheckPointers attribute), [1] ExaoneAttention (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneAttentionInner (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneConfig (class in easydel.__init__) (class in easydel.modules.exaone.__init__) (class in easydel.modules.exaone.exaone_configuration) ExaoneDecoderLayer (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneForCausalLM (class in easydel.__init__) (class in easydel.modules.exaone.__init__) (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneForSequenceClassification (class in easydel.__init__) (class in easydel.modules.exaone.__init__) (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneGatedMLP (class in easydel.modules.exaone.modeling_exaone_flax) ExaoneModel (class in easydel.__init__) (class in easydel.modules.exaone.__init__) (class in easydel.modules.exaone.modeling_exaone_flax) execute_decode() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) execute_prefill() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) execution_time (easydel.infra.loss_utils.LossMetrics attribute), [1] expand_inputs_for_generation() (in module easydel.inference.vinference._fn) expert_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) expert_gate_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) expert_parallel_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) experts (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) extra_optimizer_kwargs (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute) (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) extract() (easydel.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) extract_prompt() (in module easydel.trainers.prompt_utils) extract_static_parameters() (in module easydel.infra.utils) extrapolation_factor (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) (easydel.layers.rotary_embedding.YaRNScalingRotaryEmbedding attribute)

F

factor (easydel.layers.rotary_embedding.RopeConfig attribute), [1] FalconAttention (class in easydel.modules.falcon.modeling_falcon_flax) FalconBlock (class in easydel.modules.falcon.modeling_falcon_flax) FalconConfig (class in easydel.__init__) (class in easydel.modules.falcon.__init__) (class in easydel.modules.falcon.falcon_configuration) FalconForCausalLM (class in easydel.__init__) (class in easydel.modules.falcon.__init__) (class in easydel.modules.falcon.modeling_falcon_flax) FalconMlp (class in easydel.modules.falcon.modeling_falcon_flax) FalconModel (class in easydel.__init__) (class in easydel.modules.falcon.__init__) (class in easydel.modules.falcon.modeling_falcon_flax) fc1 (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) fc2 (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) features (easydel.utils.readme_generator.ModelInfo attribute) feed_forward (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) ffn (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) ffn_norm (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) ffn_with_glu (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) filter_value (easydel.inference.logits_process.MinPLogitsWarper attribute) (easydel.inference.logits_process.TopKLogitsWarper attribute) (easydel.inference.logits_process.TopPLogitsWarper attribute) final_layer_norm (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) final_layernorm (easydel.__init__.PhiModel attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) finetune (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) finish() (easydel.__init__.BaseTrainer static method) (easydel.trainers.__init__.BaseTrainer static method) (easydel.trainers.base_trainer.BaseTrainer static method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) finish_reason (easydel.inference.openai_api_modules.ChatCompletionResponseChoice attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionStreamResponseChoice attribute) (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.vinference.PromptOutput attribute) fire() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) first_true_indices() (in module easydel.trainers.utils) fixed_cross_entropy() (in module easydel.infra.loss_utils) flash_attention() (in module easydel.kernels.gpu_ops.flash_attention_triton._flash_attention) FLASH_ATTN2 (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] FlashAttention (class in easydel.kernels.flash_attention) FlashAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.flash) flatten_dict() (in module easydel.utils.traversals) flatten_tree() (in module easydel.utils.traversals) FlexibleAttentionModule (class in easydel.__init__) (class in easydel.layers.attention) float() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) float_tensor_to_dtype() (in module easydel.utils.parameters_transformation) flops (easydel.infra.utils.TraceResult property) flush() (easydel.utils.helpers.DummyStream method) fn (easydel.modules.mamba.modeling_mamba_flax.Lambda attribute) ForCausalLMLoss() (in module easydel.infra.loss_utils) force_use_ref_model (easydel.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] ForcedBOSTokenLogitsProcessor (class in easydel.inference.logits_process) ForcedEOSTokenLogitsProcessor (class in easydel.inference.logits_process) ForceTokensLogitsProcessor (class in easydel.inference.logits_process) format() (easydel.utils.helpers.ColorFormatter method) ForQuestionAnsweringLoss() (in module easydel.infra.loss_utils) ForSequenceClassificationLoss() (in module easydel.infra.loss_utils) ForTokenClassification() (in module easydel.infra.loss_utils) forward() (easydel.__init__.FlexibleAttentionModule method) (easydel.__init__.oEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.layers.attention.FlexibleAttentionModule method) forward_cpu() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops._base_operation.BaseOperation method) forward_cuda() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops._base_operation.BaseOperation method) forward_gpu() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops._base_operation.BaseOperation method) forward_native() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops.__init__.RecurrentGLA method) (easydel.layers.ops._base_operation.BaseOperation method) (easydel.layers.ops._gla.RecurrentGLA method) forward_rocm() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops._base_operation.BaseOperation method) forward_tpu() (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) (easydel.layers.ops.__init__.BaseOperation method) (easydel.layers.ops._base_operation.BaseOperation method) fragmentation_ratio (easydel.utils.analyze_memory.DeviceStats attribute), [1] free_gpu_in_process() (in module easydel.kernels.flash_attention) free_hbm_pages() (easydel.layers.caching.__init__.HBMPageManager method) (easydel.layers.caching.paged_attention.__init__.HBMPageManager method) free_resource() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) frequencies (easydel.__init__.DbrxModel property) (easydel.__init__.DeepseekV2Model property) (easydel.__init__.DeepseekV3Model property) (easydel.__init__.EasyDeLBaseModule property) (easydel.__init__.ExaoneModel property) (easydel.__init__.GPTJModel property) (easydel.__init__.GPTNeoXModel property) (easydel.__init__.Grok1Model property) (easydel.__init__.OpenELMModel property) (easydel.__init__.Phi3Model property) (easydel.__init__.PhiModel property) (easydel.__init__.PixtralVisionModel property) (easydel.__init__.StableLmModel property) (easydel.__init__.Xerxes2Model property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.modules.dbrx.__init__.DbrxModel property) (easydel.modules.dbrx.modeling_dbrx_flax.DbrxModel property) (easydel.modules.deepseek_v2.__init__.DeepseekV2Model property) (easydel.modules.deepseek_v2.modeling_deepseek_flax.DeepseekV2Model property) (easydel.modules.deepseek_v3.__init__.DeepseekV3Model property) (easydel.modules.deepseek_v3.modeling_deepseek_flax.DeepseekV3Model property) (easydel.modules.exaone.__init__.ExaoneModel property) (easydel.modules.exaone.modeling_exaone_flax.ExaoneModel property) (easydel.modules.gpt_j.__init__.GPTJModel property) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel property) (easydel.modules.gpt_neox.__init__.GPTNeoXModel property) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel property) (easydel.modules.grok_1.__init__.Grok1Model property) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model property) (easydel.modules.openelm.__init__.OpenELMModel property) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel property) (easydel.modules.phi.__init__.PhiModel property) (easydel.modules.phi.modeling_phi_flax.PhiModel property) (easydel.modules.phi3.__init__.Phi3Model property) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model property) (easydel.modules.pixtral.__init__.PixtralVisionModel property) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel property) (easydel.modules.stablelm.__init__.StableLmModel property) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel property) (easydel.modules.xerxes2.__init__.Xerxes2Model property) (easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2Model property) frequency_penalty (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) FrequencyPenaltyLogitsProcessor (class in easydel.inference.logits_process) from_config() (easydel.__init__.AttentionMetadata class method) (easydel.__init__.AutoShardAndGatherFunctions class method) (easydel.__init__.AutoShardAndGatherFunctions method) (easydel.layers.attention_operator.__init__.AttentionMetadata class method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata class method) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions class method) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions method) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions class method) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions method) (easydel.modules.auto.auto_modeling.BaseAutoEasyModel class method) (easydel.modules.auto.auto_modeling.BaseAutoEasyState class method) from_dict() (easydel.__init__.AttentionMetadata class method) (easydel.__init__.DPOConfig class method) (easydel.__init__.GRPOConfig class method) (easydel.__init__.LossConfig class method) (easydel.__init__.ORPOConfig class method) (easydel.__init__.RewardConfig class method) (easydel.__init__.SamplingParams class method) (easydel.__init__.SFTConfig class method) (easydel.__init__.TrainingArguments class method) (easydel.__init__.vInferenceConfig class method) (easydel.__init__.vInferencePreCompileConfig class method) (easydel.__init__.vWhisperInferenceConfig class method) (easydel.inference.__init__.SamplingParams class method) (easydel.inference.__init__.vInferenceConfig class method) (easydel.inference.__init__.vInferencePreCompileConfig class method) (easydel.inference.__init__.vWhisperInferenceConfig class method) (easydel.inference.logits_process.EmptyProcessor class method) (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor class method) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor class method) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor class method) (easydel.inference.logits_process.MinLengthLogitsProcessor class method) (easydel.inference.logits_process.MinPLogitsWarper class method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor class method) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor class method) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor class method) (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor class method) (easydel.inference.logits_process.SuppressTokensLogitsProcessor class method) (easydel.inference.logits_process.TemperatureLogitsWarper class method) (easydel.inference.logits_process.TopKLogitsWarper class method) (easydel.inference.logits_process.TopPLogitsWarper class method) (easydel.inference.utilities.SamplingParams class method) (easydel.inference.vinference.__init__.SampleState class method) (easydel.inference.vinference.__init__.vInferenceConfig class method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig class method) (easydel.inference.vinference.api_server.EndpointConfig class method) (easydel.inference.vinference.metrics.ModelMetadata class method) (easydel.inference.vinference.utilities.SampleState class method) (easydel.inference.vinference.utilities.vInferenceConfig class method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig class method) (easydel.inference.vsurge.api_server.EndpointConfig class method) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig class method) (easydel.inference.vwhisper.config.vWhisperInferenceConfig class method) (easydel.infra.__init__.LossConfig class method) (easydel.infra.base_config.EasyMethod class method) (easydel.infra.factory.ModuleRegistration class method) (easydel.infra.loss_utils.LossConfig class method) (easydel.infra.loss_utils.LossMetrics class method) (easydel.infra.mixins.generation.BeamSearchState class method) (easydel.infra.mixins.generation.GreedyState class method) (easydel.infra.mixins.generation.SampleState class method) (easydel.infra.modeling_outputs.AttentionLayerOutput class method) (easydel.infra.modeling_outputs.BaseModelOutput class method) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPast class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention class method) (easydel.infra.modeling_outputs.BeamSearchOutput class method) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions class method) (easydel.infra.modeling_outputs.CLIPOutput class method) (easydel.infra.modeling_outputs.CLIPTextModelOutput class method) (easydel.infra.modeling_outputs.DecoderLayerOutput class method) (easydel.infra.modeling_outputs.EncoderLayerOutput class method) (easydel.infra.modeling_outputs.GreedySearchOutput class method) (easydel.infra.modeling_outputs.ImageClassifierOutput class method) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention class method) (easydel.infra.modeling_outputs.MambaCausalLMOutput class method) (easydel.infra.modeling_outputs.MambaOutput class method) (easydel.infra.modeling_outputs.MaskedLMOutput class method) (easydel.infra.modeling_outputs.MoeCausalLMOutput class method) (easydel.infra.modeling_outputs.MoeModelOutput class method) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput class method) (easydel.infra.modeling_outputs.NextSentencePredictorOutput class method) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput class method) (easydel.infra.modeling_outputs.SampleOutput class method) (easydel.infra.modeling_outputs.Seq2SeqLMOutput class method) (easydel.infra.modeling_outputs.Seq2SeqModelOutput class method) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput class method) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput class method) (easydel.infra.modeling_outputs.SequenceClassifierOutput class method) (easydel.infra.modeling_outputs.TokenClassifierOutput class method) (easydel.kernels.flash_attention.AttentionConfig class method) (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes class method) (easydel.layers.attention_operator.__init__.AttentionMetadata class method) (easydel.layers.attention_operator.__init__.AttentionOutput class method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata class method) (easydel.layers.attention_operator._attention_impl.AttentionOutput class method) (easydel.layers.caching.__init__.LightningCache class method) (easydel.layers.caching.__init__.LightningCacheMetaData class method) (easydel.layers.caching.__init__.LightningCacheView class method) (easydel.layers.caching.__init__.Mamba2Cache class method) (easydel.layers.caching.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.__init__.Mamba2CacheView class method) (easydel.layers.caching.__init__.MambaCache class method) (easydel.layers.caching.__init__.MambaCacheMetaData class method) (easydel.layers.caching.__init__.MambaCacheView class method) (easydel.layers.caching.__init__.PagedAttentionCache class method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.__init__.TransformerCache class method) (easydel.layers.caching.__init__.TransformerCacheMetaData class method) (easydel.layers.caching.__init__.TransformerCacheView class method) (easydel.layers.caching.__init__.TransformerMetadata class method) (easydel.layers.caching._abstracts.BaseCacheMetadata class method) (easydel.layers.caching._abstracts.BaseRunTimeMetadata class method) (easydel.layers.caching.lightning.__init__.LightningCache class method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData class method) (easydel.layers.caching.lightning.__init__.LightningCacheView class method) (easydel.layers.caching.lightning.lightning_cache.LightningCache class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView class method) (easydel.layers.caching.mamba.__init__.MambaCache class method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData class method) (easydel.layers.caching.mamba.__init__.MambaCacheView class method) (easydel.layers.caching.mamba.mamba_cache.MambaCache class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView class method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView class method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch class method) (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages class method) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask class method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest class method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch class method) (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary class method) (easydel.layers.caching.paged_attention.__init__.NextIterationPlan class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.paged_attention.__init__.SamplingParams class method) (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata class method) (easydel.layers.caching.transformer.__init__.TransformerCache class method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData class method) (easydel.layers.caching.transformer.__init__.TransformerCacheView class method) (easydel.layers.caching.transformer.__init__.TransformerMetadata class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView class method) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata class method) (easydel.layers.linear.TensorParallelConfig class method) (easydel.layers.rotary_embedding.RopeConfig class method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast class method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast class method) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast class method) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast class method) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput class method) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput class method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput class method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output class method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast class method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput class method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput class method) (easydel.trainers.__init__.DPOConfig class method) (easydel.trainers.__init__.GRPOConfig class method) (easydel.trainers.__init__.ORPOConfig class method) (easydel.trainers.__init__.RewardConfig class method) (easydel.trainers.__init__.SFTConfig class method) (easydel.trainers.__init__.TrainingArguments class method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig class method) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig class method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig class method) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig class method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig class method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig class method) (easydel.trainers.reward_trainer.__init__.RewardConfig class method) (easydel.trainers.reward_trainer.reward_config.RewardConfig class method) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig class method) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig class method) (easydel.trainers.trainer.modeling_output.TrainerOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput class method) (easydel.trainers.trainer_protocol.TrainerOutput class method) (easydel.trainers.training_configurations.TrainingArguments class method) (easydel.trainers.utils.DataCollatorForPreference class method) (easydel.trainers.utils.DPODataCollatorWithPadding class method) (easydel.trainers.utils.RewardDataCollatorWithPadding class method) (easydel.utils.analyze_memory.DeviceStats class method) (easydel.utils.traversals.StateValidationResult class method)

from_json() (easydel.__init__.AttentionMetadata class method) (easydel.__init__.DPOConfig class method) (easydel.__init__.GRPOConfig class method) (easydel.__init__.LossConfig class method) (easydel.__init__.ORPOConfig class method) (easydel.__init__.RewardConfig class method) (easydel.__init__.SamplingParams class method) (easydel.__init__.SFTConfig class method) (easydel.__init__.TrainingArguments class method) (easydel.__init__.vInferenceConfig class method) (easydel.__init__.vInferencePreCompileConfig class method) (easydel.__init__.vWhisperInferenceConfig class method) (easydel.inference.__init__.SamplingParams class method) (easydel.inference.__init__.vInferenceConfig class method) (easydel.inference.__init__.vInferencePreCompileConfig class method) (easydel.inference.__init__.vWhisperInferenceConfig class method) (easydel.inference.logits_process.EmptyProcessor class method) (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor class method) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor class method) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor class method) (easydel.inference.logits_process.MinLengthLogitsProcessor class method) (easydel.inference.logits_process.MinPLogitsWarper class method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor class method) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor class method) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor class method) (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor class method) (easydel.inference.logits_process.SuppressTokensLogitsProcessor class method) (easydel.inference.logits_process.TemperatureLogitsWarper class method) (easydel.inference.logits_process.TopKLogitsWarper class method) (easydel.inference.logits_process.TopPLogitsWarper class method) (easydel.inference.utilities.SamplingParams class method) (easydel.inference.vinference.__init__.SampleState class method) (easydel.inference.vinference.__init__.vInferenceConfig class method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig class method) (easydel.inference.vinference.api_server.EndpointConfig class method) (easydel.inference.vinference.metrics.ModelMetadata class method) (easydel.inference.vinference.utilities.SampleState class method) (easydel.inference.vinference.utilities.vInferenceConfig class method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig class method) (easydel.inference.vsurge.api_server.EndpointConfig class method) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig class method) (easydel.inference.vwhisper.config.vWhisperInferenceConfig class method) (easydel.infra.__init__.LossConfig class method) (easydel.infra.base_config.EasyMethod class method) (easydel.infra.factory.ModuleRegistration class method) (easydel.infra.loss_utils.LossConfig class method) (easydel.infra.loss_utils.LossMetrics class method) (easydel.infra.mixins.generation.BeamSearchState class method) (easydel.infra.mixins.generation.GreedyState class method) (easydel.infra.mixins.generation.SampleState class method) (easydel.infra.modeling_outputs.AttentionLayerOutput class method) (easydel.infra.modeling_outputs.BaseModelOutput class method) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPast class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions class method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention class method) (easydel.infra.modeling_outputs.BeamSearchOutput class method) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions class method) (easydel.infra.modeling_outputs.CLIPOutput class method) (easydel.infra.modeling_outputs.CLIPTextModelOutput class method) (easydel.infra.modeling_outputs.DecoderLayerOutput class method) (easydel.infra.modeling_outputs.EncoderLayerOutput class method) (easydel.infra.modeling_outputs.GreedySearchOutput class method) (easydel.infra.modeling_outputs.ImageClassifierOutput class method) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention class method) (easydel.infra.modeling_outputs.MambaCausalLMOutput class method) (easydel.infra.modeling_outputs.MambaOutput class method) (easydel.infra.modeling_outputs.MaskedLMOutput class method) (easydel.infra.modeling_outputs.MoeCausalLMOutput class method) (easydel.infra.modeling_outputs.MoeModelOutput class method) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput class method) (easydel.infra.modeling_outputs.NextSentencePredictorOutput class method) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput class method) (easydel.infra.modeling_outputs.SampleOutput class method) (easydel.infra.modeling_outputs.Seq2SeqLMOutput class method) (easydel.infra.modeling_outputs.Seq2SeqModelOutput class method) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput class method) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput class method) (easydel.infra.modeling_outputs.SequenceClassifierOutput class method) (easydel.infra.modeling_outputs.TokenClassifierOutput class method) (easydel.kernels.flash_attention.AttentionConfig class method) (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes class method) (easydel.layers.attention_operator.__init__.AttentionMetadata class method) (easydel.layers.attention_operator.__init__.AttentionOutput class method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata class method) (easydel.layers.attention_operator._attention_impl.AttentionOutput class method) (easydel.layers.caching.__init__.LightningCache class method) (easydel.layers.caching.__init__.LightningCacheMetaData class method) (easydel.layers.caching.__init__.LightningCacheView class method) (easydel.layers.caching.__init__.Mamba2Cache class method) (easydel.layers.caching.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.__init__.Mamba2CacheView class method) (easydel.layers.caching.__init__.MambaCache class method) (easydel.layers.caching.__init__.MambaCacheMetaData class method) (easydel.layers.caching.__init__.MambaCacheView class method) (easydel.layers.caching.__init__.PagedAttentionCache class method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.__init__.TransformerCache class method) (easydel.layers.caching.__init__.TransformerCacheMetaData class method) (easydel.layers.caching.__init__.TransformerCacheView class method) (easydel.layers.caching.__init__.TransformerMetadata class method) (easydel.layers.caching._abstracts.BaseCacheMetadata class method) (easydel.layers.caching._abstracts.BaseRunTimeMetadata class method) (easydel.layers.caching.lightning.__init__.LightningCache class method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData class method) (easydel.layers.caching.lightning.__init__.LightningCacheView class method) (easydel.layers.caching.lightning.lightning_cache.LightningCache class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView class method) (easydel.layers.caching.mamba.__init__.MambaCache class method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData class method) (easydel.layers.caching.mamba.__init__.MambaCacheView class method) (easydel.layers.caching.mamba.mamba_cache.MambaCache class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView class method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView class method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch class method) (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages class method) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask class method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest class method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch class method) (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary class method) (easydel.layers.caching.paged_attention.__init__.NextIterationPlan class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.paged_attention.__init__.SamplingParams class method) (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata class method) (easydel.layers.caching.transformer.__init__.TransformerCache class method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData class method) (easydel.layers.caching.transformer.__init__.TransformerCacheView class method) (easydel.layers.caching.transformer.__init__.TransformerMetadata class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView class method) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata class method) (easydel.layers.linear.TensorParallelConfig class method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast class method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast class method) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast class method) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast class method) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput class method) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput class method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput class method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output class method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast class method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput class method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput class method) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput class method) (easydel.trainers.__init__.DPOConfig class method) (easydel.trainers.__init__.GRPOConfig class method) (easydel.trainers.__init__.ORPOConfig class method) (easydel.trainers.__init__.RewardConfig class method) (easydel.trainers.__init__.SFTConfig class method) (easydel.trainers.__init__.TrainingArguments class method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig class method) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig class method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig class method) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig class method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig class method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig class method) (easydel.trainers.reward_trainer.__init__.RewardConfig class method) (easydel.trainers.reward_trainer.reward_config.RewardConfig class method) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig class method) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig class method) (easydel.trainers.trainer.modeling_output.TrainerOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput class method) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput class method) (easydel.trainers.trainer_protocol.TrainerOutput class method) (easydel.trainers.training_configurations.TrainingArguments class method) (easydel.trainers.utils.DataCollatorForPreference class method) (easydel.trainers.utils.DPODataCollatorWithPadding class method) (easydel.trainers.utils.RewardDataCollatorWithPadding class method) (easydel.utils.analyze_memory.DeviceStats class method) (easydel.utils.traversals.StateValidationResult class method) from_linear() (easydel.layers.quantization.__init__.Linear8bit class method) (easydel.layers.quantization.__init__.LinearNF4 class method) (easydel.layers.quantization.linear_8bit.Linear8bit class method) (easydel.layers.quantization.linear_nf4.LinearNF4 class method) from_output() (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary class method) from_params() (easydel.__init__.AutoShardAndGatherFunctions static method) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions static method) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions static method) from_pretrained() (easydel.__init__.AutoEasyDeLConfig class method) (easydel.__init__.AutoShardAndGatherFunctions class method) (easydel.__init__.AutoShardAndGatherFunctions method) (easydel.__init__.DbrxAttentionConfig class method) (easydel.__init__.DbrxFFNConfig class method) (easydel.__init__.EasyDeLBaseConfig class method) (easydel.__init__.MptAttentionConfig class method) (easydel.infra.__init__.EasyDeLBaseConfig class method) (easydel.infra.base_config.EasyDeLBaseConfig class method) (easydel.infra.mixins.__init__.EasyBridgeMixin class method) (easydel.infra.mixins.bridge.EasyBridgeMixin class method) (easydel.modules.auto.__init__.AutoEasyDeLConfig class method) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions class method) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions method) (easydel.modules.auto.auto_configuration.AutoEasyDeLConfig class method) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions class method) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions method) (easydel.modules.auto.auto_modeling.BaseAutoEasyModel class method) (easydel.modules.auto.auto_modeling.BaseAutoEasyState class method) (easydel.modules.dbrx.__init__.DbrxAttentionConfig class method) (easydel.modules.dbrx.__init__.DbrxFFNConfig class method) (easydel.modules.dbrx.dbrx_configuration.DbrxAttentionConfig class method) (easydel.modules.dbrx.dbrx_configuration.DbrxFFNConfig class method) (easydel.modules.mosaic_mpt.__init__.MptAttentionConfig class method) (easydel.modules.mosaic_mpt.mosaic_configuration.MptAttentionConfig class method) from_pure() (easydel.layers.caching.__init__.TransformerCache class method) (easydel.layers.caching.transformer.__init__.TransformerCache class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache class method) from_sampling_params() (easydel.__init__.vSurgeRequest class method) (easydel.inference.__init__.vSurgeRequest class method) (easydel.inference.vsurge.__init__.vSurgeRequest class method) (easydel.inference.vsurge.vsurge.vSurgeRequest class method) from_seed() (easydel.utils.__init__.JaxRNG class method) (easydel.utils.rngs_utils.JaxRNG class method) from_text_vision_configs() (easydel.__init__.CLIPConfig class method) (easydel.__init__.SiglipConfig class method) (easydel.modules.clip.__init__.CLIPConfig class method) (easydel.modules.clip.clip_configuration.CLIPConfig class method) (easydel.modules.siglip.__init__.SiglipConfig class method) (easydel.modules.siglip.configuration_siglip.SiglipConfig class method) from_tuple() (easydel.layers.rotary_embedding.RopeConfig method) frozen_parameters (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) fully_gather() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) fully_shard() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) fully_sharded_data_parallel_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) function (easydel.inference.openai_api_modules.ToolDefinition attribute) function_call (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.ChatMessage attribute) (easydel.inference.openai_api_modules.DeltaMessage attribute) FunctionDefinition (class in easydel.inference.openai_api_modules) functions (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) FunctionTracer (class in easydel.infra.utils)

G

gate (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) gate_loss (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) gate_proj (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) gate_up_proj (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) gather_fns (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) gather_model() (easydel.__init__.EasyDeLBaseModule method) (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.base_state.EasyDeLState method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) gather_optimizer_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) gather_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) Gemma2Attention (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2Config (class in easydel.__init__) (class in easydel.modules.gemma2.__init__) (class in easydel.modules.gemma2.gemma2_configuration) Gemma2DecoderLayer (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2ForCausalLM (class in easydel.__init__) (class in easydel.modules.gemma2.__init__) (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.gemma2.__init__) (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2MLP (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2Model (class in easydel.__init__) (class in easydel.modules.gemma2.__init__) (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma2RMSNorm (class in easydel.modules.gemma2.modeling_gemma2_flax) Gemma3Attention (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3CausalLMOutputWithPast (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3Config (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.gemma3_configuration) Gemma3DecoderLayer (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3ForCausalLM (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3ForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3MLP (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3MultiModalProjector (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3RMSNorm (class in easydel.modules.gemma3.modeling_gemma3_flax) Gemma3TextConfig (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.gemma3_configuration) Gemma3TextModel (class in easydel.__init__) (class in easydel.modules.gemma3.__init__) (class in easydel.modules.gemma3.modeling_gemma3_flax) GemmaAttention (class in easydel.modules.gemma.modeling_gemma_flax) GemmaConfig (class in easydel.__init__) (class in easydel.modules.gemma.__init__) (class in easydel.modules.gemma.gemma_configuration) GemmaDecoderLayer (class in easydel.modules.gemma.modeling_gemma_flax) GemmaForCausalLM (class in easydel.__init__) (class in easydel.modules.gemma.__init__) (class in easydel.modules.gemma.modeling_gemma_flax) GemmaForSequenceClassification (class in easydel.__init__) (class in easydel.modules.gemma.__init__) (class in easydel.modules.gemma.modeling_gemma_flax) GemmaMLP (class in easydel.modules.gemma.modeling_gemma_flax) GemmaModel (class in easydel.__init__) (class in easydel.modules.gemma.__init__) (class in easydel.modules.gemma.modeling_gemma_flax) GemmaRMSNorm (class in easydel.modules.gemma.modeling_gemma_flax) generate() (easydel.__init__.vInference method) (easydel.__init__.vSurge method) (easydel.__init__.vWhisperInference method) (easydel.__init__.WhisperForConditionalGeneration method) (easydel.inference.__init__.vInference method) (easydel.inference.__init__.vSurge method) (easydel.inference.__init__.vWhisperInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) (easydel.inference.vwhisper.__init__.vWhisperInference method) (easydel.inference.vwhisper.core.vWhisperInference method) (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method) generate_block_attention_mask() (in module easydel.modules.pixtral.modeling_pixtral_flax) generate_dequeue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) generate_during_eval (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] generate_enqueue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) generate_readme() (easydel.utils.readme_generator.ReadmeGenerator method) generate_timestep_added (easydel.inference.vsurge.utils.ActiveRequest attribute) generated_tokens (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.inference.vinference.vinference.PromptOutput attribute) GenerateRNG (class in easydel.utils.__init__) (class in easydel.utils.rngs_utils) generation_config (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vinference.vinference.vInferenceMetaData attribute) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) generation_length (easydel.inference.vinference.metrics.vInferenceMetrics attribute) GenerationStepTask (class in easydel.layers.caching.paged_attention.__init__) get() (easydel.__init__.AttentionRegistry class method) (easydel.layers.attention_operator.__init__.AttentionRegistry class method) (easydel.layers.attention_operator._attention_impl.AttentionRegistry class method) get_activation_fn() (in module easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) get_axis_dims() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_axis_names() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_backend() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_basic_causal_mask() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_basic_frequencies() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_basic_inv_frequencies() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_basic_rope() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_batch_logps() (in module easydel.trainers.odds_ratio_preference_optimization_trainer._fn) get_cache_dir() (in module easydel.utils.__init__) (in module easydel.utils.helpers) get_compiled_funcs() (in module easydel.inference.vinference._fn) get_config() (easydel.infra.factory.Registry method) get_count() (easydel.utils.traversals.MetaValueRecreator method) get_decoder() (easydel.__init__.OPTForCausalLM method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) get_decoder_input_ids() (in module easydel.inference.vwhisper.__init__) (in module easydel.inference.vwhisper.generation) get_default() (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes class method) get_default_config() (easydel.__init__.JaxDistributedConfig static method) (easydel.trainers.__init__.JaxDistributedConfig static method) (easydel.trainers.utils.JaxDistributedConfig static method) get_default_hash() (easydel.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) get_device_history() (easydel.utils.analyze_memory.SMPMemoryMonitor method) get_device_memory_usage() (in module easydel.kernels.flash_attention) get_device_stats() (easydel.utils.analyze_memory.MemoryMonitorServer method) get_dot_general_by_bits() (in module easydel.infra.utils) get_dtype() (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VisionTransformerPretrainedModel method) (in module easydel.utils.parameters_transformation) get_fcm_mask() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) get_formatting_func_from_dataset() (in module easydel.trainers.__init__) (in module easydel.trainers.utils) get_frequencies() (in module easydel.layers.rotary_embedding) get_gradient_checkpoint_policy() (in module easydel.infra.utils) get_hash_of_lowering() (in module easydel.utils.compiling_utils) get_image_features() (easydel.__init__.AyaVisionForConditionalGeneration method) (easydel.__init__.CLIPModel method) (easydel.__init__.Gemma3ForConditionalGeneration method) (easydel.__init__.Llama4ForConditionalGeneration method) (easydel.__init__.LlavaForConditionalGeneration method) (easydel.__init__.SiglipModel method) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration method) (easydel.modules.clip.__init__.CLIPModel method) (easydel.modules.clip.modeling_clip_flax.CLIPModel method) (easydel.modules.gemma3.__init__.Gemma3ForConditionalGeneration method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration method) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration method) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration method) (easydel.modules.llava.__init__.LlavaForConditionalGeneration method) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration method) (easydel.modules.siglip.__init__.SiglipModel method) (easydel.modules.siglip.modeling_siglip_flax.SiglipModel method) get_impl_metadata() (easydel.layers.attention_operator.__init__.AttentionImpl method) (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.__init__.FlashAttn method) (easydel.layers.attention_operator.__init__.PagedAttn method) (easydel.layers.attention_operator.__init__.RingAttn method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.__init__.SplashAttn method) (easydel.layers.attention_operator.__init__.VanillaAttn method) (easydel.layers.attention_operator._attention_impl.AttentionImpl method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn method) (easydel.layers.attention_operator.modules.__init__.FlashAttn method) (easydel.layers.attention_operator.modules.__init__.PagedAttn method) (easydel.layers.attention_operator.modules.__init__.RingAttn method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.__init__.SplashAttn method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn method) (easydel.layers.attention_operator.modules.flash.FlashAttn method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn method) (easydel.layers.attention_operator.modules.ring.RingAttn method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn method) (easydel.layers.attention_operator.modules.splash.SplashAttn method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn method) get_impl_name() (easydel.layers.attention_operator.__init__.AttentionImpl class method) (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn class method) (easydel.layers.attention_operator.__init__.FlashAttn class method) (easydel.layers.attention_operator.__init__.PagedAttn class method) (easydel.layers.attention_operator.__init__.RingAttn class method) (easydel.layers.attention_operator.__init__.ScaledDotProductAttn class method) (easydel.layers.attention_operator.__init__.SplashAttn class method) (easydel.layers.attention_operator.__init__.VanillaAttn class method) (easydel.layers.attention_operator._attention_impl.AttentionImpl class method) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn class method) (easydel.layers.attention_operator.modules.__init__.FlashAttn class method) (easydel.layers.attention_operator.modules.__init__.PagedAttn class method) (easydel.layers.attention_operator.modules.__init__.RingAttn class method) (easydel.layers.attention_operator.modules.__init__.ScaledDotProductAttn class method) (easydel.layers.attention_operator.modules.__init__.SplashAttn class method) (easydel.layers.attention_operator.modules.__init__.VanillaAttn class method) (easydel.layers.attention_operator.modules.flash.FlashAttn class method) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn class method) (easydel.layers.attention_operator.modules.ring.RingAttn class method) (easydel.layers.attention_operator.modules.scaled_dot_product.ScaledDotProductAttn class method) (easydel.layers.attention_operator.modules.splash.SplashAttn class method) (easydel.layers.attention_operator.modules.vanilla.VanillaAttn class method) get_input_embeddings() (easydel.__init__.OPTForCausalLM method) (easydel.__init__.OPTModel method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.__init__.OPTModel method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTModel method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) get_inv_frequencies() (in module easydel.layers.rotary_embedding) get_kernel() (easydel.layers.quantization.__init__.Linear8bit method) (easydel.layers.quantization.__init__.LinearNF4 method) (easydel.layers.quantization.linear_8bit.Linear8bit method) (easydel.layers.quantization.linear_nf4.LinearNF4 method) get_leaf_signature() (in module easydel.utils.compiling_utils) get_logger() (in module easydel.utils.__init__) (in module easydel.utils.helpers) get_logits_processor() (easydel.__init__.SamplingParams method) (easydel.inference.__init__.SamplingParams method) (easydel.inference.utilities.SamplingParams method) get_logits_warper() (easydel.__init__.SamplingParams method) (easydel.inference.__init__.SamplingParams method) (easydel.inference.utilities.SamplingParams method) get_loss_function() (in module easydel.trainers.direct_preference_optimization_trainer._fn) get_loss_normalizing_factor_and_weights() (in module easydel.infra.loss_utils) get_matmul_output_sharding() (in module easydel.layers.linear) get_mesh_names() (easydel.__init__.FalconConfig static method) (easydel.__init__.GPTJConfig static method) (easydel.__init__.GPTNeoXConfig static method) (easydel.modules.falcon.__init__.FalconConfig static method) (easydel.modules.falcon.falcon_configuration.FalconConfig static method) (easydel.modules.gpt_j.__init__.GPTJConfig static method) (easydel.modules.gpt_j.gpt_j_configuration.GPTJConfig static method) (easydel.modules.gpt_neox.__init__.GPTNeoXConfig static method) (easydel.modules.gpt_neox.gpt_neox_configuration.GPTNeoXConfig static method) get_mha_cost_estimate() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) get_mode() (easydel.layers.attention_operator.__init__.AttentionImpl method) (easydel.layers.attention_operator._attention_impl.AttentionImpl method) get_module_from_path() (in module easydel.utils.graph_utils) get_module_registration() (easydel.infra.factory.Registry method) get_module_repr() (in module easydel.infra.mixins.protocol) get_modules_by_type() (in module easydel.__init__) (in module easydel.modules.auto.__init__) (in module easydel.modules.auto.auto_configuration) get_optimal_config() (in module easydel.layers.attention) get_optimizer_and_scheduler() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) get_output_embeddings() (easydel.__init__.OPTForCausalLM method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) get_output_partition_spec() (in module easydel.layers.linear) get_partition_rules() (easydel.__init__.ArcticConfig method) (easydel.__init__.AyaVisionConfig method) (easydel.__init__.CLIPConfig method) (easydel.__init__.CLIPTextConfig method) (easydel.__init__.CLIPVisionConfig method) (easydel.__init__.Cohere2Config method) (easydel.__init__.CohereConfig method) (easydel.__init__.DbrxConfig method) (easydel.__init__.DeepseekV2Config method) (easydel.__init__.DeepseekV3Config method) (easydel.__init__.EasyDeLBaseConfig method) (easydel.__init__.ExaoneConfig method) (easydel.__init__.FalconConfig method) (easydel.__init__.Gemma2Config method) (easydel.__init__.Gemma3Config method) (easydel.__init__.Gemma3TextConfig method) (easydel.__init__.GemmaConfig method) (easydel.__init__.GPT2Config method) (easydel.__init__.GPTJConfig method) (easydel.__init__.GPTNeoXConfig method) (easydel.__init__.Grok1Config method) (easydel.__init__.InternLM2Config method) (easydel.__init__.Llama4Config method) (easydel.__init__.Llama4TextConfig method) (easydel.__init__.Llama4VisionConfig method) (easydel.__init__.LlamaConfig method) (easydel.__init__.LlavaConfig method) (easydel.__init__.Mamba2Config method) (easydel.__init__.MambaConfig method) (easydel.__init__.MistralConfig method) (easydel.__init__.MixtralConfig method) (easydel.__init__.MptConfig method) (easydel.__init__.Olmo2Config method) (easydel.__init__.OlmoConfig method) (easydel.__init__.OpenELMConfig method) (easydel.__init__.OPTConfig method) (easydel.__init__.Phi3Config method) (easydel.__init__.PhiConfig method) (easydel.__init__.PhiMoeConfig method) (easydel.__init__.PixtralVisionConfig method) (easydel.__init__.Qwen2Config method) (easydel.__init__.Qwen2MoeConfig method) (easydel.__init__.Qwen2VLConfig method) (easydel.__init__.Qwen3Config method) (easydel.__init__.Qwen3MoeConfig method) (easydel.__init__.RobertaConfig method) (easydel.__init__.SiglipConfig method) (easydel.__init__.SiglipTextConfig method) (easydel.__init__.SiglipVisionConfig method) (easydel.__init__.StableLmConfig method) (easydel.__init__.vInferenceConfig method) (easydel.__init__.WhisperConfig method) (easydel.__init__.Xerxes2Config method) (easydel.__init__.XerxesConfig method) (easydel.inference.__init__.vInferenceConfig method) (easydel.inference.vinference.__init__.vInferenceConfig method) (easydel.inference.vinference.utilities.vInferenceConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) (easydel.modules.arctic.__init__.ArcticConfig method) (easydel.modules.arctic.arctic_configuration.ArcticConfig method) (easydel.modules.aya_vision.__init__.AyaVisionConfig method) (easydel.modules.aya_vision.aya_vision_configuration.AyaVisionConfig method) (easydel.modules.clip.__init__.CLIPConfig method) (easydel.modules.clip.__init__.CLIPTextConfig method) (easydel.modules.clip.__init__.CLIPVisionConfig method) (easydel.modules.clip.clip_configuration.CLIPConfig method) (easydel.modules.clip.clip_configuration.CLIPTextConfig method) (easydel.modules.clip.clip_configuration.CLIPVisionConfig method) (easydel.modules.cohere.__init__.CohereConfig method) (easydel.modules.cohere.cohere_configuration.CohereConfig method) (easydel.modules.cohere2.__init__.Cohere2Config method) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config method) (easydel.modules.dbrx.__init__.DbrxConfig method) (easydel.modules.dbrx.dbrx_configuration.DbrxConfig method) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config method) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config method) (easydel.modules.deepseek_v3.__init__.DeepseekV3Config method) (easydel.modules.deepseek_v3.deepseek_configuration.DeepseekV3Config method) (easydel.modules.exaone.__init__.ExaoneConfig method) (easydel.modules.exaone.exaone_configuration.ExaoneConfig method) (easydel.modules.falcon.__init__.FalconConfig method) (easydel.modules.falcon.falcon_configuration.FalconConfig method) (easydel.modules.gemma.__init__.GemmaConfig method) (easydel.modules.gemma.gemma_configuration.GemmaConfig method) (easydel.modules.gemma2.__init__.Gemma2Config method) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config method) (easydel.modules.gemma3.__init__.Gemma3Config method) (easydel.modules.gemma3.__init__.Gemma3TextConfig method) (easydel.modules.gemma3.gemma3_configuration.Gemma3Config method) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig method) (easydel.modules.gpt2.__init__.GPT2Config method) (easydel.modules.gpt2.gpt2_configuration.GPT2Config method) (easydel.modules.gpt_j.__init__.GPTJConfig method) (easydel.modules.gpt_j.gpt_j_configuration.GPTJConfig method) (easydel.modules.gpt_neox.__init__.GPTNeoXConfig method) (easydel.modules.gpt_neox.gpt_neox_configuration.GPTNeoXConfig method) (easydel.modules.grok_1.__init__.Grok1Config method) (easydel.modules.grok_1.grok_1_configuration.Grok1Config method) (easydel.modules.internlm2.__init__.InternLM2Config method) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config method) (easydel.modules.llama.__init__.LlamaConfig method) (easydel.modules.llama.__init__.VisionLlamaConfig method) (easydel.modules.llama.llama_configuration.LlamaConfig method) (easydel.modules.llama.llama_configuration.VisionLlamaConfig method) (easydel.modules.llama4.__init__.Llama4Config method) (easydel.modules.llama4.__init__.Llama4TextConfig method) (easydel.modules.llama4.__init__.Llama4VisionConfig method) (easydel.modules.llama4.llama4_configuration.Llama4Config method) (easydel.modules.llama4.llama4_configuration.Llama4TextConfig method) (easydel.modules.llama4.llama4_configuration.Llama4VisionConfig method) (easydel.modules.llava.__init__.LlavaConfig method) (easydel.modules.llava.llava_configuration.LlavaConfig method) (easydel.modules.mamba.__init__.MambaConfig method) (easydel.modules.mamba.mamba_configuration.MambaConfig method) (easydel.modules.mamba2.__init__.Mamba2Config method) (easydel.modules.mamba2.mamba2_configuration.Mamba2Config method) (easydel.modules.minimax_text_v1.__init__.MiniMaxText01Config method) (easydel.modules.minimax_text_v1.minimax_text_01_configuration.MiniMaxText01Config method) (easydel.modules.mistral.__init__.MistralConfig method) (easydel.modules.mistral.mistral_configuration.MistralConfig method) (easydel.modules.mixtral.__init__.MixtralConfig method) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig method) (easydel.modules.mosaic_mpt.__init__.MptConfig method) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig method) (easydel.modules.olmo.__init__.OlmoConfig method) (easydel.modules.olmo.olmo_configuration.OlmoConfig method) (easydel.modules.olmo2.__init__.Olmo2Config method) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config method) (easydel.modules.openelm.__init__.OpenELMConfig method) (easydel.modules.openelm.openelm_configuration.OpenELMConfig method) (easydel.modules.opt.__init__.OPTConfig method) (easydel.modules.opt.opt_configuration.OPTConfig method) (easydel.modules.phi.__init__.PhiConfig method) (easydel.modules.phi.phi_configuration.PhiConfig method) (easydel.modules.phi3.__init__.Phi3Config method) (easydel.modules.phi3.phi3_configuration.Phi3Config method) (easydel.modules.phimoe.__init__.PhiMoeConfig method) (easydel.modules.phimoe.phimoe_configuration.PhiMoeConfig method) (easydel.modules.pixtral.__init__.PixtralVisionConfig method) (easydel.modules.pixtral.pixtral_configuration.PixtralVisionConfig method) (easydel.modules.qwen2.__init__.Qwen2Config method) (easydel.modules.qwen2.qwen_configuration.Qwen2Config method) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig method) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig method) (easydel.modules.qwen2_vl.__init__.Qwen2VLConfig method) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLConfig method) (easydel.modules.qwen3.__init__.Qwen3Config method) (easydel.modules.qwen3.qwen3_configuration.Qwen3Config method) (easydel.modules.qwen3_moe.__init__.Qwen3MoeConfig method) (easydel.modules.qwen3_moe.qwen3_moe_configuration.Qwen3MoeConfig method) (easydel.modules.roberta.__init__.RobertaConfig method) (easydel.modules.roberta.roberta_configuration.RobertaConfig method) (easydel.modules.rwkv.__init__.RwkvConfig method) (easydel.modules.rwkv.rwkv_configuration.RwkvConfig method) (easydel.modules.siglip.__init__.SiglipConfig method) (easydel.modules.siglip.__init__.SiglipTextConfig method) (easydel.modules.siglip.__init__.SiglipVisionConfig method) (easydel.modules.siglip.configuration_siglip.SiglipConfig method) (easydel.modules.siglip.configuration_siglip.SiglipTextConfig method) (easydel.modules.siglip.configuration_siglip.SiglipVisionConfig method) (easydel.modules.stablelm.__init__.StableLmConfig method) (easydel.modules.stablelm.stablelm_configuration.StableLmConfig method) (easydel.modules.whisper.__init__.WhisperConfig method) (easydel.modules.whisper.whisper_configuration.WhisperConfig method) (easydel.modules.xerxes.__init__.XerxesConfig method) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig method) (easydel.modules.xerxes2.__init__.Xerxes2Config method) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config method)

get_path() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) get_per_token_logps() (in module easydel.trainers.group_relative_policy_optimization._fn) get_prefix_destination_sharding() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) get_quantized_kernel() (easydel.layers.quantization.__init__.Linear8bit method) (easydel.layers.quantization.__init__.LinearNF4 method) (easydel.layers.quantization.linear_8bit.Linear8bit method) (easydel.layers.quantization.linear_nf4.LinearNF4 method) get_rng() (easydel.utils.traversals.MetaValueRecreator method) get_rope() (in module easydel.layers.rotary_embedding) get_rope_index() (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) get_runstage_flops() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) get_safe_arr() (in module easydel.trainers.training_configurations) get_safe_hash_int() (in module easydel.utils.compiling_utils) get_sharding() (in module easydel.kernels.gpu_ops._utils) (in module easydel.layers.linear) get_sharding_safely() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) get_shardings() (easydel.__init__.AttentionMetadata method) (easydel.layers.attention_operator.__init__.AttentionMetadata method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata method) get_signature_tree_util() (in module easydel.utils.compiling_utils) get_standalones() (easydel.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) get_state_shardings() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) get_static_arguments() (easydel.__init__.EasyDeLBaseModule method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) get_streaming_checkpointer() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) get_stride() (in module easydel.kernels.utils) get_strides() (in module easydel.kernels.gpu_ops._utils) get_summary() (easydel.utils.analyze_memory.SMPMemoryMonitor method) get_tensorboard() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) get_text_features() (easydel.__init__.CLIPModel method) (easydel.__init__.SiglipModel method) (easydel.modules.clip.__init__.CLIPModel method) (easydel.modules.clip.modeling_clip_flax.CLIPModel method) (easydel.modules.siglip.__init__.SiglipModel method) (easydel.modules.siglip.modeling_siglip_flax.SiglipModel method) get_torch() (in module easydel.utils.parameters_transformation) get_torch_loader() (easydel.infra.mixins.__init__.EasyBridgeMixin class method) (easydel.infra.mixins.bridge.EasyBridgeMixin class method) get_total_concurrent_requests() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method) get_wandb_init() (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.training_configurations.TrainingArguments method) get_weight_decay_exclusions() (easydel.__init__.ArcticConfig static method) (easydel.__init__.Cohere2Config static method) (easydel.__init__.CohereConfig static method) (easydel.__init__.DeepseekV2Config static method) (easydel.__init__.ExaoneConfig static method) (easydel.__init__.Gemma2Config static method) (easydel.__init__.Gemma3TextConfig static method) (easydel.__init__.GemmaConfig static method) (easydel.__init__.Grok1Config static method) (easydel.__init__.InternLM2Config static method) (easydel.__init__.LlamaConfig static method) (easydel.__init__.MistralConfig static method) (easydel.__init__.MixtralConfig static method) (easydel.__init__.OpenELMConfig static method) (easydel.__init__.Qwen2Config static method) (easydel.__init__.Qwen2MoeConfig static method) (easydel.__init__.Xerxes2Config static method) (easydel.__init__.XerxesConfig static method) (easydel.modules.arctic.__init__.ArcticConfig static method) (easydel.modules.arctic.arctic_configuration.ArcticConfig static method) (easydel.modules.cohere.__init__.CohereConfig static method) (easydel.modules.cohere.cohere_configuration.CohereConfig static method) (easydel.modules.cohere2.__init__.Cohere2Config static method) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config static method) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config static method) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config static method) (easydel.modules.exaone.__init__.ExaoneConfig static method) (easydel.modules.exaone.exaone_configuration.ExaoneConfig static method) (easydel.modules.gemma.__init__.GemmaConfig static method) (easydel.modules.gemma.gemma_configuration.GemmaConfig static method) (easydel.modules.gemma2.__init__.Gemma2Config static method) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config static method) (easydel.modules.gemma3.__init__.Gemma3TextConfig static method) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig static method) (easydel.modules.grok_1.__init__.Grok1Config static method) (easydel.modules.grok_1.grok_1_configuration.Grok1Config static method) (easydel.modules.internlm2.__init__.InternLM2Config static method) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config static method) (easydel.modules.llama.__init__.LlamaConfig static method) (easydel.modules.llama.llama_configuration.LlamaConfig static method) (easydel.modules.mistral.__init__.MistralConfig static method) (easydel.modules.mistral.mistral_configuration.MistralConfig static method) (easydel.modules.mixtral.__init__.MixtralConfig static method) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig static method) (easydel.modules.openelm.__init__.OpenELMConfig static method) (easydel.modules.openelm.openelm_configuration.OpenELMConfig static method) (easydel.modules.qwen2.__init__.Qwen2Config static method) (easydel.modules.qwen2.qwen_configuration.Qwen2Config static method) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig static method) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig static method) (easydel.modules.xerxes.__init__.XerxesConfig static method) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig static method) (easydel.modules.xerxes2.__init__.Xerxes2Config static method) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config static method) GLU (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) GPT2Attention (class in easydel.modules.gpt2.modeling_gpt2_flax) GPT2Block (class in easydel.modules.gpt2.modeling_gpt2_flax) GPT2Config (class in easydel.__init__) (class in easydel.modules.gpt2.__init__) (class in easydel.modules.gpt2.gpt2_configuration) GPT2LMHeadModel (class in easydel.__init__) (class in easydel.modules.gpt2.__init__) (class in easydel.modules.gpt2.modeling_gpt2_flax) GPT2MLP (class in easydel.modules.gpt2.modeling_gpt2_flax) GPT2Model (class in easydel.__init__) (class in easydel.modules.gpt2.__init__) (class in easydel.modules.gpt2.modeling_gpt2_flax) GPTJAttention (class in easydel.modules.gpt_j.modeling_gpt_j_flax) GPTJBlock (class in easydel.modules.gpt_j.modeling_gpt_j_flax) GPTJConfig (class in easydel.__init__) (class in easydel.modules.gpt_j.__init__) (class in easydel.modules.gpt_j.gpt_j_configuration) GPTJForCausalLM (class in easydel.__init__) (class in easydel.modules.gpt_j.__init__) (class in easydel.modules.gpt_j.modeling_gpt_j_flax) GPTJMLP (class in easydel.modules.gpt_j.modeling_gpt_j_flax) GPTJModel (class in easydel.__init__) (class in easydel.modules.gpt_j.__init__) (class in easydel.modules.gpt_j.modeling_gpt_j_flax) GPTNeoXAttention (class in easydel.modules.gpt_neox.modeling_gpt_neox_flax) GPTNeoXBlock (class in easydel.modules.gpt_neox.modeling_gpt_neox_flax) GPTNeoXConfig (class in easydel.__init__) (class in easydel.modules.gpt_neox.__init__) (class in easydel.modules.gpt_neox.gpt_neox_configuration) GPTNeoXForCausalLM (class in easydel.__init__) (class in easydel.modules.gpt_neox.__init__) (class in easydel.modules.gpt_neox.modeling_gpt_neox_flax) GPTNeoXMlp (class in easydel.modules.gpt_neox.modeling_gpt_neox_flax) GPTNeoXModel (class in easydel.__init__) (class in easydel.modules.gpt_neox.__init__) (class in easydel.modules.gpt_neox.modeling_gpt_neox_flax) GPU (easydel.__init__.EasyDeLBackends attribute), [1] (easydel.infra.etils.EasyDeLBackends attribute), [1] (easydel.kernels.flash_attention.Backend attribute) grad_norms (easydel.infra.loss_utils.LossMetrics attribute), [1] gradient_accumulation_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) gradient_checkpointing (easydel.__init__.InternLM2Model attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmModel attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) granted_freq_max_position_embedding (easydel.__init__.ArcticConfig property) (easydel.__init__.Cohere2Config property) (easydel.__init__.CohereConfig property) (easydel.__init__.DbrxConfig property) (easydel.__init__.DeepseekV2Config property) (easydel.__init__.EasyDeLBaseConfig property) (easydel.__init__.ExaoneConfig property) (easydel.__init__.FalconConfig property) (easydel.__init__.Gemma2Config property) (easydel.__init__.Gemma3TextConfig property) (easydel.__init__.GemmaConfig property) (easydel.__init__.Grok1Config property) (easydel.__init__.InternLM2Config property) (easydel.__init__.LlamaConfig property) (easydel.__init__.MistralConfig property) (easydel.__init__.MixtralConfig property) (easydel.__init__.MptConfig property) (easydel.__init__.Olmo2Config property) (easydel.__init__.OlmoConfig property) (easydel.__init__.OpenELMConfig property) (easydel.__init__.Phi3Config property) (easydel.__init__.PhiConfig property) (easydel.__init__.PhiMoeConfig property) (easydel.__init__.Qwen2Config property) (easydel.__init__.Qwen2MoeConfig property) (easydel.__init__.StableLmConfig property) (easydel.__init__.Xerxes2Config property) (easydel.__init__.XerxesConfig property) (easydel.infra.__init__.EasyDeLBaseConfig property) (easydel.infra.base_config.EasyDeLBaseConfig property) (easydel.modules.arctic.__init__.ArcticConfig property) (easydel.modules.arctic.arctic_configuration.ArcticConfig property) (easydel.modules.cohere.__init__.CohereConfig property) (easydel.modules.cohere.cohere_configuration.CohereConfig property) (easydel.modules.cohere2.__init__.Cohere2Config property) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config property) (easydel.modules.dbrx.__init__.DbrxConfig property) (easydel.modules.dbrx.dbrx_configuration.DbrxConfig property) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config property) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config property) (easydel.modules.exaone.__init__.ExaoneConfig property) (easydel.modules.exaone.exaone_configuration.ExaoneConfig property) (easydel.modules.falcon.__init__.FalconConfig property) (easydel.modules.falcon.falcon_configuration.FalconConfig property) (easydel.modules.gemma.__init__.GemmaConfig property) (easydel.modules.gemma.gemma_configuration.GemmaConfig property) (easydel.modules.gemma2.__init__.Gemma2Config property) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config property) (easydel.modules.gemma3.__init__.Gemma3TextConfig property) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig property) (easydel.modules.grok_1.__init__.Grok1Config property) (easydel.modules.grok_1.grok_1_configuration.Grok1Config property) (easydel.modules.internlm2.__init__.InternLM2Config property) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config property) (easydel.modules.llama.__init__.LlamaConfig property) (easydel.modules.llama.__init__.VisionLlamaConfig property) (easydel.modules.llama.llama_configuration.LlamaConfig property) (easydel.modules.llama.llama_configuration.VisionLlamaConfig property) (easydel.modules.mistral.__init__.MistralConfig property) (easydel.modules.mistral.mistral_configuration.MistralConfig property) (easydel.modules.mixtral.__init__.MixtralConfig property) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig property) (easydel.modules.mosaic_mpt.__init__.MptConfig property) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig property) (easydel.modules.olmo.__init__.OlmoConfig property) (easydel.modules.olmo.olmo_configuration.OlmoConfig property) (easydel.modules.olmo2.__init__.Olmo2Config property) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config property) (easydel.modules.openelm.__init__.OpenELMConfig property) (easydel.modules.openelm.openelm_configuration.OpenELMConfig property) (easydel.modules.phi.__init__.PhiConfig property) (easydel.modules.phi.phi_configuration.PhiConfig property) (easydel.modules.phi3.__init__.Phi3Config property) (easydel.modules.phi3.phi3_configuration.Phi3Config property) (easydel.modules.phimoe.__init__.PhiMoeConfig property) (easydel.modules.phimoe.phimoe_configuration.PhiMoeConfig property) (easydel.modules.qwen2.__init__.Qwen2Config property) (easydel.modules.qwen2.qwen_configuration.Qwen2Config property) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig property) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig property) (easydel.modules.stablelm.__init__.StableLmConfig property) (easydel.modules.stablelm.stablelm_configuration.StableLmConfig property) (easydel.modules.xerxes.__init__.XerxesConfig property) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig property) (easydel.modules.xerxes2.__init__.Xerxes2Config property) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config property) granted_mask_max_position_embedding (easydel.__init__.ArcticConfig property) (easydel.__init__.Cohere2Config property) (easydel.__init__.CohereConfig property) (easydel.__init__.DbrxConfig property) (easydel.__init__.DeepseekV2Config property) (easydel.__init__.EasyDeLBaseConfig property) (easydel.__init__.ExaoneConfig property) (easydel.__init__.FalconConfig property) (easydel.__init__.Gemma2Config property) (easydel.__init__.Gemma3TextConfig property) (easydel.__init__.GemmaConfig property) (easydel.__init__.Grok1Config property) (easydel.__init__.InternLM2Config property) (easydel.__init__.LlamaConfig property) (easydel.__init__.MistralConfig property) (easydel.__init__.MixtralConfig property) (easydel.__init__.MptConfig property) (easydel.__init__.Olmo2Config property) (easydel.__init__.OlmoConfig property) (easydel.__init__.OpenELMConfig property) (easydel.__init__.Phi3Config property) (easydel.__init__.PhiConfig property) (easydel.__init__.PhiMoeConfig property) (easydel.__init__.Qwen2Config property) (easydel.__init__.Qwen2MoeConfig property) (easydel.__init__.StableLmConfig property) (easydel.__init__.Xerxes2Config property) (easydel.__init__.XerxesConfig property) (easydel.infra.__init__.EasyDeLBaseConfig property) (easydel.infra.base_config.EasyDeLBaseConfig property) (easydel.modules.arctic.__init__.ArcticConfig property) (easydel.modules.arctic.arctic_configuration.ArcticConfig property) (easydel.modules.cohere.__init__.CohereConfig property) (easydel.modules.cohere.cohere_configuration.CohereConfig property) (easydel.modules.cohere2.__init__.Cohere2Config property) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config property) (easydel.modules.dbrx.__init__.DbrxConfig property) (easydel.modules.dbrx.dbrx_configuration.DbrxConfig property) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config property) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config property) (easydel.modules.exaone.__init__.ExaoneConfig property) (easydel.modules.exaone.exaone_configuration.ExaoneConfig property) (easydel.modules.falcon.__init__.FalconConfig property) (easydel.modules.falcon.falcon_configuration.FalconConfig property) (easydel.modules.gemma.__init__.GemmaConfig property) (easydel.modules.gemma.gemma_configuration.GemmaConfig property) (easydel.modules.gemma2.__init__.Gemma2Config property) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config property) (easydel.modules.gemma3.__init__.Gemma3TextConfig property) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig property) (easydel.modules.grok_1.__init__.Grok1Config property) (easydel.modules.grok_1.grok_1_configuration.Grok1Config property) (easydel.modules.internlm2.__init__.InternLM2Config property) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config property) (easydel.modules.llama.__init__.LlamaConfig property) (easydel.modules.llama.__init__.VisionLlamaConfig property) (easydel.modules.llama.llama_configuration.LlamaConfig property) (easydel.modules.llama.llama_configuration.VisionLlamaConfig property) (easydel.modules.mistral.__init__.MistralConfig property) (easydel.modules.mistral.mistral_configuration.MistralConfig property) (easydel.modules.mixtral.__init__.MixtralConfig property) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig property) (easydel.modules.mosaic_mpt.__init__.MptConfig property) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig property) (easydel.modules.olmo.__init__.OlmoConfig property) (easydel.modules.olmo.olmo_configuration.OlmoConfig property) (easydel.modules.olmo2.__init__.Olmo2Config property) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config property) (easydel.modules.openelm.__init__.OpenELMConfig property) (easydel.modules.openelm.openelm_configuration.OpenELMConfig property) (easydel.modules.phi.__init__.PhiConfig property) (easydel.modules.phi.phi_configuration.PhiConfig property) (easydel.modules.phi3.__init__.Phi3Config property) (easydel.modules.phi3.phi3_configuration.Phi3Config property) (easydel.modules.phimoe.__init__.PhiMoeConfig property) (easydel.modules.phimoe.phimoe_configuration.PhiMoeConfig property) (easydel.modules.qwen2.__init__.Qwen2Config property) (easydel.modules.qwen2.qwen_configuration.Qwen2Config property) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig property) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig property) (easydel.modules.stablelm.__init__.StableLmConfig property) (easydel.modules.stablelm.stablelm_configuration.StableLmConfig property) (easydel.modules.xerxes.__init__.XerxesConfig property) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig property) (easydel.modules.xerxes2.__init__.Xerxes2Config property) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config property) graphdef (easydel.__init__.EasyDeLBaseModule property) (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.base_state.EasyDeLState attribute), [1] (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) graphother (easydel.__init__.EasyDeLBaseModule property) (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.base_state.EasyDeLState attribute), [1] (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) graphstate (easydel.__init__.EasyDeLBaseModule property) (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.base_state.EasyDeLState attribute), [1] (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) graphtree_params_shape (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) graphtree_shape (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) GreedySearchOutput (class in easydel.infra.modeling_outputs) GreedyState (class in easydel.infra.mixins.generation) Grok1Attention (class in easydel.modules.grok_1.modeling_grok_1_flax) Grok1BLockSparseMLP (class in easydel.modules.grok_1.modeling_grok_1_flax) Grok1Config (class in easydel.__init__) (class in easydel.modules.grok_1.__init__) (class in easydel.modules.grok_1.grok_1_configuration) Grok1DecoderLayer (class in easydel.modules.grok_1.modeling_grok_1_flax) Grok1ForCausalLM (class in easydel.__init__) (class in easydel.modules.grok_1.__init__) (class in easydel.modules.grok_1.modeling_grok_1_flax) Grok1Model (class in easydel.__init__) (class in easydel.modules.grok_1.__init__) (class in easydel.modules.grok_1.modeling_grok_1_flax) Grok1SparseMoeBlock (class in easydel.modules.grok_1.modeling_grok_1_flax) grpo_step() (in module easydel.trainers.group_relative_policy_optimization._fn) GRPOConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.group_relative_policy_optimization.__init__) (class in easydel.trainers.group_relative_policy_optimization.grpo_config) GRPOTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.group_relative_policy_optimization.__init__) (class in easydel.trainers.group_relative_policy_optimization.grpo_trainer)

H

half() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) handler (easydel.inference.vinference.api_server.EndpointConfig attribute) (easydel.inference.vsurge.api_server.EndpointConfig attribute) has_backward_blocks (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes property) hash_fn() (in module easydel.utils.compiling_utils) hbm_utilization (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] HBMPageManager (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.paged_attention.__init__) head_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) head_dim (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) head_size (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) hf_torch_auto_loader (easydel.infra.mixins.__init__.EasyBridgeMixin attribute) (easydel.infra.mixins.bridge.EasyBridgeMixin attribute) hidden_dropout (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) hidden_size (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute)

hidden_state_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) hidden_states (easydel.infra.modeling_outputs.BaseModelOutput attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPast attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention attribute) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.CLIPTextModelOutput attribute) (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) (easydel.infra.modeling_outputs.EncoderLayerOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.MambaCausalLMOutput attribute) (easydel.infra.modeling_outputs.MambaOutput attribute) (easydel.infra.modeling_outputs.MaskedLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput attribute) (easydel.infra.modeling_outputs.NextSentencePredictorOutput attribute) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.TokenClassifierOutput attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput attribute) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput attribute) high_freq_factor (easydel.layers.rotary_embedding.Llama3RotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute), [1] host_memory_used (easydel.inference.vinference.metrics.vInferenceMetrics attribute) hostname (easydel.utils.analyze_memory.DeviceStats attribute), [1]

I

id (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] ids_to_pop_from_dataset (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute) (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) ignore_index (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] IMAGE_CLASSIFICATION (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] image_embeds (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput attribute) image_hidden_states (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) IMAGE_TEXT_TO_TEXT (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] ImageClassifierOutput (class in easydel.infra.modeling_outputs) ImageClassifierOutputWithNoAttention (class in easydel.infra.modeling_outputs) in_compiling_process (easydel.inference.vinference.vinference.vInferenceMetaData attribute) in_features (easydel.layers.linear.ParallelLinear attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) index (easydel.inference.openai_api_modules.ChatCompletionResponseChoice attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionStreamResponseChoice attribute) (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching.__init__.TransformerMetadata attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.__init__.TransformerMetadata attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata attribute) inference_latency (easydel.inference.vinference.metrics.vInferenceMetrics attribute) inference_name (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) (easydel.inference.vinference.vinference.vInferenceMetaData attribute) inference_requests (easydel.inference.vinference.metrics.vInferenceMetrics attribute) InferenceScheduler (class in easydel.layers.caching.paged_attention.__init__) init() (easydel.layers.caching.__init__.LightningCacheView class method) (easydel.layers.caching.__init__.Mamba2CacheView class method) (easydel.layers.caching.__init__.MambaCacheView class method) (easydel.layers.caching.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.__init__.TransformerCacheView class method) (easydel.layers.caching._abstracts.BaseCacheView class method) (easydel.layers.caching.lightning.__init__.LightningCacheView class method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView class method) (easydel.layers.caching.mamba.__init__.MambaCacheView class method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView class method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView class method) (easydel.layers.caching.transformer.__init__.TransformerCacheView class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView class method) init_cache() (easydel.__init__.AyaVisionForConditionalGeneration method) (easydel.__init__.Gemma3ForConditionalGeneration method) (easydel.__init__.Llama4ForConditionalGeneration method) (easydel.__init__.LlavaForConditionalGeneration method) (easydel.__init__.Mamba2ForCausalLM method) (easydel.__init__.MambaForCausalLM method) (easydel.__init__.MambaModel method) (easydel.__init__.Xerxes2ForCausalLM method) (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) (easydel.layers.caching.__init__.LightningCache class method) (easydel.layers.caching.__init__.Mamba2Cache class method) (easydel.layers.caching.__init__.MambaCache class method) (easydel.layers.caching.__init__.PagedAttentionCache class method) (easydel.layers.caching.__init__.TransformerCache class method) (easydel.layers.caching._abstracts.BaseCache class method) (easydel.layers.caching.lightning.__init__.LightningCache class method) (easydel.layers.caching.lightning.lightning_cache.LightningCache class method) (easydel.layers.caching.mamba.__init__.MambaCache class method) (easydel.layers.caching.mamba.mamba_cache.MambaCache class method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache class method) (easydel.layers.caching.transformer.__init__.TransformerCache class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache class method) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration method) (easydel.modules.gemma3.__init__.Gemma3ForConditionalGeneration method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration method) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration method) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration method) (easydel.modules.llava.__init__.LlavaForConditionalGeneration method) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration method) (easydel.modules.mamba.__init__.MambaForCausalLM method) (easydel.modules.mamba.__init__.MambaModel method) (easydel.modules.mamba.modeling_mamba_flax.MambaForCausalLM method) (easydel.modules.mamba.modeling_mamba_flax.MambaModel method) (easydel.modules.mamba2.__init__.Mamba2ForCausalLM method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2ForCausalLM method) (easydel.modules.xerxes2.__init__.Xerxes2ForCausalLM method) (easydel.modules.xerxes2.modeling_xerxes2_flax.Xerxes2ForCausalLM method) init_decode_state() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) init_deepseek_rotary_embedding() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) init_empty() (easydel.layers.caching.__init__.LightningCache class method) (easydel.layers.caching.__init__.Mamba2Cache class method) (easydel.layers.caching.__init__.MambaCache class method) (easydel.layers.caching.__init__.PagedAttentionCache method) (easydel.layers.caching.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.__init__.TransformerCache class method) (easydel.layers.caching._abstracts.BaseCache class method) (easydel.layers.caching.lightning.__init__.LightningCache class method) (easydel.layers.caching.lightning.lightning_cache.LightningCache class method) (easydel.layers.caching.mamba.__init__.MambaCache class method) (easydel.layers.caching.mamba.mamba_cache.MambaCache class method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache class method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache class method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch class method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest class method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch class method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata class method) (easydel.layers.caching.paged_attention.__init__.SamplingParams class method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata class method) (easydel.layers.caching.transformer.__init__.TransformerCache class method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache class method) init_garphstate() (in module easydel.utils.traversals) init_jax() (easydel.layers.caching.paged_attention.__init__.SamplingParams class method)

init_numpy() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch class method) (easydel.layers.caching.paged_attention.__init__.SamplingParams class method) init_pages() (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) init_state() (in module easydel.modules.rwkv.modeling_rwkv_flax) init_to_value() (in module easydel.modules.mamba.modeling_mamba_flax) (in module easydel.modules.mamba2.modeling_mamba2_flax) init_tx (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) init_tx() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) initialize() (easydel.__init__.JaxDistributedConfig class method) (easydel.trainers.__init__.JaxDistributedConfig class method) (easydel.trainers.utils.JaxDistributedConfig class method) initialize_trainer_utils() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) InitialSequenceRequest (class in easydel.layers.caching.paged_attention.__init__) input_ids (easydel.layers.caching.paged_attention.__init__.ModelInputBatch attribute), [1] input_layernorm (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) input_partition_spec (easydel.inference.vinference.vinference.vInferenceMetaData attribute) insert() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) insert_decode_state() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (easydel.layers.caching.paged_attention.__init__.SamplingParams method) insert_from_task() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (easydel.layers.caching.paged_attention.__init__.SamplingParams method) insert_index() (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) insert_starts() (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) instructions_formatting_function() (in module easydel.trainers.__init__) (in module easydel.trainers.utils) int_key_to_string() (in module easydel.utils.traversals) intermediate_size (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) InternLM2Attention (class in easydel.modules.internlm2.modeling_internlm2_flax) InternLM2Block (class in easydel.modules.internlm2.modeling_internlm2_flax) InternLM2Config (class in easydel.__init__) (class in easydel.modules.internlm2.__init__) (class in easydel.modules.internlm2.internlm2_configuration) InternLM2ForCausalLM (class in easydel.__init__) (class in easydel.modules.internlm2.__init__) (class in easydel.modules.internlm2.modeling_internlm2_flax) InternLM2ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.internlm2.__init__) (class in easydel.modules.internlm2.modeling_internlm2_flax) InternLM2MLP (class in easydel.modules.internlm2.modeling_internlm2_flax) InternLM2Model (class in easydel.__init__) (class in easydel.modules.internlm2.__init__) (class in easydel.modules.internlm2.modeling_internlm2_flax) interpolate() (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionEmbeddings method) inv_frequencies (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) invalid_types (easydel.utils.traversals.StateValidationResult attribute) is_active (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch property) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest property) is_byte_token() (in module easydel.inference.vsurge.utils) is_causal (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) is_client_side_tokenization (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) is_conversational() (in module easydel.trainers.prompt_utils) is_cross_attention (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) is_decode_mode() (easydel.layers.caching.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata method) is_empty (easydel.layers.caching.__init__.TransformerCacheView property) (easydel.layers.caching.transformer.__init__.TransformerCacheView property) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView property) is_encoder_decoder (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.utils.DataCollatorForPreference attribute) (easydel.trainers.utils.DPODataCollatorWithPadding attribute) is_fine_tuning (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) is_flatten() (in module easydel.infra.utils) (in module easydel.modules.auto.auto_configuration) (in module easydel.utils.traversals) is_iterable() (in module easydel.utils.traversals) is_jit_wrapped() (in module easydel.utils.compiling_utils) is_multilingual (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) is_neox_style (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) is_package_available() (in module easydel.utils.__init__) (in module easydel.utils.lazy_import) is_prefill_mode() (easydel.layers.caching.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata method) is_process_zero (easydel.__init__.BaseTrainer property) (easydel.__init__.TrainingArguments property) (easydel.trainers.__init__.BaseTrainer property) (easydel.trainers.__init__.TrainingArguments property) (easydel.trainers.base_trainer.BaseTrainer property) (easydel.trainers.training_configurations.TrainingArguments property) is_residual_mlp (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) is_sent_finished (easydel.infra.mixins.generation.BeamSearchState attribute), [1] (easydel.infra.mixins.generation.GreedyState attribute), [1] (easydel.infra.mixins.generation.SampleState attribute), [1] is_sequence_finished (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] is_valid (easydel.utils.traversals.StateValidationResult attribute) items() (easydel.layers.rotary_embedding.RopeConfig method) iter_module_search() (in module easydel.utils.graph_utils)

J

JAX (easydel.__init__.EasyDeLPlatforms attribute), [1] (easydel.infra.etils.EasyDeLPlatforms attribute), [1] (easydel.kernels.flash_attention.Platform attribute) jax2pt() (in module easydel.utils.parameters_transformation) jax_call() (easydel.trainers.utils.DataCollatorForCompletionOnlyLM method) jax_distributed_config (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) jax_flash_attention() (in module easydel.kernels.cpu_ops.__init__) (in module easydel.kernels.cpu_ops.flash_attention_jax.__init__) jax_flash_attn_2_mu() (in module easydel.kernels.cpu_ops.flash_attention_jax._flash_attention) jax_mask_tokens() (easydel.trainers.utils.DataCollatorForCompletionOnlyLM method)

jax_memory_used (easydel.inference.vinference.metrics.vInferenceMetrics attribute) jax_mesh() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) jax_scatter() (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) JaxDistributedConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.utils) JaxRNG (class in easydel.utils.__init__) (class in easydel.utils.rngs_utils) jitter() (easydel.modules.dbrx.modeling_dbrx_flax.DbrxRouter method) json (easydel.inference.vwhisper.server.ResponseFormat attribute) JSONProgressBar (class in easydel.trainers.trainer_protocol)

K

k_layernorm (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) k_norm (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) k_proj (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) kernel_init (easydel.layers.linear.ParallelLinear attribute) key (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) key_dim (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) key_heads (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute)

key_pages (easydel.layers.caching.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView attribute), [1] key_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) key_value (easydel.layers.caching.__init__.LightningCacheView attribute) (easydel.layers.caching.lightning.__init__.LightningCacheView attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView attribute) keys() (easydel.layers.rotary_embedding.RopeConfig method) keys_to_ignore_at_inference (easydel.__init__.DeepseekV3Config attribute) (easydel.__init__.GPT2Config attribute) (easydel.__init__.Olmo2Config attribute) (easydel.__init__.OPTConfig attribute) (easydel.__init__.Qwen2VLConfig attribute) (easydel.modules.deepseek_v3.__init__.DeepseekV3Config attribute) (easydel.modules.deepseek_v3.deepseek_configuration.DeepseekV3Config attribute) (easydel.modules.gpt2.__init__.GPT2Config attribute) (easydel.modules.gpt2.gpt2_configuration.GPT2Config attribute) (easydel.modules.minimax_text_v1.__init__.MiniMaxText01Config attribute) (easydel.modules.minimax_text_v1.minimax_text_01_configuration.MiniMaxText01Config attribute) (easydel.modules.olmo2.__init__.Olmo2Config attribute) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config attribute) (easydel.modules.opt.__init__.OPTConfig attribute) (easydel.modules.opt.opt_configuration.OPTConfig attribute) (easydel.modules.qwen2_vl.__init__.Qwen2VLConfig attribute) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLConfig attribute) kv (easydel.kernels.tpu_ops.ring_attention_pallas._utils.SegmentIds attribute), [1] kv_head_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) kv_head_dim_size (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1]

L

label_pad_token_id (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.utils.DataCollatorForPreference attribute) (easydel.trainers.utils.DPODataCollatorWithPadding attribute) label_smoothing (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] Lambda (class in easydel.modules.mamba.modeling_mamba_flax) language (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) last_hidden_state (easydel.infra.modeling_outputs.BaseModelOutput attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPast attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention attribute) (easydel.infra.modeling_outputs.CLIPTextModelOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutput attribute) (easydel.infra.modeling_outputs.MambaOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput attribute) last_save_file_name (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) (easydel.trainers.trainer_protocol.TrainerOutput attribute) layer_idx (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) layer_index (easydel.layers.caching.__init__.LightningCacheView attribute) (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.__init__.MambaCacheView attribute) (easydel.layers.caching.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching._abstracts.BaseCacheView attribute) (easydel.layers.caching.lightning.__init__.LightningCacheView attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView attribute) (easydel.layers.caching.mamba.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) layer_norm (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) layernorm_names (easydel.infra.factory.ModuleRegistration attribute), [1] layers (easydel.__init__.InternLM2Model attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmModel attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) lazy_init() (easydel.__init__.EasyDeLBaseModule class method) (easydel.infra.__init__.EasyDeLBaseModule class method) (easydel.infra.base_module.EasyDeLBaseModule class method) (easydel.infra.mixins.__init__.BaseModuleProtocol class method) (easydel.infra.mixins.protocol.BaseModuleProtocol class method) LazyLogger (class in easydel.utils.helpers) LazyModule (class in easydel.utils.__init__) (class in easydel.utils.lazy_import) learning_rate (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.__init__.SFTConfig attribute) (easydel.__init__.TrainingArguments attribute) (easydel.infra.loss_utils.LossMetrics attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) learning_rate_end (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) leave_alone_context_manager() (in module easydel.trainers.utils) length (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] lightning_attention() (in module easydel.layers.ops._lightning_attention) LightningCache (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.lightning.__init__) (class in easydel.layers.caching.lightning.lightning_cache) LightningCacheMetaData (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.lightning.__init__) (class in easydel.layers.caching.lightning.lightning_cache) LightningCacheView (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.lightning.__init__) (class in easydel.layers.caching.lightning.lightning_cache) LightningMetadata (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.lightning.__init__) (class in easydel.layers.caching.lightning.lightning_cache) limitations (easydel.utils.readme_generator.ModelInfo attribute) LINEAR (easydel.__init__.EasyDeLSchedulers attribute), [1] (easydel.infra.etils.EasyDeLSchedulers attribute), [1] Linear8bit (class in easydel.layers.quantization.__init__) (class in easydel.layers.quantization.linear_8bit) LinearNF4 (class in easydel.layers.quantization.__init__) (class in easydel.layers.quantization.linear_nf4) LinearScalingRotaryEmbedding (class in easydel.layers.rotary_embedding) LION (easydel.__init__.EasyDeLOptimizers attribute), [1] (easydel.infra.etils.EasyDeLOptimizers attribute), [1] list_implementations() (easydel.__init__.AttentionRegistry class method) (easydel.layers.attention_operator.__init__.AttentionRegistry class method) (easydel.layers.attention_operator._attention_impl.AttentionRegistry class method) liveness() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) Llama3RotaryEmbedding (class in easydel.layers.rotary_embedding) Llama4CausalLMOutputWithPast (class in easydel.modules.llama4.modeling_llama4_flax) Llama4Config (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.llama4_configuration) Llama4ForCausalLM (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.modeling_llama4_flax) Llama4ForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.modeling_llama4_flax) Llama4ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.modeling_llama4_flax) Llama4MultiModalProjector (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextAttention (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextConfig (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.llama4_configuration) Llama4TextDecoderLayer (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextExperts (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextL2Norm (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextMLP (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextModel (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.modeling_llama4_flax) Llama4TextMoe (class in easydel.modules.llama4.modeling_llama4_flax) Llama4UnfoldConvolution (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionAttention (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionConfig (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.llama4_configuration) Llama4VisionEncoder (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionEncoderLayer (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionMLP (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionMLP2 (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionModel (class in easydel.__init__) (class in easydel.modules.llama4.__init__) (class in easydel.modules.llama4.modeling_llama4_flax) Llama4VisionPixelShuffleMLP (class in easydel.modules.llama4.modeling_llama4_flax) LlamaAttention (class in easydel.modules.llama.modeling_llama_flax) LlamaConfig (class in easydel.__init__) (class in easydel.modules.llama.__init__) (class in easydel.modules.llama.llama_configuration)

LlamaDecoderLayer (class in easydel.modules.llama.modeling_llama_flax) LlamaForCausalLM (class in easydel.__init__) (class in easydel.modules.llama.__init__) (class in easydel.modules.llama.modeling_llama_flax) LlamaForSequenceClassification (class in easydel.__init__) (class in easydel.modules.llama.__init__) (class in easydel.modules.llama.modeling_llama_flax) LlamaMLP (class in easydel.modules.llama.modeling_llama_flax) LlamaModel (class in easydel.__init__) (class in easydel.modules.llama.__init__) (class in easydel.modules.llama.modeling_llama_flax) LlavaCausalLMOutputWithPast (class in easydel.modules.llava.modeling_llava_flax) LlavaConfig (class in easydel.__init__) (class in easydel.modules.llava.__init__) (class in easydel.modules.llava.llava_configuration) LlavaForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.llava.__init__) (class in easydel.modules.llava.modeling_llava_flax) LlavaMultiModalProjector (class in easydel.modules.llava.modeling_llava_flax) lm_head (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) ln_1 (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) ln_2 (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) ln_post (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) ln_pre (easydel.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) load_checkpoint() (easydel.utils.checkpoint_managers.__init__.CheckpointManager static method) (easydel.utils.checkpoint_managers.streamer.CheckpointManager static method) load_compiled_fn() (in module easydel.utils.__init__) (in module easydel.utils.compiling_utils) load_inference() (easydel.__init__.vInference class method) (easydel.inference.__init__.vInference class method) (easydel.inference.vinference.__init__.vInference class method) (easydel.inference.vinference.vinference.vInference class method) load_optimizer() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) load_state() (easydel.__init__.EasyDeLState class method) (easydel.infra.__init__.EasyDeLState class method) (easydel.infra.base_state.EasyDeLState class method) log() (easydel.utils.__init__.Timers method) (easydel.utils.helpers.Timers method) log_all_workers (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) log_grad_norms (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) log_metrics() (easydel.__init__.BaseTrainer method) (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) (easydel.trainers.training_configurations.TrainingArguments method) log_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) log_weight_distribution() (easydel.__init__.BaseTrainer method) (easydel.__init__.TrainingArguments method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) (easydel.trainers.training_configurations.TrainingArguments method) logit_bias (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) logits (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.MambaCausalLMOutput attribute) (easydel.infra.modeling_outputs.MaskedLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput attribute) (easydel.infra.modeling_outputs.NextSentencePredictorOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.TokenClassifierOutput attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput attribute) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput attribute) logits_per_image (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) logits_per_text (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) logits_processor (easydel.__init__.SamplingParams property) (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.SamplingParams property) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.utilities.SamplingParams property) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) logits_warper (easydel.__init__.SamplingParams property) (easydel.inference.__init__.SamplingParams property) (easydel.inference.utilities.SamplingParams property) LogitsProcessor (class in easydel.inference.logits_process) LogitsProcessorList (class in easydel.inference.logits_process) LogitsWarper (class in easydel.inference.logits_process) logprobs (easydel.inference.openai_api_modules.CompletionResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionStreamResponseChoice attribute) long_factor (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute), [1] long_mscale (easydel.layers.rotary_embedding.RopeConfig attribute), [1] loss (easydel.infra.loss_utils.LossMetrics attribute), [1] (easydel.infra.modeling_outputs.BaseModelOutput attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPast attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention attribute) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention attribute) (easydel.infra.modeling_outputs.MambaCausalLMOutput attribute) (easydel.infra.modeling_outputs.MambaOutput attribute) (easydel.infra.modeling_outputs.MaskedLMOutput attribute) (easydel.infra.modeling_outputs.MoeCausalLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput attribute) (easydel.infra.modeling_outputs.NextSentencePredictorOutput attribute) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.TokenClassifierOutput attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) loss_config (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) loss_function (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) loss_normalizing_factor (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] loss_type (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.Gemma3ForConditionalGeneration attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.LlavaForConditionalGeneration attribute) (easydel.__init__.Qwen2VLForConditionalGeneration attribute) (easydel.__init__.WhisperForConditionalGeneration attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute) (easydel.modules.gemma3.__init__.Gemma3ForConditionalGeneration attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute) (easydel.modules.llava.__init__.LlavaForConditionalGeneration attribute) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] LossConfig (class in easydel.__init__) (class in easydel.infra.__init__) (class in easydel.infra.loss_utils) LossMetrics (class in easydel.infra.loss_utils) low_freq_factor (easydel.layers.rotary_embedding.Llama3RotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute), [1] low_mem_usage (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) lower_function() (in module easydel.inference.utilities) (in module easydel.utils.compiling_utils)

M

main() (in module easydel.inference.vwhisper.cli) (in module easydel.scripts.install_on_hosts) make_assertions_and_get_sizes() (in module easydel.trainers.training_utils) make_choice_type_function() (in module easydel.utils.cli_helpers) make_divisible() (in module easydel.modules.openelm.openelm_configuration) make_flexible_sliding_window() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) Mamba2Block (class in easydel.modules.mamba2.modeling_mamba2_flax) Mamba2Cache (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba2.__init__) (class in easydel.layers.caching.mamba2.mamba2_cache) Mamba2CacheMetaData (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba2.__init__) (class in easydel.layers.caching.mamba2.mamba2_cache) Mamba2CacheView (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba2.__init__) (class in easydel.layers.caching.mamba2.mamba2_cache) Mamba2CausalLMOutput (class in easydel.modules.mamba2.modeling_mamba2_flax) Mamba2Config (class in easydel.__init__) (class in easydel.modules.mamba2.__init__) (class in easydel.modules.mamba2.mamba2_configuration) Mamba2ForCausalLM (class in easydel.__init__) (class in easydel.modules.mamba2.__init__) (class in easydel.modules.mamba2.modeling_mamba2_flax) Mamba2Metadata (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba2.__init__) (class in easydel.layers.caching.mamba2.mamba2_cache) Mamba2Mixer (class in easydel.modules.mamba2.modeling_mamba2_flax) Mamba2Model (class in easydel.__init__) (class in easydel.modules.mamba2.__init__) (class in easydel.modules.mamba2.modeling_mamba2_flax) Mamba2Output (class in easydel.modules.mamba2.modeling_mamba2_flax) MambaBlock (class in easydel.modules.mamba.modeling_mamba_flax) MambaCache (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba.__init__) (class in easydel.layers.caching.mamba.mamba_cache) MambaCacheMetaData (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba.__init__) (class in easydel.layers.caching.mamba.mamba_cache) MambaCacheView (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba.__init__) (class in easydel.layers.caching.mamba.mamba_cache) MambaCausalLMOutput (class in easydel.infra.modeling_outputs) (class in easydel.modules.mamba.modeling_mamba_flax) MambaConfig (class in easydel.__init__) (class in easydel.modules.mamba.__init__) (class in easydel.modules.mamba.mamba_configuration) MambaConv1D (class in easydel.modules.mamba.modeling_mamba_flax) MambaForCausalLM (class in easydel.__init__) (class in easydel.modules.mamba.__init__) (class in easydel.modules.mamba.modeling_mamba_flax) MambaMetadata (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.mamba.__init__) (class in easydel.layers.caching.mamba.mamba_cache) MambaMixer (class in easydel.modules.mamba.modeling_mamba_flax) MambaModel (class in easydel.__init__) (class in easydel.modules.mamba.__init__) (class in easydel.modules.mamba.modeling_mamba_flax) MambaOutput (class in easydel.infra.modeling_outputs) (class in easydel.modules.mamba.modeling_mamba_flax) MambaRMSNormGated (class in easydel.modules.mamba2.modeling_mamba2_flax) manager (easydel.layers.caching.paged_attention.__init__.InferenceScheduler attribute) mapped_idx (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment attribute), [1] MaskedLMOutput (class in easydel.infra.modeling_outputs) match_keywords() (in module easydel.utils.parameters_transformation) matmul() (in module easydel.kernels.matmul) matmul_benchmark() (in module easydel.kernels.matmul) matmul_grad_test() (in module easydel.kernels.matmul) matmul_method (easydel.layers.linear.TensorParallelConfig attribute) matmul_test() (in module easydel.kernels.matmul) matmul_type (easydel.layers.linear.TensorParallelConfig attribute) MatrixMultiplyMethod (class in easydel.kernels.collective_matmul) max_completion_length (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.utils.DataCollatorForPreference attribute) (easydel.trainers.utils.DPODataCollatorWithPadding attribute) max_concurrent_decodes (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) max_concurrent_prefill (easydel.__init__.oEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.vsurge.__init__.oEngine property) max_evaluation_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) max_grad_norm (easydel.infra.loss_utils.LossMetrics attribute), [1] max_length (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.oEngine property) (easydel.__init__.ORPOConfig attribute), [1] (easydel.__init__.vEngine property) (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor attribute) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.utils.RewardDataCollatorWithPadding attribute) max_new_tokens (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] max_position_embeddings (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) max_prefill_length (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) max_prompt_length (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.utils.DataCollatorForPreference attribute) (easydel.trainers.utils.DPODataCollatorWithPadding attribute) max_seq_len (easydel.layers.caching.paged_attention.__init__.InferenceScheduler attribute) max_seq_length (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) max_sequence_length (easydel.__init__.ORPOConfig attribute) (easydel.__init__.RewardConfig attribute) (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.ORPOConfig attribute) (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute) (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) max_sequences (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] max_so_far (easydel.layers.attention_operator.modules.ring.Carry attribute) max_source_positions (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) max_target_positions (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) max_tokens (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) (easydel.layers.caching.paged_attention.__init__.SamplingParams attribute), [1] max_training_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) maybe_apply_chat_template() (in module easydel.trainers.prompt_utils) maybe_extract_prompt() (in module easydel.trainers.prompt_utils) maybe_unpair_preference_dataset() (in module easydel.trainers.prompt_utils) mean_grad_norm (easydel.infra.loss_utils.LossMetrics attribute), [1] measure_flops() (in module easydel.inference.vinference._fn) measure_inference_afterward() (easydel.inference.vinference.metrics.vInferenceMetrics method) measure_inference_first_step() (easydel.inference.vinference.metrics.vInferenceMetrics method) memory_monitor (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) memory_pressure (easydel.utils.analyze_memory.DeviceStats attribute), [1] MemoryMonitorClient (class in easydel.utils.analyze_memory) MemoryMonitorServer (class in easydel.utils.analyze_memory) merge() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) merge_lora_params() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (in module easydel.infra.utils) merge_model_and_tree() (in module easydel.utils.traversals) merge_module() (easydel.__init__.EasyDeLBaseModule static method) (easydel.infra.__init__.EasyDeLBaseModule static method) (easydel.infra.base_module.EasyDeLBaseModule static method) merge_params() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) merge_params_dict() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) merge_state_and_tree() (in module easydel.utils.traversals) merge_to_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) mesh (easydel.__init__.AttentionMetadata attribute), [1] (easydel.__init__.BaseTrainer property) (easydel.__init__.EasyDeLBaseConfig property) (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseConfig property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_config.EasyDeLBaseConfig property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] (easydel.layers.linear.TensorParallelConfig attribute), [1] (easydel.trainers.__init__.BaseTrainer property) (easydel.trainers.base_trainer.BaseTrainer property) (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol property) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput attribute) (easydel.trainers.trainer_protocol.TrainerOutput attribute) message (easydel.inference.openai_api_modules.ChatCompletionResponseChoice attribute) messages (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) metadata (easydel.__init__.FlexibleAttentionModule attribute) (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) (easydel.layers.attention.FlexibleAttentionModule attribute) (easydel.layers.attention_operator.__init__.AutoRegressiveDecodeAttn attribute) (easydel.layers.attention_operator.__init__.PagedAttn attribute) (easydel.layers.attention_operator.modules.__init__.AutoRegressiveDecodeAttn attribute) (easydel.layers.attention_operator.modules.__init__.PagedAttn attribute) (easydel.layers.attention_operator.modules.paged_attn.PagedAttn attribute) (easydel.layers.caching.__init__.HBMPageManager property) (easydel.layers.caching.__init__.LightningCacheView attribute) (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.__init__.MambaCacheView attribute) (easydel.layers.caching.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching._abstracts.BaseCacheView attribute) (easydel.layers.caching.lightning.__init__.LightningCacheView attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView attribute) (easydel.layers.caching.mamba.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) (easydel.layers.caching.paged_attention.__init__.HBMPageManager property) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) metadata() (easydel.layers.quantization.__init__.Linear8bit static method) (easydel.layers.quantization.__init__.LinearNF4 static method) (easydel.layers.quantization.base_quant.QauntModule static method) (easydel.layers.quantization.linear_8bit.Linear8bit static method) (easydel.layers.quantization.linear_nf4.LinearNF4 static method) MetaValueRecreator (class in easydel.utils.traversals) methods (easydel.inference.vinference.api_server.EndpointConfig attribute) (easydel.inference.vsurge.api_server.EndpointConfig attribute) metrics (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) metrics_to_show_in_rich_pbar (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) MetricsColumn (class in easydel.trainers.trainer_protocol) MetricsTracker (class in easydel.trainers.trainer_protocol) min_length (easydel.inference.logits_process.MinLengthLogitsProcessor attribute) min_p (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.MinPLogitsWarper attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) min_tokens_to_keep (easydel.inference.logits_process.MinPLogitsWarper attribute) (easydel.inference.logits_process.TopKLogitsWarper attribute) (easydel.inference.logits_process.TopPLogitsWarper attribute) minibatch_call() (in module easydel.trainers.training_utils) MiniMaxText01Attention (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01BlockSparseTop2MLP (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01Config (class in easydel.modules.minimax_text_v1.__init__) (class in easydel.modules.minimax_text_v1.minimax_text_01_configuration) MiniMaxText01DecoderLayer (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01ForCausalLM (class in easydel.modules.minimax_text_v1.__init__) (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01LightningAttention (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01MLP (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01Model (class in easydel.modules.minimax_text_v1.__init__) (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MiniMaxText01SparseMoeBlock (class in easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax) MinLengthLogitsProcessor (class in easydel.inference.logits_process) MinPLogitsWarper (class in easydel.inference.logits_process) missing_keys (easydel.utils.traversals.StateValidationResult attribute) MistralAttention (class in easydel.modules.mistral.modeling_mistral_flax) MistralConfig (class in easydel.__init__) (class in easydel.modules.mistral.__init__) (class in easydel.modules.mistral.mistral_configuration) MistralDecoderLayer (class in easydel.modules.mistral.modeling_mistral_flax) MistralForCausalLM (class in easydel.__init__) (class in easydel.modules.mistral.__init__) (class in easydel.modules.mistral.modeling_mistral_flax) MistralForSequenceClassification (class in easydel.__init__) (class in easydel.modules.mistral.__init__) (class in easydel.modules.mistral.modeling_mistral_flax) MistralMLP (class in easydel.modules.mistral.modeling_mistral_flax) MistralModel (class in easydel.__init__) (class in easydel.modules.mistral.__init__) (class in easydel.modules.mistral.modeling_mistral_flax) MixtralAttention (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralBLockSparseTop2MLP (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralConfig (class in easydel.__init__) (class in easydel.modules.mixtral.__init__) (class in easydel.modules.mixtral.mixtral_configuration) MixtralDecoderLayer (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralForCausalLM (class in easydel.__init__) (class in easydel.modules.mixtral.__init__) (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralForSequenceClassification (class in easydel.__init__) (class in easydel.modules.mixtral.__init__) (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralModel (class in easydel.__init__) (class in easydel.modules.mixtral.__init__) (class in easydel.modules.mixtral.modeling_mixtral_flax) MixtralSparseMoeBlock (class in easydel.modules.mixtral.modeling_mixtral_flax) mlp (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) mlp3d_sharding() (in module easydel.infra.utils) mlp_intermediate_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) model (easydel.__init__.BaseTrainer property) (easydel.__init__.EasyDeLState property) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CountTokenRequest attribute) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) (easydel.infra.__init__.EasyDeLState property) (easydel.infra.base_state.EasyDeLState property) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.trainers.__init__.BaseTrainer property) (easydel.trainers.base_trainer.BaseTrainer property) (easydel.trainers.trainer_protocol.BaseTrainerProtocol property) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput attribute) model_config (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionResponseChoice attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponseChoice attribute) (easydel.inference.openai_api_modules.ChatMessage attribute) (easydel.inference.openai_api_modules.CompletionLogprobs attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponseChoice attribute) (easydel.inference.openai_api_modules.CountTokenRequest attribute) (easydel.inference.openai_api_modules.DeltaMessage attribute) (easydel.inference.openai_api_modules.FunctionDefinition attribute) (easydel.inference.openai_api_modules.ToolDefinition attribute) (easydel.inference.openai_api_modules.UsageInfo attribute) (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.vinference.PromptOutput attribute) (easydel.inference.vinference.vinference.vInferenceMetaData attribute) (easydel.inference.vwhisper.server.TranscriptionResponse attribute) model_info (easydel.inference.vinference.metrics.vInferenceMetrics attribute) model_init_kwargs (easydel.__init__.ORPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute) model_kwargs (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.infra.mixins.generation.BeamSearchState attribute), [1] (easydel.infra.mixins.generation.GreedyState attribute), [1] (easydel.infra.mixins.generation.SampleState attribute), [1] model_name (easydel.__init__.DPOConfig attribute) (easydel.__init__.GRPOConfig attribute) (easydel.__init__.ORPOConfig attribute), [1] (easydel.__init__.RewardConfig attribute) (easydel.__init__.SFTConfig attribute) (easydel.__init__.TrainingArguments attribute) (easydel.inference.vinference.metrics.vInferenceMetrics attribute) (easydel.trainers.__init__.DPOConfig attribute) (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) model_parameters (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) model_prefill_length (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) model_state (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) model_task (easydel.__init__.AutoEasyDeLModel attribute) (easydel.__init__.AutoEasyDeLModelForCausalLM attribute) (easydel.__init__.AutoEasyDeLModelForImageTextToText attribute) (easydel.__init__.AutoEasyDeLModelForSeq2SeqLM attribute) (easydel.__init__.AutoEasyDeLModelForSequenceClassification attribute) (easydel.__init__.AutoEasyDeLModelForSpeechSeq2Seq attribute) (easydel.__init__.AutoEasyDeLModelForZeroShotImageClassification attribute) (easydel.__init__.AutoEasyDeLVisionModel attribute) (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.modules.auto.__init__.AutoEasyDeLModel attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForCausalLM attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForImageTextToText attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForSeq2SeqLM attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForSequenceClassification attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForSpeechSeq2Seq attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForZeroShotImageClassification attribute) (easydel.modules.auto.__init__.AutoEasyDeLVisionModel attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModel attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForCausalLM attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForImageTextToText attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForSeq2SeqLM attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForSequenceClassification attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForSpeechSeq2Seq attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForZeroShotImageClassification attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLVisionModel attribute) (easydel.modules.auto.auto_modeling.BaseAutoEasyModel attribute), [1] (easydel.utils.readme_generator.ModelInfo attribute)

model_type (easydel.__init__.ArcticConfig attribute) (easydel.__init__.AyaVisionConfig attribute) (easydel.__init__.CLIPConfig attribute) (easydel.__init__.CLIPTextConfig attribute) (easydel.__init__.CLIPVisionConfig attribute) (easydel.__init__.Cohere2Config attribute) (easydel.__init__.CohereConfig attribute) (easydel.__init__.DbrxConfig attribute) (easydel.__init__.DeepseekV2Config attribute) (easydel.__init__.DeepseekV3Config attribute) (easydel.__init__.EasyDeLBaseModule property) (easydel.__init__.ExaoneConfig attribute) (easydel.__init__.FalconConfig attribute) (easydel.__init__.Gemma2Config attribute) (easydel.__init__.Gemma3Config attribute) (easydel.__init__.Gemma3TextConfig attribute) (easydel.__init__.GemmaConfig attribute) (easydel.__init__.GPT2Config attribute) (easydel.__init__.GPTJConfig attribute) (easydel.__init__.GPTNeoXConfig attribute) (easydel.__init__.Grok1Config attribute) (easydel.__init__.InternLM2Config attribute) (easydel.__init__.Llama4Config attribute) (easydel.__init__.Llama4TextConfig attribute) (easydel.__init__.Llama4VisionConfig attribute) (easydel.__init__.LlamaConfig attribute) (easydel.__init__.LlavaConfig attribute) (easydel.__init__.Mamba2Config attribute) (easydel.__init__.MambaConfig attribute) (easydel.__init__.MistralConfig attribute) (easydel.__init__.MixtralConfig attribute) (easydel.__init__.MptConfig attribute) (easydel.__init__.Olmo2Config attribute) (easydel.__init__.OlmoConfig attribute) (easydel.__init__.OpenELMConfig attribute) (easydel.__init__.OPTConfig attribute) (easydel.__init__.Phi3Config attribute) (easydel.__init__.PhiConfig attribute) (easydel.__init__.PhiMoeConfig attribute) (easydel.__init__.PixtralVisionConfig attribute) (easydel.__init__.Qwen2Config attribute) (easydel.__init__.Qwen2MoeConfig attribute) (easydel.__init__.Qwen2VLConfig attribute) (easydel.__init__.Qwen3Config attribute) (easydel.__init__.Qwen3MoeConfig attribute) (easydel.__init__.RobertaConfig attribute) (easydel.__init__.SiglipConfig attribute) (easydel.__init__.SiglipTextConfig attribute) (easydel.__init__.SiglipVisionConfig attribute) (easydel.__init__.StableLmConfig attribute) (easydel.__init__.WhisperConfig attribute) (easydel.__init__.Xerxes2Config attribute) (easydel.__init__.XerxesConfig attribute) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.modules.arctic.__init__.ArcticConfig attribute) (easydel.modules.arctic.arctic_configuration.ArcticConfig attribute) (easydel.modules.aya_vision.__init__.AyaVisionConfig attribute) (easydel.modules.aya_vision.aya_vision_configuration.AyaVisionConfig attribute) (easydel.modules.clip.__init__.CLIPConfig attribute) (easydel.modules.clip.__init__.CLIPTextConfig attribute) (easydel.modules.clip.__init__.CLIPVisionConfig attribute) (easydel.modules.clip.clip_configuration.CLIPConfig attribute) (easydel.modules.clip.clip_configuration.CLIPTextConfig attribute) (easydel.modules.clip.clip_configuration.CLIPVisionConfig attribute) (easydel.modules.cohere.__init__.CohereConfig attribute) (easydel.modules.cohere.cohere_configuration.CohereConfig attribute) (easydel.modules.cohere2.__init__.Cohere2Config attribute) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config attribute) (easydel.modules.dbrx.__init__.DbrxConfig attribute) (easydel.modules.dbrx.dbrx_configuration.DbrxConfig attribute) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config attribute) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config attribute) (easydel.modules.deepseek_v3.__init__.DeepseekV3Config attribute) (easydel.modules.deepseek_v3.deepseek_configuration.DeepseekV3Config attribute) (easydel.modules.exaone.__init__.ExaoneConfig attribute) (easydel.modules.exaone.exaone_configuration.ExaoneConfig attribute) (easydel.modules.falcon.__init__.FalconConfig attribute) (easydel.modules.falcon.falcon_configuration.FalconConfig attribute) (easydel.modules.gemma.__init__.GemmaConfig attribute) (easydel.modules.gemma.gemma_configuration.GemmaConfig attribute) (easydel.modules.gemma2.__init__.Gemma2Config attribute) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config attribute) (easydel.modules.gemma3.__init__.Gemma3Config attribute) (easydel.modules.gemma3.__init__.Gemma3TextConfig attribute) (easydel.modules.gemma3.gemma3_configuration.Gemma3Config attribute) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig attribute) (easydel.modules.gpt2.__init__.GPT2Config attribute) (easydel.modules.gpt2.gpt2_configuration.GPT2Config attribute) (easydel.modules.gpt_j.__init__.GPTJConfig attribute) (easydel.modules.gpt_j.gpt_j_configuration.GPTJConfig attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXConfig attribute) (easydel.modules.gpt_neox.gpt_neox_configuration.GPTNeoXConfig attribute) (easydel.modules.grok_1.__init__.Grok1Config attribute) (easydel.modules.grok_1.grok_1_configuration.Grok1Config attribute) (easydel.modules.internlm2.__init__.InternLM2Config attribute) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config attribute) (easydel.modules.llama.__init__.LlamaConfig attribute) (easydel.modules.llama.llama_configuration.LlamaConfig attribute) (easydel.modules.llama4.__init__.Llama4Config attribute) (easydel.modules.llama4.__init__.Llama4TextConfig attribute) (easydel.modules.llama4.__init__.Llama4VisionConfig attribute) (easydel.modules.llama4.llama4_configuration.Llama4Config attribute) (easydel.modules.llama4.llama4_configuration.Llama4TextConfig attribute) (easydel.modules.llama4.llama4_configuration.Llama4VisionConfig attribute) (easydel.modules.llava.__init__.LlavaConfig attribute) (easydel.modules.llava.llava_configuration.LlavaConfig attribute) (easydel.modules.mamba.__init__.MambaConfig attribute) (easydel.modules.mamba.mamba_configuration.MambaConfig attribute) (easydel.modules.mamba2.__init__.Mamba2Config attribute) (easydel.modules.mamba2.mamba2_configuration.Mamba2Config attribute) (easydel.modules.minimax_text_v1.__init__.MiniMaxText01Config attribute) (easydel.modules.minimax_text_v1.minimax_text_01_configuration.MiniMaxText01Config attribute) (easydel.modules.mistral.__init__.MistralConfig attribute) (easydel.modules.mistral.mistral_configuration.MistralConfig attribute) (easydel.modules.mixtral.__init__.MixtralConfig attribute) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig attribute) (easydel.modules.mosaic_mpt.__init__.MptConfig attribute) (easydel.modules.mosaic_mpt.mosaic_configuration.MptConfig attribute) (easydel.modules.olmo.__init__.OlmoConfig attribute) (easydel.modules.olmo.olmo_configuration.OlmoConfig attribute) (easydel.modules.olmo2.__init__.Olmo2Config attribute) (easydel.modules.olmo2.olmo2_configuration.Olmo2Config attribute) (easydel.modules.openelm.__init__.OpenELMConfig attribute) (easydel.modules.openelm.openelm_configuration.OpenELMConfig attribute) (easydel.modules.opt.__init__.OPTConfig attribute) (easydel.modules.opt.opt_configuration.OPTConfig attribute) (easydel.modules.phi.__init__.PhiConfig attribute) (easydel.modules.phi.phi_configuration.PhiConfig attribute) (easydel.modules.phi3.__init__.Phi3Config attribute) (easydel.modules.phi3.phi3_configuration.Phi3Config attribute) (easydel.modules.phimoe.__init__.PhiMoeConfig attribute) (easydel.modules.phimoe.phimoe_configuration.PhiMoeConfig attribute) (easydel.modules.pixtral.__init__.PixtralVisionConfig attribute) (easydel.modules.pixtral.pixtral_configuration.PixtralVisionConfig attribute) (easydel.modules.qwen2.__init__.Qwen2Config attribute) (easydel.modules.qwen2.qwen_configuration.Qwen2Config attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig attribute) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig attribute) (easydel.modules.qwen2_vl.__init__.Qwen2VLConfig attribute) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLConfig attribute) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLVisionConfig attribute) (easydel.modules.qwen3.__init__.Qwen3Config attribute) (easydel.modules.qwen3.qwen3_configuration.Qwen3Config attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeConfig attribute) (easydel.modules.qwen3_moe.qwen3_moe_configuration.Qwen3MoeConfig attribute) (easydel.modules.roberta.__init__.RobertaConfig attribute) (easydel.modules.roberta.roberta_configuration.RobertaConfig attribute) (easydel.modules.rwkv.__init__.RwkvConfig attribute) (easydel.modules.rwkv.rwkv_configuration.RwkvConfig attribute) (easydel.modules.siglip.__init__.SiglipConfig attribute) (easydel.modules.siglip.__init__.SiglipTextConfig attribute) (easydel.modules.siglip.__init__.SiglipVisionConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipTextConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipVisionConfig attribute) (easydel.modules.stablelm.__init__.StableLmConfig attribute) (easydel.modules.stablelm.stablelm_configuration.StableLmConfig attribute) (easydel.modules.whisper.__init__.WhisperConfig attribute) (easydel.modules.whisper.whisper_configuration.WhisperConfig attribute) (easydel.modules.xerxes.__init__.XerxesConfig attribute) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig attribute) (easydel.modules.xerxes2.__init__.Xerxes2Config attribute) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config attribute) (easydel.utils.readme_generator.ModelInfo attribute) ModelInfo (class in easydel.utils.readme_generator) ModelInputBatch (class in easydel.layers.caching.paged_attention.__init__) ModelIOProcessor (class in easydel.layers.caching.paged_attention.__init__) ModelMetadata (class in easydel.inference.vinference.metrics) ModelOutput (class in easydel.infra.modeling_outputs) ModelOutputBatch (class in easydel.layers.caching.paged_attention.__init__) ModelOutputSummary (class in easydel.layers.caching.paged_attention.__init__) module easydel.__init__ easydel.inference.__init__ easydel.inference.logits_process easydel.inference.openai_api_modules easydel.inference.utilities easydel.inference.vinference.__init__ easydel.inference.vinference._fn easydel.inference.vinference.api_server easydel.inference.vinference.metrics easydel.inference.vinference.utilities easydel.inference.vinference.vinference easydel.inference.vsurge.__init__ easydel.inference.vsurge.api_server easydel.inference.vsurge.utils easydel.inference.vsurge.vsurge easydel.inference.vwhisper.__init__ easydel.inference.vwhisper.cli easydel.inference.vwhisper.config easydel.inference.vwhisper.core easydel.inference.vwhisper.generation easydel.inference.vwhisper.server easydel.inference.vwhisper.utils easydel.infra.__init__ easydel.infra.base_config easydel.infra.base_module easydel.infra.base_state easydel.infra.errors easydel.infra.etils easydel.infra.factory easydel.infra.loss_utils easydel.infra.mixins.__init__ easydel.infra.mixins.bridge easydel.infra.mixins.generation easydel.infra.mixins.protocol easydel.infra.modeling_outputs easydel.infra.utils easydel.kernels.__init__ easydel.kernels.collective_matmul easydel.kernels.cpu_ops.__init__ easydel.kernels.cpu_ops.flash_attention_jax.__init__ easydel.kernels.cpu_ops.flash_attention_jax._backward_jax easydel.kernels.cpu_ops.flash_attention_jax._flash_attention easydel.kernels.cpu_ops.flash_attention_jax._forward_jax easydel.kernels.cpu_ops.ring_attention_jax.__init__ easydel.kernels.cpu_ops.ring_attention_jax._backward_jax easydel.kernels.cpu_ops.ring_attention_jax._forward_jax easydel.kernels.cpu_ops.ring_attention_jax._ring_attention easydel.kernels.cpu_ops.ring_attention_jax._utils easydel.kernels.flash_attention easydel.kernels.gpu_ops.__init__ easydel.kernels.gpu_ops._deprecated.pallas_gemm easydel.kernels.gpu_ops._deprecated.pallas_gqa_flash_attention_2 easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2 easydel.kernels.gpu_ops._deprecated.triton_gqa_flash_attention_2 easydel.kernels.gpu_ops._utils easydel.kernels.gpu_ops.flash_attention_triton.__init__ easydel.kernels.gpu_ops.flash_attention_triton._backward_triton easydel.kernels.gpu_ops.flash_attention_triton._flash_attention easydel.kernels.gpu_ops.flash_attention_triton._forward_triton easydel.kernels.gpu_ops.flash_attention_triton._utils easydel.kernels.gpu_ops.matmul_triton.__init__ easydel.kernels.gpu_ops.matmul_triton._caller easydel.kernels.matmul easydel.kernels.ring_attention easydel.kernels.rms_norm easydel.kernels.tpu_ops.__init__ easydel.kernels.tpu_ops.matmul_pallas.__init__ easydel.kernels.tpu_ops.matmul_pallas._caller easydel.kernels.tpu_ops.paged_attention_pallas.__init__ easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention easydel.kernels.tpu_ops.ragged_attention_pallas.__init__ easydel.kernels.tpu_ops.ragged_attention_pallas._caller easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas easydel.kernels.tpu_ops.ring_attention_pallas.__init__ easydel.kernels.tpu_ops.ring_attention_pallas._backward_pallas easydel.kernels.tpu_ops.ring_attention_pallas._forward_pallas easydel.kernels.tpu_ops.ring_attention_pallas._ring_attention easydel.kernels.tpu_ops.ring_attention_pallas._utils easydel.kernels.utils easydel.layers.__init__ easydel.layers.attention easydel.layers.attention_operator.__init__ easydel.layers.attention_operator._attention_impl easydel.layers.attention_operator.modules.__init__ easydel.layers.attention_operator.modules.flash easydel.layers.attention_operator.modules.paged_attn easydel.layers.attention_operator.modules.ring easydel.layers.attention_operator.modules.scaled_dot_product easydel.layers.attention_operator.modules.splash easydel.layers.attention_operator.modules.vanilla easydel.layers.caching.__init__ easydel.layers.caching._abstracts easydel.layers.caching.lightning.__init__ easydel.layers.caching.lightning.lightning_cache easydel.layers.caching.mamba.__init__ easydel.layers.caching.mamba.mamba_cache easydel.layers.caching.mamba2.__init__ easydel.layers.caching.mamba2.mamba2_cache easydel.layers.caching.paged_attention.__init__ easydel.layers.caching.paged_attention.paged_attention_cache easydel.layers.caching.transformer.__init__ easydel.layers.caching.transformer.transformer_cache easydel.layers.linear easydel.layers.norms easydel.layers.ops.__init__ easydel.layers.ops._base_operation easydel.layers.ops._gla easydel.layers.ops._lightning_attention easydel.layers.quantization._8bit_quantizer easydel.layers.quantization.__init__ easydel.layers.quantization._nf4_quantizer easydel.layers.quantization.base_quant easydel.layers.quantization.linear_8bit easydel.layers.quantization.linear_nf4 easydel.layers.quantization.quantizers easydel.layers.rotary_embedding easydel.layers.rotary_embedding_test easydel.modules.__init__ easydel.modules.arctic.__init__ easydel.modules.arctic.arctic_configuration easydel.modules.arctic.modeling_arctic_flax easydel.modules.auto.__init__ easydel.modules.auto.auto_configuration easydel.modules.auto.auto_modeling easydel.modules.aya_vision.__init__ easydel.modules.aya_vision.aya_vision_configuration easydel.modules.aya_vision.modeling_aya_vision_flax easydel.modules.clip.__init__ easydel.modules.clip.clip_configuration easydel.modules.clip.modeling_clip_flax easydel.modules.cohere.__init__ easydel.modules.cohere.cohere_configuration easydel.modules.cohere.modeling_cohere_flax easydel.modules.cohere2.__init__ easydel.modules.cohere2.cohere2_configuration easydel.modules.cohere2.modeling_cohere2_flax easydel.modules.dbrx.__init__ easydel.modules.dbrx.dbrx_configuration easydel.modules.dbrx.modeling_dbrx_flax easydel.modules.deepseek_v2.__init__ easydel.modules.deepseek_v2.deepseek_configuration easydel.modules.deepseek_v2.modeling_deepseek_flax easydel.modules.deepseek_v3.__init__ easydel.modules.deepseek_v3.deepseek_configuration easydel.modules.deepseek_v3.modeling_deepseek_flax easydel.modules.exaone.__init__ easydel.modules.exaone.exaone_configuration easydel.modules.exaone.modeling_exaone_flax easydel.modules.falcon.__init__ easydel.modules.falcon.falcon_configuration easydel.modules.falcon.modeling_falcon_flax easydel.modules.gemma.__init__ easydel.modules.gemma.gemma_configuration easydel.modules.gemma.modeling_gemma_flax easydel.modules.gemma2.__init__ easydel.modules.gemma2.gemma2_configuration easydel.modules.gemma2.modeling_gemma2_flax easydel.modules.gemma3.__init__ easydel.modules.gemma3.gemma3_configuration easydel.modules.gemma3.modeling_gemma3_flax easydel.modules.gpt2.__init__ easydel.modules.gpt2.gpt2_configuration easydel.modules.gpt2.modeling_gpt2_flax easydel.modules.gpt_j.__init__ easydel.modules.gpt_j.gpt_j_configuration easydel.modules.gpt_j.modeling_gpt_j_flax easydel.modules.gpt_neox.__init__ easydel.modules.gpt_neox.gpt_neox_configuration easydel.modules.gpt_neox.modeling_gpt_neox_flax easydel.modules.grok_1.__init__ easydel.modules.grok_1.grok_1_configuration easydel.modules.grok_1.modeling_grok_1_flax easydel.modules.internlm2.__init__ easydel.modules.internlm2.internlm2_configuration easydel.modules.internlm2.modeling_internlm2_flax easydel.modules.llama.__init__ easydel.modules.llama.llama_configuration easydel.modules.llama.modeling_llama_flax easydel.modules.llama4.__init__ easydel.modules.llama4.llama4_configuration easydel.modules.llama4.modeling_llama4_flax easydel.modules.llava.__init__ easydel.modules.llava.llava_configuration easydel.modules.llava.modeling_llava_flax easydel.modules.mamba.__init__ easydel.modules.mamba.mamba_configuration easydel.modules.mamba.modeling_mamba_flax easydel.modules.mamba2.__init__ easydel.modules.mamba2.mamba2_configuration easydel.modules.mamba2.modeling_mamba2_flax easydel.modules.minimax_text_v1.__init__ easydel.modules.minimax_text_v1.minimax_text_01_configuration easydel.modules.minimax_text_v1.modeling_minimax_text_01_flax easydel.modules.mistral.__init__ easydel.modules.mistral.mistral_configuration easydel.modules.mistral.modeling_mistral_flax easydel.modules.mixtral.__init__ easydel.modules.mixtral.mixtral_configuration easydel.modules.mixtral.modeling_mixtral_flax easydel.modules.mosaic_mpt.__init__ easydel.modules.mosaic_mpt.modeling_mosaic_flax easydel.modules.mosaic_mpt.mosaic_configuration easydel.modules.olmo.__init__ easydel.modules.olmo.modeling_olmo_flax easydel.modules.olmo.olmo_configuration easydel.modules.olmo2.__init__ easydel.modules.olmo2.modeling_olmo2_flax easydel.modules.olmo2.olmo2_configuration easydel.modules.openelm.__init__ easydel.modules.openelm.modeling_openelm_flax easydel.modules.openelm.openelm_configuration easydel.modules.opt.__init__ easydel.modules.opt.modeling_opt_flax easydel.modules.opt.opt_configuration easydel.modules.phi.__init__ easydel.modules.phi.modeling_phi_flax easydel.modules.phi.phi_configuration easydel.modules.phi3.__init__ easydel.modules.phi3.modeling_phi3_flax easydel.modules.phi3.phi3_configuration easydel.modules.phimoe.__init__ easydel.modules.phimoe.modeling_phimoe_flax easydel.modules.phimoe.phimoe_configuration easydel.modules.pixtral.__init__ easydel.modules.pixtral.modeling_pixtral_flax easydel.modules.pixtral.pixtral_configuration easydel.modules.qwen2.__init__ easydel.modules.qwen2.modeling_qwen_flax easydel.modules.qwen2.qwen_configuration easydel.modules.qwen2_moe.__init__ easydel.modules.qwen2_moe.configuration_qwen2_moe easydel.modules.qwen2_moe.modeling_qwen2_moe_flax easydel.modules.qwen2_vl.__init__ easydel.modules.qwen2_vl.modeling_qwen2_vl_flax easydel.modules.qwen2_vl.qwen2_vl_configuration easydel.modules.qwen3.__init__ easydel.modules.qwen3.modeling_qwen3_flax easydel.modules.qwen3.qwen3_configuration easydel.modules.qwen3_moe.__init__ easydel.modules.qwen3_moe.modeling_qwen3_moe_flax easydel.modules.qwen3_moe.qwen3_moe_configuration easydel.modules.roberta.__init__ easydel.modules.roberta.modeling_roberta_flax easydel.modules.roberta.roberta_configuration easydel.modules.rwkv.__init__ easydel.modules.rwkv.modeling_rwkv_flax easydel.modules.rwkv.rwkv_configuration easydel.modules.siglip.__init__ easydel.modules.siglip.configuration_siglip easydel.modules.siglip.modeling_siglip_flax easydel.modules.stablelm.__init__ easydel.modules.stablelm.modeling_stablelm_flax easydel.modules.stablelm.stablelm_configuration easydel.modules.whisper.__init__ easydel.modules.whisper.modeling_whisper_flax easydel.modules.whisper.whisper_configuration easydel.modules.xerxes.__init__ easydel.modules.xerxes.modeling_xerxes_flax easydel.modules.xerxes.xerxes_configuration easydel.modules.xerxes2.__init__ easydel.modules.xerxes2.modeling_xerxes2_flax easydel.modules.xerxes2.xerxes2_configuration easydel.scripts.__init__ easydel.scripts.finetune.__init__ easydel.scripts.install_on_hosts easydel.trainers.__init__ easydel.trainers.base_trainer easydel.trainers.direct_preference_optimization_trainer.__init__ easydel.trainers.direct_preference_optimization_trainer._fn easydel.trainers.direct_preference_optimization_trainer.dpo_config easydel.trainers.direct_preference_optimization_trainer.dpo_trainer easydel.trainers.group_relative_policy_optimization.__init__ easydel.trainers.group_relative_policy_optimization._fn easydel.trainers.group_relative_policy_optimization.grpo_config easydel.trainers.group_relative_policy_optimization.grpo_trainer easydel.trainers.odds_ratio_preference_optimization_trainer.__init__ easydel.trainers.odds_ratio_preference_optimization_trainer._fn easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer easydel.trainers.packer easydel.trainers.prompt_utils easydel.trainers.reward_trainer.__init__ easydel.trainers.reward_trainer._fn easydel.trainers.reward_trainer.reward_config easydel.trainers.reward_trainer.reward_trainer easydel.trainers.supervised_fine_tuning_trainer.__init__ easydel.trainers.supervised_fine_tuning_trainer.sft_config easydel.trainers.supervised_fine_tuning_trainer.sft_trainer easydel.trainers.trainer.__init__ easydel.trainers.trainer._fn easydel.trainers.trainer.modeling_output easydel.trainers.trainer.trainer easydel.trainers.trainer_protocol easydel.trainers.training_configurations easydel.trainers.training_utils easydel.trainers.utils easydel.utils.__init__ easydel.utils.analyze_memory easydel.utils.checkpoint_managers.__init__ easydel.utils.checkpoint_managers.streamer easydel.utils.cli_helpers easydel.utils.compiling_utils easydel.utils.graph_utils easydel.utils.helpers easydel.utils.lazy_import easydel.utils.parameters_transformation easydel.utils.readme_generator easydel.utils.rngs_utils easydel.utils.traversals module (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.infra.factory.ModuleRegistration attribute), [1] (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) MODULE_CONFIG (easydel.__init__.ConfigType attribute), [1] (easydel.infra.factory.ConfigType attribute), [1] module_dtype (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) module_to_huggingface_model() (in module easydel.__init__) (in module easydel.utils.parameters_transformation) module_to_torch() (in module easydel.__init__) (in module easydel.utils.parameters_transformation) ModuleCaches (class in easydel.infra.utils) ModuleRegistration (class in easydel.infra.factory) moe_infer() (easydel.modules.deepseek_v3.modeling_deepseek_flax.DeepseekV3MoE method) MoeCausalLMOutput (class in easydel.infra.modeling_outputs) MoEGate (class in easydel.modules.deepseek_v2.modeling_deepseek_flax) (class in easydel.modules.deepseek_v3.modeling_deepseek_flax) MoeModelOutput (class in easydel.infra.modeling_outputs) MptAttention (class in easydel.modules.mosaic_mpt.modeling_mosaic_flax) MptAttentionConfig (class in easydel.__init__) (class in easydel.modules.mosaic_mpt.__init__) (class in easydel.modules.mosaic_mpt.mosaic_configuration) MptBlock (class in easydel.modules.mosaic_mpt.modeling_mosaic_flax) MptConfig (class in easydel.__init__) (class in easydel.modules.mosaic_mpt.__init__) (class in easydel.modules.mosaic_mpt.mosaic_configuration) MptForCausalLM (class in easydel.__init__) (class in easydel.modules.mosaic_mpt.__init__) (class in easydel.modules.mosaic_mpt.modeling_mosaic_flax) MptMLP (class in easydel.modules.mosaic_mpt.modeling_mosaic_flax) MptModel (class in easydel.__init__) (class in easydel.modules.mosaic_mpt.__init__) (class in easydel.modules.mosaic_mpt.modeling_mosaic_flax) mscale (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) mscale_all_dim (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute) MultiheadAttention (class in easydel.modules.siglip.modeling_siglip_flax) MultiPageAsyncCopyDescriptor (class in easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas) MultipleChoiceModelOutput (class in easydel.infra.modeling_outputs)

N

n (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) n_groups (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) n_heads (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) name (easydel.inference.openai_api_modules.ChatMessage attribute) (easydel.inference.openai_api_modules.FunctionDefinition attribute) (easydel.utils.readme_generator.ModelInfo attribute) named_tree_map() (in module easydel.utils.traversals) native_forward() (easydel.layers.linear.ParallelLinear method) new_decodes_requests (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] next_rng() (in module easydel.utils.rngs_utils) next_sampling_params (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] NextIterationPlan (class in easydel.layers.caching.paged_attention.__init__) NextSentencePredictorOutput (class in easydel.infra.modeling_outputs) NF4 (easydel.__init__.EasyDeLQuantizationMethods attribute), [1] (easydel.infra.etils.EasyDeLQuantizationMethods attribute), [1] ngram_size (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor attribute) nnx_init() (in module easydel.utils.traversals) NO_WEIGHT_NUM_REAL_TARGET_TOKENS (easydel.infra.loss_utils.SpecialLossNormalizingFactor attribute), [1] None (easydel.__init__.AutoEasyDeLModelForCausalLM attribute) (easydel.__init__.AutoEasyDeLModelForSpeechSeq2Seq attribute) (easydel.__init__.AutoShardAndGatherFunctions attribute) NONE (easydel.__init__.EasyDeLGradientCheckPointers attribute), [1] (easydel.__init__.EasyDeLQuantizationMethods attribute), [1] (easydel.__init__.EasyDeLSchedulers attribute), [1] (easydel.infra.etils.EasyDeLGradientCheckPointers attribute), [1] (easydel.infra.etils.EasyDeLQuantizationMethods attribute), [1] (easydel.infra.etils.EasyDeLSchedulers attribute), [1] None (easydel.modules.auto.__init__.AutoEasyDeLModelForCausalLM attribute) (easydel.modules.auto.__init__.AutoEasyDeLModelForSpeechSeq2Seq attribute) (easydel.modules.auto.__init__.AutoShardAndGatherFunctions attribute) (easydel.modules.auto.auto_configuration.AutoShardAndGatherFunctions attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForCausalLM attribute) (easydel.modules.auto.auto_modeling.AutoEasyDeLModelForSpeechSeq2Seq attribute) NoRepeatNGramLogitsProcessor (class in easydel.inference.logits_process) norm (easydel.__init__.InternLM2Model attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmModel attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) norm_1 (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) norm_2 (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) norm_f (easydel.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) norms (easydel.modules.stablelm.modeling_stablelm_flax.StableLmLayerNormPerHead attribute) NOTHING_SAVEABLE (easydel.__init__.EasyDeLGradientCheckPointers attribute), [1] (easydel.infra.etils.EasyDeLGradientCheckPointers attribute), [1] NullProgressBar (class in easydel.trainers.trainer_protocol) num_classification_labels (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] num_experts (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) num_experts_per_tok (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute)

num_generated_tokens (easydel.inference.vsurge.utils.ReturnSample attribute), [1] num_groups (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) num_heads (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) num_hidden_layers (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) num_k_heads (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) num_key_value_groups (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) num_key_value_heads (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) num_kv_heads (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] num_labels (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] num_mel_bins (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) num_of_sequences (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) num_pages_per_layer (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] num_pages_per_sequence (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] num_q_heads (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) NUM_REAL_TARGET_TOKENS (easydel.infra.loss_utils.SpecialLossNormalizingFactor attribute), [1] num_return_sequences (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] NUM_TOTAL_TARGET_TOKENS (easydel.infra.loss_utils.SpecialLossNormalizingFactor attribute), [1] num_train_epochs (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) num_used_slots (easydel.__init__.oDriver property) (easydel.inference.__init__.oDriver property) (easydel.inference.vsurge.__init__.oDriver property) num_v_heads (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) numerator (easydel.layers.attention_operator.modules.ring.Carry attribute)

O

o_proj (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) object (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) odds_ratio_loss() (in module easydel.trainers.odds_ratio_preference_optimization_trainer._fn) oDriver (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) oEngine (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) offload_dataset (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) offload_device (easydel.__init__.TrainingArguments property) (easydel.trainers.__init__.TrainingArguments property) (easydel.trainers.training_configurations.TrainingArguments property) offload_device_index (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) offload_device_type (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) offset (easydel.modules.opt.modeling_opt_flax.OPTLearnedPositionalEmbedding attribute) Olmo2Attention (class in easydel.modules.olmo2.modeling_olmo2_flax) Olmo2Config (class in easydel.__init__) (class in easydel.modules.olmo2.__init__) (class in easydel.modules.olmo2.olmo2_configuration) Olmo2DecoderLayer (class in easydel.modules.olmo2.modeling_olmo2_flax) Olmo2ForCausalLM (class in easydel.__init__) (class in easydel.modules.olmo2.__init__) (class in easydel.modules.olmo2.modeling_olmo2_flax) Olmo2ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.olmo2.modeling_olmo2_flax) Olmo2MLP (class in easydel.modules.olmo2.modeling_olmo2_flax) Olmo2Model (class in easydel.__init__) (class in easydel.modules.olmo2.__init__) (class in easydel.modules.olmo2.modeling_olmo2_flax) OlmoAttention (class in easydel.modules.olmo.modeling_olmo_flax) OlmoConfig (class in easydel.__init__) (class in easydel.modules.olmo.__init__) (class in easydel.modules.olmo.olmo_configuration) OlmoDecoderLayer (class in easydel.modules.olmo.modeling_olmo_flax) OlmoForCausalLM (class in easydel.__init__) (class in easydel.modules.olmo.__init__) (class in easydel.modules.olmo.modeling_olmo_flax) OlmoForSequenceClassification (class in easydel.modules.olmo.__init__) (class in easydel.modules.olmo.modeling_olmo_flax) OlmoMLP (class in easydel.modules.olmo.modeling_olmo_flax) OlmoModel (class in easydel.__init__) (class in easydel.modules.olmo.__init__) (class in easydel.modules.olmo.modeling_olmo_flax) on_step_end() (easydel.__init__.BaseTrainer method) (easydel.__init__.DPOTrainer method) (easydel.__init__.GRPOTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.__init__.DPOTrainer method) (easydel.trainers.__init__.GRPOTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer method) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method)

on_step_start() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) onehot() (in module easydel.infra.loss_utils) online_flops (easydel.infra.utils.CompilationTracker property) OpenELMConfig (class in easydel.__init__) (class in easydel.modules.openelm.__init__) (class in easydel.modules.openelm.openelm_configuration) OpenELMDecoderLayer (class in easydel.modules.openelm.modeling_openelm_flax) OpenELMFeedForwardNetwork (class in easydel.modules.openelm.modeling_openelm_flax) OpenELMForCausalLM (class in easydel.__init__) (class in easydel.modules.openelm.__init__) (class in easydel.modules.openelm.modeling_openelm_flax) OpenELMModel (class in easydel.__init__) (class in easydel.modules.openelm.__init__) (class in easydel.modules.openelm.modeling_openelm_flax) OpenELMMultiHeadCausalAttention (class in easydel.modules.openelm.modeling_openelm_flax) opt_state (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_state.EasyDeLState attribute), [1] OPTAttention (class in easydel.modules.opt.modeling_opt_flax) OPTConfig (class in easydel.__init__) (class in easydel.modules.opt.__init__) (class in easydel.modules.opt.opt_configuration) OPTDecoder (class in easydel.modules.opt.modeling_opt_flax) OPTDecoderLayer (class in easydel.modules.opt.modeling_opt_flax) OPTForCausalLM (class in easydel.__init__) (class in easydel.modules.opt.__init__) (class in easydel.modules.opt.modeling_opt_flax) optimizer (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) OPTLearnedPositionalEmbedding (class in easydel.modules.opt.modeling_opt_flax) OPTModel (class in easydel.__init__) (class in easydel.modules.opt.__init__) (class in easydel.modules.opt.modeling_opt_flax) orig_max_position (easydel.layers.rotary_embedding.Llama3RotaryEmbedding attribute) original_max_position_embeddings (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute), [1] (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) orpo_step() (in module easydel.trainers.odds_ratio_preference_optimization_trainer._fn) ORPOConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.odds_ratio_preference_optimization_trainer.__init__) (class in easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config) ORPOTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.odds_ratio_preference_optimization_trainer.__init__) (class in easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer) other_metrics (easydel.infra.loss_utils.LossMetrics attribute), [1] out_features (easydel.layers.linear.ParallelLinear attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) out_proj (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) output_arrays_only (easydel.trainers.utils.DPODataCollatorWithPadding attribute) overview (easydel.utils.readme_generator.ModelInfo attribute) OverWriteWithGradient (class in easydel.infra.utils)

P

pack_sequences() (in module easydel.__init__) (in module easydel.trainers.__init__) (in module easydel.trainers.packer) packing (easydel.__init__.SFTConfig attribute) (easydel.trainers.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig attribute) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig attribute) pad() (in module easydel.trainers.utils) pad_sequence() (in module easydel.trainers.utils) pad_tensor_by_size() (in module easydel.modules.mamba2.modeling_mamba2_flax) pad_to_length() (in module easydel.trainers.utils) pad_token_id (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) (easydel.trainers.utils.DataCollatorForPreference attribute) (easydel.trainers.utils.DPODataCollatorWithPadding attribute) pad_tokens() (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (in module easydel.inference.vsurge.utils) padded_length (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] padding (easydel.trainers.utils.RewardDataCollatorWithPadding attribute) padding_idx (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) padding_value (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.ORPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.ORPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig attribute), [1] (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig attribute), [1] page_idx (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment attribute), [1] page_indices (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages attribute), [1] (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] page_size (easydel.layers.caching.__init__.HBMPageManager property) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.__init__.HBMPageManager property) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData attribute), [1] page_table (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] page_update_mapped_idxs (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] page_update_page_idxs (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] page_update_slots (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] PAGED_ATTENTION (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] paged_attention() (in module easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention) paged_flash_attention_kernel() (in module easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas) paged_flash_attention_kernel_inline_seq_dim() (in module easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas) PagedAttention (class in easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention) PagedAttentionCache (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.paged_attention.__init__) (class in easydel.layers.caching.paged_attention.paged_attention_cache) PagedAttentionCacheMetaData (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.paged_attention.__init__) (class in easydel.layers.caching.paged_attention.paged_attention_cache) PagedAttentionCacheView (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.paged_attention.__init__) (class in easydel.layers.caching.paged_attention.paged_attention_cache) PagedAttentionMetadata (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.paged_attention.__init__) (class in easydel.layers.caching.paged_attention.paged_attention_cache) PagedAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.paged_attn) PALLAS (easydel.__init__.EasyDeLPlatforms attribute), [1] (easydel.infra.etils.EasyDeLPlatforms attribute), [1] (easydel.kernels.flash_attention.Platform attribute) pallas_gqa_flash_attention2_gpu() (in module easydel.kernels.gpu_ops._deprecated.pallas_gqa_flash_attention_2) pallas_mha_flash_attention2_gpu() (in module easydel.kernels.gpu_ops._deprecated.pallas_mha_flash_attention_2) pallas_paged_attention() (in module easydel.kernels.tpu_ops.__init__) (in module easydel.kernels.tpu_ops.paged_attention_pallas.__init__) pallas_prefill_attention() (in module easydel.kernels.tpu_ops.__init__) (in module easydel.kernels.tpu_ops.paged_attention_pallas.__init__) pallas_ragged_decode() (in module easydel.kernels.tpu_ops.__init__) (in module easydel.kernels.tpu_ops.ragged_attention_pallas.__init__) parallel_config (easydel.layers.linear.ParallelLinear attribute) ParallelLinear (class in easydel.layers.linear) param_dtype (easydel.__init__.ArcticForCausalLM attribute) (easydel.__init__.ArcticModel attribute) (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.CLIPForImageClassification attribute) (easydel.__init__.CLIPTextModel attribute) (easydel.__init__.CLIPTextModelWithProjection attribute) (easydel.__init__.CLIPVisionModel attribute) (easydel.__init__.CohereForSequenceClassification attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.GPT2Model attribute) (easydel.__init__.GPTJForCausalLM attribute) (easydel.__init__.GPTJModel attribute) (easydel.__init__.GPTNeoXForCausalLM attribute) (easydel.__init__.GPTNeoXModel attribute) (easydel.__init__.Grok1ForCausalLM attribute) (easydel.__init__.Grok1Model attribute) (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.InternLM2Model attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.Llama4ForSequenceClassification attribute) (easydel.__init__.LlamaForCausalLM attribute) (easydel.__init__.LlamaForSequenceClassification attribute) (easydel.__init__.LlamaModel attribute) (easydel.__init__.MistralForCausalLM attribute) (easydel.__init__.MistralForSequenceClassification attribute) (easydel.__init__.MistralModel attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.MptModel attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.OPTModel attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.StableLmModel attribute) (easydel.layers.linear.ParallelLinear attribute) (easydel.modules.arctic.__init__.ArcticForCausalLM attribute) (easydel.modules.arctic.__init__.ArcticForSequenceClassification attribute) (easydel.modules.arctic.__init__.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticAttention attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForCausalLM attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForSequenceClassification attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute), [1] (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector attribute) (easydel.modules.clip.__init__.CLIPForImageClassification attribute) (easydel.modules.clip.__init__.CLIPTextModel attribute) (easydel.modules.clip.__init__.CLIPTextModelWithProjection attribute) (easydel.modules.clip.__init__.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPAttention attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification attribute) (easydel.modules.clip.modeling_clip_flax.CLIPMLP attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer attribute) (easydel.modules.cohere.__init__.CohereForSequenceClassification attribute) (easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.__init__.GPT2Model attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute), [1] (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model attribute) (easydel.modules.gpt_j.__init__.GPTJForCausalLM attribute) (easydel.modules.gpt_j.__init__.GPTJModel attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXModel attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXAttention attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXBlock attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXMlp attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel attribute) (easydel.modules.grok_1.__init__.Grok1ForCausalLM attribute) (easydel.modules.grok_1.__init__.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1BLockSparseMLP attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1DecoderLayer attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1ForCausalLM attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1SparseMoeBlock attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.llama.__init__.LlamaForCausalLM attribute) (easydel.modules.llama.__init__.LlamaForSequenceClassification attribute) (easydel.modules.llama.__init__.LlamaModel attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForCausalLM attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForSequenceClassification attribute) (easydel.modules.llama.modeling_llama_flax.LlamaModel attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.__init__.Llama4ForSequenceClassification attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute), [1] (easydel.modules.llama4.modeling_llama4_flax.Llama4ForSequenceClassification attribute) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.mistral.__init__.MistralForCausalLM attribute) (easydel.modules.mistral.__init__.MistralForSequenceClassification attribute) (easydel.modules.mistral.__init__.MistralModel attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForCausalLM attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForSequenceClassification attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralModel attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.__init__.OPTModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTModel attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) parameters (easydel.__init__.EasyDeLBaseModule property) (easydel.inference.openai_api_modules.FunctionDefinition attribute) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) params (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) params_sharding (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) parse_args_into_dataclasses() (easydel.utils.__init__.DataClassArgumentParser method) (easydel.utils.cli_helpers.DataClassArgumentParser method) parse_dict() (easydel.utils.__init__.DataClassArgumentParser method) (easydel.utils.cli_helpers.DataClassArgumentParser method) parse_json_file() (easydel.utils.__init__.DataClassArgumentParser method) (easydel.utils.cli_helpers.DataClassArgumentParser method) parse_yaml_file() (easydel.utils.__init__.DataClassArgumentParser method) (easydel.utils.cli_helpers.DataClassArgumentParser method) partial_rotary_factor (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) partition_axis (easydel.__init__.AttentionMetadata attribute), [1] (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) partition_manager (easydel.__init__.AttentionMetadata attribute) (easydel.__init__.EasyDeLBaseConfig property) (easydel.infra.__init__.EasyDeLBaseConfig property) (easydel.infra.base_config.EasyDeLBaseConfig property) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute) (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute) partition_rules (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] PartitionAxis (class in easydel.__init__) (class in easydel.infra.__init__) past_key_values (easydel.infra.modeling_outputs.BaseModelOutput attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPast attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions attribute) (easydel.infra.modeling_outputs.MaskedLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqLMOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput attribute) (easydel.infra.modeling_outputs.SequenceClassifierOutput attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast attribute) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) patch_conv (easydel.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) PatchBlockSpec (class in easydel.kernels.tpu_ops.ring_attention_pallas._utils) PatchEmbed (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) PatchMerger (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) path (easydel.inference.vinference.api_server.EndpointConfig attribute) (easydel.inference.vsurge.api_server.EndpointConfig attribute) peak_utilization_percent (easydel.utils.analyze_memory.DeviceStats attribute), [1] perform_all_gather_matmul() (in module easydel.kernels.collective_matmul) perform_reduce_scatter_matmul() (in module easydel.kernels.collective_matmul) performance_mode (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) Phi3Attention (class in easydel.modules.phi3.modeling_phi3_flax) Phi3Config (class in easydel.__init__) (class in easydel.modules.phi3.__init__) (class in easydel.modules.phi3.phi3_configuration) Phi3DecoderLayer (class in easydel.modules.phi3.modeling_phi3_flax) Phi3ForCausalLM (class in easydel.__init__) (class in easydel.modules.phi3.__init__) (class in easydel.modules.phi3.modeling_phi3_flax) Phi3LongRoPEScaledRotaryEmbedding (class in easydel.layers.rotary_embedding) Phi3MLP (class in easydel.modules.phi3.modeling_phi3_flax) Phi3Model (class in easydel.__init__) (class in easydel.modules.phi3.__init__) (class in easydel.modules.phi3.modeling_phi3_flax) PhiAttention (class in easydel.modules.phi.modeling_phi_flax) PhiConfig (class in easydel.__init__) (class in easydel.modules.phi.__init__) (class in easydel.modules.phi.phi_configuration) PhiDecoderLayer (class in easydel.modules.phi.modeling_phi_flax) PhiForCausalLM (class in easydel.__init__) (class in easydel.modules.phi.__init__) (class in easydel.modules.phi.modeling_phi_flax) PhiMLP (class in easydel.modules.phi.modeling_phi_flax) PhiModel (class in easydel.__init__) (class in easydel.modules.phi.__init__) (class in easydel.modules.phi.modeling_phi_flax) PhiMoEAttention (class in easydel.modules.phimoe.modeling_phimoe_flax) PhiMoEBlockSparseTop2MLP (class in easydel.modules.phimoe.modeling_phimoe_flax) PhiMoeConfig (class in easydel.__init__) (class in easydel.modules.phimoe.__init__) (class in easydel.modules.phimoe.phimoe_configuration) PhiMoeDecoderLayer (class in easydel.modules.phimoe.modeling_phimoe_flax) PhiMoeForCausalLM (class in easydel.__init__) (class in easydel.modules.phimoe.__init__) (class in easydel.modules.phimoe.modeling_phimoe_flax) PhiMoeModel (class in easydel.__init__) (class in easydel.modules.phimoe.__init__) (class in easydel.modules.phimoe.modeling_phimoe_flax) PhiMoeSparseMoeBlock (class in easydel.modules.phimoe.modeling_phimoe_flax) pixel_shuffle() (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector method) (in module easydel.modules.llama4.modeling_llama4_flax) PixtralAttention (class in easydel.modules.pixtral.modeling_pixtral_flax) PixtralBlock (class in easydel.modules.pixtral.modeling_pixtral_flax) PixtralMLP (class in easydel.modules.pixtral.modeling_pixtral_flax) PixtralTransformer (class in easydel.modules.pixtral.modeling_pixtral_flax) PixtralVisionConfig (class in easydel.__init__) (class in easydel.modules.pixtral.__init__) (class in easydel.modules.pixtral.pixtral_configuration) PixtralVisionModel (class in easydel.__init__) (class in easydel.modules.pixtral.__init__) (class in easydel.modules.pixtral.modeling_pixtral_flax) place_request_on_prefill_queue() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method)

Platform (class in easydel.kernels.flash_attention) platform (easydel.__init__.AttentionMetadata attribute), [1] (easydel.inference.vinference.metrics.ModelMetadata attribute) (easydel.kernels.flash_attention.AttentionConfig attribute) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] platfrom (easydel.inference.vinference.metrics.ModelMetadata attribute) pooler_output (easydel.infra.modeling_outputs.BaseModelOutputWithPooling attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions attribute) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention attribute) pop() (easydel.infra.modeling_outputs.ModelOutput method) position (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] position_ids_in_meshgrid() (in module easydel.modules.pixtral.modeling_pixtral_flax) positions (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelInputBatch attribute), [1] post_attention_layernorm (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) postpadded (easydel.layers.caching.__init__.TransformerMetadata attribute) (easydel.layers.caching.transformer.__init__.TransformerMetadata attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata attribute) precision (easydel.__init__.ArcticForCausalLM attribute) (easydel.__init__.ArcticModel attribute) (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.CLIPForImageClassification attribute) (easydel.__init__.CLIPTextModel attribute) (easydel.__init__.CLIPTextModelWithProjection attribute) (easydel.__init__.CLIPVisionModel attribute) (easydel.__init__.CohereForSequenceClassification attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.GPT2Model attribute) (easydel.__init__.GPTJForCausalLM attribute) (easydel.__init__.GPTJModel attribute) (easydel.__init__.GPTNeoXForCausalLM attribute) (easydel.__init__.GPTNeoXModel attribute) (easydel.__init__.Grok1ForCausalLM attribute) (easydel.__init__.Grok1Model attribute) (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.InternLM2Model attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.Llama4ForSequenceClassification attribute) (easydel.__init__.LlamaForCausalLM attribute) (easydel.__init__.LlamaForSequenceClassification attribute) (easydel.__init__.LlamaModel attribute) (easydel.__init__.MistralForCausalLM attribute) (easydel.__init__.MistralForSequenceClassification attribute) (easydel.__init__.MistralModel attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.MptModel attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.OPTForCausalLM attribute) (easydel.__init__.OPTModel attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.StableLmModel attribute) (easydel.layers.linear.ParallelLinear attribute) (easydel.modules.arctic.__init__.ArcticForCausalLM attribute) (easydel.modules.arctic.__init__.ArcticForSequenceClassification attribute) (easydel.modules.arctic.__init__.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticAttention attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForCausalLM attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForSequenceClassification attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute), [1] (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector attribute) (easydel.modules.clip.__init__.CLIPForImageClassification attribute) (easydel.modules.clip.__init__.CLIPTextModel attribute) (easydel.modules.clip.__init__.CLIPTextModelWithProjection attribute) (easydel.modules.clip.__init__.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPAttention attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification attribute) (easydel.modules.clip.modeling_clip_flax.CLIPMLP attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer attribute) (easydel.modules.cohere.__init__.CohereForSequenceClassification attribute) (easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.__init__.GPT2Model attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute), [1] (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model attribute) (easydel.modules.gpt_j.__init__.GPTJForCausalLM attribute) (easydel.modules.gpt_j.__init__.GPTJModel attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXModel attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXAttention attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXBlock attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXMlp attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel attribute) (easydel.modules.grok_1.__init__.Grok1ForCausalLM attribute) (easydel.modules.grok_1.__init__.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1BLockSparseMLP attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1DecoderLayer attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1ForCausalLM attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1SparseMoeBlock attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Block attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) (easydel.modules.llama.__init__.LlamaForCausalLM attribute) (easydel.modules.llama.__init__.LlamaForSequenceClassification attribute) (easydel.modules.llama.__init__.LlamaModel attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForCausalLM attribute) (easydel.modules.llama.modeling_llama_flax.LlamaForSequenceClassification attribute) (easydel.modules.llama.modeling_llama_flax.LlamaModel attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.__init__.Llama4ForSequenceClassification attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute), [1] (easydel.modules.llama4.modeling_llama4_flax.Llama4ForSequenceClassification attribute) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.mistral.__init__.MistralForCausalLM attribute) (easydel.modules.mistral.__init__.MistralForSequenceClassification attribute) (easydel.modules.mistral.__init__.MistralModel attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForCausalLM attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralForSequenceClassification attribute) (easydel.modules.mistral.modeling_mistral_flax.MistralModel attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.opt.__init__.OPTForCausalLM attribute) (easydel.modules.opt.__init__.OPTModel attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM attribute) (easydel.modules.opt.modeling_opt_flax.OPTModel attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3MLP attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) precompile() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) precompiled_configs (easydel.inference.vinference.vinference.vInferenceMetaData attribute) precompute_ref_log_probs (easydel.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] precompute_vl_rotary() (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) prefill() (easydel.__init__.oEngine method) (easydel.__init__.vEngine method) (easydel.inference.__init__.oEngine method) (easydel.inference.__init__.vEngine method) (easydel.inference.vsurge.__init__.oEngine method) (easydel.inference.vsurge.__init__.vEngine method) prefill_attention() (in module easydel.kernels.tpu_ops.paged_attention_pallas._paged_attention) prefill_attention_impl() (in module easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas) prefill_complete (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] prefill_content (easydel.inference.vsurge.utils.ActiveRequest attribute) prefill_dequeue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) prefill_enqueue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) prefill_fn() (in module easydel.inference.vinference._fn) prefill_length (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] (easydel.layers.caching.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata attribute), [1] prefill_lengths (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) prefill_next_position (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] prefill_page_table (easydel.layers.caching.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata attribute), [1] prefill_pages_update (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] prefill_position (easydel.layers.caching.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata attribute), [1] prefill_queue (easydel.layers.caching.paged_attention.__init__.InferenceScheduler attribute) prefill_request (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] prefill_request_id (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] prefill_result (easydel.inference.vsurge.utils.ActiveRequest attribute) prefill_token_id (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary attribute), [1] prepadded (easydel.trainers.utils.DPODataCollatorWithPadding attribute) prepare_inputs_for_call() (easydel.__init__.EasyDeLBaseModule method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) prepare_inputs_for_generation() (easydel.__init__.AyaVisionForConditionalGeneration method) (easydel.__init__.Gemma3ForConditionalGeneration method) (easydel.__init__.Llama4ForConditionalGeneration method) (easydel.__init__.LlavaForConditionalGeneration method) (easydel.__init__.Mamba2ForCausalLM method) (easydel.__init__.MambaForCausalLM method) (easydel.__init__.OPTForCausalLM method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.__init__.WhisperForConditionalGeneration method) (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration method) (easydel.modules.gemma3.__init__.Gemma3ForConditionalGeneration method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration method) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration method) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration method) (easydel.modules.llava.__init__.LlavaForConditionalGeneration method) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration method) (easydel.modules.mamba.__init__.MambaForCausalLM method) (easydel.modules.mamba.modeling_mamba_flax.MambaForCausalLM method) (easydel.modules.mamba2.__init__.Mamba2ForCausalLM method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2ForCausalLM method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method) prepare_matrix_for_all_gather() (in module easydel.kernels.collective_matmul) prepare_matrix_for_reduce_scatter() (in module easydel.kernels.collective_matmul) prepare_model_input() (easydel.layers.caching.paged_attention.__init__.ModelIOProcessor static method) prepare_model_output() (easydel.layers.caching.paged_attention.__init__.ModelIOProcessor static method) presence_penalty (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) PresencePenaltyLogitsProcessor (class in easydel.inference.logits_process) prettify_nnx() (in module easydel.infra.mixins.protocol) print_current_status() (easydel.utils.analyze_memory.SMPMemoryMonitor method) print_history_summary() (easydel.utils.analyze_memory.SMPMemoryMonitor method) printify_nnx() (in module easydel.infra.mixins.protocol) prng_key (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) (easydel.infra.mixins.generation.SampleState attribute), [1] problem_type (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] process_audio_input() (in module easydel.inference.vwhisper.__init__) (in module easydel.inference.vwhisper.utils) process_client_side_tokenization_response() (easydel.__init__.vSurge method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) process_prompt() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) process_prompts_concurrently() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) process_result_tokens() (in module easydel.inference.vsurge.utils) process_row() (easydel.__init__.DPOTrainer static method) (easydel.trainers.__init__.DPOTrainer static method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer static method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer static method) process_server_side_tokenization_response() (easydel.__init__.vSurge method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) process_tensor() (in module easydel.utils.parameters_transformation) process_zero_is_admin (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) processing_time (easydel.inference.openai_api_modules.UsageInfo attribute) processor (easydel.__init__.oDriver property) (easydel.__init__.vDriver property) (easydel.__init__.vSurge property) (easydel.inference.__init__.oDriver property) (easydel.inference.__init__.vDriver property) (easydel.inference.__init__.vSurge property) (easydel.inference.vsurge.__init__.oDriver property) (easydel.inference.vsurge.__init__.vDriver property) (easydel.inference.vsurge.__init__.vSurge property) (easydel.inference.vsurge.vsurge.vSurge property) progress_bar_type (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) proj_1 (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) proj_2 (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) project_in (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) project_out (easydel.modules.opt.modeling_opt_flax.OPTDecoder attribute) prompt (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) prompt_token_ids (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] prompt_tokens (easydel.inference.openai_api_modules.UsageInfo attribute) PromptOutput (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.vinference) pruning_module (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) pt2jax() (in module easydel.utils.parameters_transformation) pure_transform_fn (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) push_to_hub() (easydel.infra.mixins.__init__.EasyBridgeMixin method) (easydel.infra.mixins.bridge.EasyBridgeMixin method) put_compiled_funcs() (in module easydel.inference.vinference._fn) PyTree (class in easydel.__init__) (class in easydel.infra.__init__)

Q

q_layernorm (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) q_norm (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) q_proj (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) QauntModule (class in easydel.layers.quantization.base_quant) qk_layernorm (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) qkv_proj (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) quantization_mapping() (easydel.layers.quantization.__init__.Linear8bit static method) (easydel.layers.quantization.__init__.LinearNF4 static method) (easydel.layers.quantization.base_quant.QauntModule static method) (easydel.layers.quantization.linear_8bit.Linear8bit static method) (easydel.layers.quantization.linear_nf4.LinearNF4 static method) quantize() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) quantize_8bit() (in module easydel.layers.quantization.linear_8bit) quantize_and_pack_nf4() (in module easydel.layers.quantization._nf4_quantizer) (in module easydel.layers.quantization.linear_nf4) quantize_linear_layers() (in module easydel.infra.utils) quantize_linears() (easydel.layers.quantization.__init__.EasyQuantizer method) (easydel.layers.quantization.quantizers.EasyQuantizer method) quantize_row_q8_0() (in module easydel.layers.quantization._8bit_quantizer) quantized_matmul_bwd() (in module easydel.layers.quantization.linear_8bit) quantized_matmul_fwd() (in module easydel.layers.quantization.linear_8bit) quantizer (easydel.__init__.AttentionModule property) (easydel.layers.attention.AttentionModule property) QuantParam (class in easydel.layers.quantization.base_quant) query (easydel.kernels.tpu_ops.ring_attention_pallas._utils.SegmentIds attribute), [1] query_sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) QuestionAnsweringModelOutput (class in easydel.infra.modeling_outputs) queue_size (easydel.inference.vinference.metrics.vInferenceMetrics attribute) quick_gelu() (in module easydel.infra.utils) quiet() (in module easydel.utils.helpers) Qwen2Attention (class in easydel.modules.qwen2.modeling_qwen_flax) Qwen2Config (class in easydel.__init__) (class in easydel.modules.qwen2.__init__) (class in easydel.modules.qwen2.qwen_configuration) Qwen2DecoderLayer (class in easydel.modules.qwen2.modeling_qwen_flax) Qwen2ForCausalLM (class in easydel.__init__) (class in easydel.modules.qwen2.__init__) (class in easydel.modules.qwen2.modeling_qwen_flax) Qwen2ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.qwen2.__init__) (class in easydel.modules.qwen2.modeling_qwen_flax) Qwen2MLP (class in easydel.modules.qwen2.modeling_qwen_flax) Qwen2Model (class in easydel.__init__) (class in easydel.modules.qwen2.__init__) (class in easydel.modules.qwen2.modeling_qwen_flax)

Qwen2MoeAttention (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeConfig (class in easydel.__init__) (class in easydel.modules.qwen2_moe.__init__) (class in easydel.modules.qwen2_moe.configuration_qwen2_moe) Qwen2MoeDecoderLayer (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeForCausalLM (class in easydel.__init__) (class in easydel.modules.qwen2_moe.__init__) (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeForSequenceClassification (class in easydel.__init__) (class in easydel.modules.qwen2_moe.__init__) (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeMLP (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeModel (class in easydel.__init__) (class in easydel.modules.qwen2_moe.__init__) (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2MoeSparseMoeBlock (class in easydel.modules.qwen2_moe.modeling_qwen2_moe_flax) Qwen2VisionTransformerPretrainedModel (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLAttention (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLCausalLMOutputWithPast (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLConfig (class in easydel.__init__) (class in easydel.modules.qwen2_vl.__init__) (class in easydel.modules.qwen2_vl.qwen2_vl_configuration) Qwen2VLDecoderLayer (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.qwen2_vl.__init__) (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLMLP (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLModel (class in easydel.__init__) (class in easydel.modules.qwen2_vl.__init__) (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLVisionBlock (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) Qwen2VLVisionConfig (class in easydel.modules.qwen2_vl.qwen2_vl_configuration) Qwen3Attention (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3Config (class in easydel.__init__) (class in easydel.modules.qwen3.__init__) (class in easydel.modules.qwen3.qwen3_configuration) Qwen3DecoderLayer (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3ForCausalLM (class in easydel.__init__) (class in easydel.modules.qwen3.__init__) (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3ForSequenceClassification (class in easydel.__init__) (class in easydel.modules.qwen3.__init__) (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3MLP (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3Model (class in easydel.__init__) (class in easydel.modules.qwen3.__init__) (class in easydel.modules.qwen3.modeling_qwen3_flax) Qwen3MoeAttention (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeConfig (class in easydel.__init__) (class in easydel.modules.qwen3_moe.__init__) (class in easydel.modules.qwen3_moe.qwen3_moe_configuration) Qwen3MoeDecoderLayer (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeForCausalLM (class in easydel.__init__) (class in easydel.modules.qwen3_moe.__init__) (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeForSequenceClassification (class in easydel.__init__) (class in easydel.modules.qwen3_moe.__init__) (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeMLP (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeModel (class in easydel.__init__) (class in easydel.modules.qwen3_moe.__init__) (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax) Qwen3MoeSparseMoeBlock (class in easydel.modules.qwen3_moe.modeling_qwen3_moe_flax)

R

ragged_attention() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._caller) ragged_flash_attention_kernel() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) ragged_gqa() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) ragged_mha() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) ragged_mqa() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) raw_stats (easydel.utils.analyze_memory.DeviceStats attribute), [1] raw_value (easydel.infra.utils.ModuleCaches attribute) (easydel.infra.utils.OverWriteWithGradient attribute) (easydel.layers.quantization.base_quant.QuantParam attribute) read_basics_from_config() (easydel.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) readiness() (easydel.__init__.vInferenceApiServer method) (easydel.__init__.vSurgeApiServer method) (easydel.inference.__init__.vInferenceApiServer method) (easydel.inference.__init__.vSurgeApiServer method) (easydel.inference.vinference.__init__.vInferenceApiServer method) (easydel.inference.vinference.api_server.vInferenceApiServer method) (easydel.inference.vsurge.__init__.vSurgeApiServer method) (easydel.inference.vsurge.api_server.vSurgeApiServer method) ReadmeGenerator (class in easydel.utils.readme_generator) record_model_metadata() (easydel.inference.vinference.metrics.vInferenceMetrics method) recreate_meta_values() (in module easydel.utils.traversals) recurrent_gla() (in module easydel.layers.ops._gla) RecurrentGLA (class in easydel.layers.ops.__init__) (class in easydel.layers.ops._gla) redefine_state() (in module easydel.utils.traversals) reduce_output (easydel.layers.linear.TensorParallelConfig attribute) REDUCE_SCATTER (easydel.kernels.collective_matmul.MatrixMultiplyMethod attribute), [1] reduce_scatter_output (easydel.layers.linear.TensorParallelConfig attribute), [1] reduction (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] ref_model_mixup_alpha (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) ref_model_sync_steps (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) reference_free (easydel.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] reference_gqa() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) reference_mha() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) reference_mqa() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._forward_pallas) reference_ragged_attention() (in module easydel.kernels.tpu_ops.ragged_attention_pallas._caller) refine_graphs() (in module easydel.utils.traversals) register() (easydel.__init__.AttentionRegistry class method) (easydel.layers.attention_operator.__init__.AttentionRegistry class method) (easydel.layers.attention_operator._attention_impl.AttentionRegistry class method) register_config() (easydel.infra.factory.Registry method) (in module easydel.__init__) (in module easydel.infra.factory) register_module() (easydel.infra.factory.Registry method) (in module easydel.__init__) (in module easydel.infra.factory) Registry (class in easydel.infra.factory) REGRESSIVE_DECODE (easydel.__init__.AttentionMechanisms attribute) (easydel.layers.attention.AttentionMechanisms attribute) rejected_rewards (easydel.infra.loss_utils.LossMetrics attribute), [1] remove_ckpt_after_load (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) remove_unused_columns (easydel.__init__.GRPOConfig attribute) (easydel.__init__.RewardConfig attribute) (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.__init__.RewardConfig attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) (easydel.trainers.reward_trainer.__init__.RewardConfig attribute) (easydel.trainers.reward_trainer.reward_config.RewardConfig attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) render() (easydel.trainers.trainer_protocol.MetricsColumn method) repeat_key_value() (easydel.__init__.AttentionModule static method) (easydel.layers.attention.AttentionModule static method) repeat_kv() (in module easydel.modules.cohere.modeling_cohere_flax) repeat_kv_heads() (easydel.kernels.flash_attention.FlashAttention static method) (easydel.layers.attention_operator.__init__.AttentionImpl static method) (easydel.layers.attention_operator._attention_impl.AttentionImpl static method) repetition_penalty (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) RepetitionPenaltyLogitsProcessor (class in easydel.inference.logits_process) replace() (easydel.__init__.AttentionMetadata method) (easydel.__init__.DPOConfig method) (easydel.__init__.EasyDeLState method) (easydel.__init__.GRPOConfig method) (easydel.__init__.LossConfig method) (easydel.__init__.ORPOConfig method) (easydel.__init__.PartitionAxis method) (easydel.__init__.RewardConfig method) (easydel.__init__.SamplingParams method) (easydel.__init__.SFTConfig method) (easydel.__init__.TrainingArguments method) (easydel.__init__.vInferenceConfig method) (easydel.__init__.vInferencePreCompileConfig method) (easydel.__init__.vWhisperInferenceConfig method) (easydel.inference.__init__.SamplingParams method) (easydel.inference.__init__.vInferenceConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vWhisperInferenceConfig method) (easydel.inference.logits_process.EmptyProcessor method) (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor method) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor method) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor method) (easydel.inference.logits_process.MinLengthLogitsProcessor method) (easydel.inference.logits_process.MinPLogitsWarper method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor method) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor method) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensLogitsProcessor method) (easydel.inference.logits_process.TemperatureLogitsWarper method) (easydel.inference.logits_process.TopKLogitsWarper method) (easydel.inference.logits_process.TopPLogitsWarper method) (easydel.inference.utilities.SamplingParams method) (easydel.inference.vinference.__init__.SampleState method) (easydel.inference.vinference.__init__.vInferenceConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.api_server.EndpointConfig method) (easydel.inference.vinference.metrics.ModelMetadata method) (easydel.inference.vinference.utilities.SampleState method) (easydel.inference.vinference.utilities.vInferenceConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) (easydel.inference.vsurge.api_server.EndpointConfig method) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig method) (easydel.inference.vwhisper.config.vWhisperInferenceConfig method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.__init__.LossConfig method) (easydel.infra.__init__.PartitionAxis method) (easydel.infra.base_config.EasyMethod method) (easydel.infra.base_state.EasyDeLState method) (easydel.infra.factory.ModuleRegistration method) (easydel.infra.loss_utils.LossConfig method) (easydel.infra.loss_utils.LossMetrics method) (easydel.infra.mixins.generation.BeamSearchState method) (easydel.infra.mixins.generation.GreedyState method) (easydel.infra.mixins.generation.SampleState method) (easydel.infra.modeling_outputs.AttentionLayerOutput method) (easydel.infra.modeling_outputs.BaseModelOutput method) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention method) (easydel.infra.modeling_outputs.BaseModelOutputWithPast method) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention method) (easydel.infra.modeling_outputs.BeamSearchOutput method) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions method) (easydel.infra.modeling_outputs.CLIPOutput method) (easydel.infra.modeling_outputs.CLIPTextModelOutput method) (easydel.infra.modeling_outputs.DecoderLayerOutput method) (easydel.infra.modeling_outputs.EncoderLayerOutput method) (easydel.infra.modeling_outputs.GreedySearchOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention method) (easydel.infra.modeling_outputs.MambaCausalLMOutput method) (easydel.infra.modeling_outputs.MambaOutput method) (easydel.infra.modeling_outputs.MaskedLMOutput method) (easydel.infra.modeling_outputs.MoeCausalLMOutput method) (easydel.infra.modeling_outputs.MoeModelOutput method) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput method) (easydel.infra.modeling_outputs.NextSentencePredictorOutput method) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.SampleOutput method) (easydel.infra.modeling_outputs.Seq2SeqLMOutput method) (easydel.infra.modeling_outputs.Seq2SeqModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput method) (easydel.infra.modeling_outputs.SequenceClassifierOutput method) (easydel.infra.modeling_outputs.TokenClassifierOutput method) (easydel.kernels.flash_attention.AttentionConfig method) (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes method) (easydel.layers.attention_operator.__init__.AttentionMetadata method) (easydel.layers.attention_operator.__init__.AttentionOutput method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata method) (easydel.layers.attention_operator._attention_impl.AttentionOutput method) (easydel.layers.caching.__init__.LightningCache method) (easydel.layers.caching.__init__.LightningCacheMetaData method) (easydel.layers.caching.__init__.LightningCacheView method) (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheMetaData method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.__init__.PagedAttentionCache method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.__init__.PagedAttentionMetadata method) (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.__init__.TransformerCacheMetaData method) (easydel.layers.caching.__init__.TransformerCacheView method) (easydel.layers.caching.__init__.TransformerMetadata method) (easydel.layers.caching._abstracts.BaseCacheMetadata method) (easydel.layers.caching._abstracts.BaseRunTimeMetadata method) (easydel.layers.caching.lightning.__init__.LightningCache method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData method) (easydel.layers.caching.lightning.__init__.LightningCacheView method) (easydel.layers.caching.lightning.lightning_cache.LightningCache method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages method) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest method) (easydel.layers.caching.paged_attention.__init__.ModelInputBatch method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch method) (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary method) (easydel.layers.caching.paged_attention.__init__.NextIterationPlan method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.__init__.SamplingParams method) (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData method) (easydel.layers.caching.transformer.__init__.TransformerCacheView method) (easydel.layers.caching.transformer.__init__.TransformerMetadata method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView method) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata method) (easydel.layers.linear.TensorParallelConfig method) (easydel.layers.rotary_embedding.RopeConfig method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast method) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast method) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast method) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput method) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput method) (easydel.trainers.__init__.DPOConfig method) (easydel.trainers.__init__.GRPOConfig method) (easydel.trainers.__init__.ORPOConfig method) (easydel.trainers.__init__.RewardConfig method) (easydel.trainers.__init__.SFTConfig method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig method) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig method) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig method) (easydel.trainers.reward_trainer.__init__.RewardConfig method) (easydel.trainers.reward_trainer.reward_config.RewardConfig method) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig method) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig method) (easydel.trainers.trainer.modeling_output.TrainerOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput method) (easydel.trainers.trainer_protocol.TrainerOutput method) (easydel.trainers.training_configurations.TrainingArguments method) (easydel.trainers.utils.DataCollatorForPreference method) (easydel.trainers.utils.DPODataCollatorWithPadding method) (easydel.trainers.utils.RewardDataCollatorWithPadding method) (easydel.utils.analyze_memory.DeviceStats method) (easydel.utils.traversals.StateValidationResult method) replace_dot_general_with_matmul() (in module easydel.kernels.matmul) repo_id (easydel.utils.readme_generator.ModelInfo attribute) report_metrics (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) report_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) required (easydel.inference.openai_api_modules.FunctionDefinition attribute) required_props (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] reset() (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) (easydel.trainers.trainer_protocol.BaseProgressBar method) (easydel.trainers.trainer_protocol.JSONProgressBar method) (easydel.trainers.trainer_protocol.MetricsTracker method) (easydel.trainers.trainer_protocol.NullProgressBar method) (easydel.trainers.trainer_protocol.RichProgressBar method) (easydel.trainers.trainer_protocol.TqdmProgressBar method) (easydel.utils.__init__.Timer method) (easydel.utils.helpers.Timer method) reshape_for_broadcast() (in module easydel.modules.llama4.modeling_llama4_flax) reshape_into_chunks() (in module easydel.modules.mamba2.modeling_mamba2_flax) resid_attn_dropout (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) resid_dropout (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) residual_states (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) (easydel.infra.modeling_outputs.EncoderLayerOutput attribute) resolve_spec() (easydel.__init__.PartitionAxis method) (easydel.infra.__init__.PartitionAxis method) ResponseFormat (class in easydel.inference.vwhisper.server) return_channel (easydel.inference.vsurge.utils.ActiveRequest attribute) return_timestamps (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) return_type_adjuster() (in module easydel.infra.mixins.protocol) ReturnSample (class in easydel.inference.vsurge.utils) reverse_openai_format() (in module easydel.trainers.prompt_utils) RewardConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.reward_trainer.__init__) (class in easydel.trainers.reward_trainer.reward_config) RewardDataCollatorWithPadding (class in easydel.trainers.utils) RewardTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.reward_trainer.__init__) (class in easydel.trainers.reward_trainer.reward_trainer) RichProgressBar (class in easydel.trainers.trainer_protocol) RING (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] ring_attention() (in module easydel.kernels.ring_attention) RingAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.ring) rms_norm() (in module easydel.kernels.rms_norm) RMSNorm (class in easydel.layers.norms) (class in easydel.modules.cohere.modeling_cohere_flax) (class in easydel.modules.xerxes.modeling_xerxes_flax) RMSPROP (easydel.__init__.EasyDeLOptimizers attribute), [1] (easydel.infra.etils.EasyDeLOptimizers attribute), [1] rng (easydel.utils.__init__.GenerateRNG property) (easydel.utils.rngs_utils.GenerateRNG property) rng_keys() (easydel.__init__.ArcticConfig static method) (easydel.__init__.Cohere2Config static method) (easydel.__init__.CohereConfig static method) (easydel.__init__.DeepseekV2Config static method) (easydel.__init__.ExaoneConfig static method) (easydel.__init__.Gemma2Config static method) (easydel.__init__.Gemma3TextConfig static method) (easydel.__init__.GemmaConfig static method) (easydel.__init__.Grok1Config static method) (easydel.__init__.InternLM2Config static method) (easydel.__init__.LlamaConfig static method) (easydel.__init__.MistralConfig static method) (easydel.__init__.MixtralConfig static method) (easydel.__init__.OpenELMConfig static method) (easydel.__init__.Qwen2Config static method) (easydel.__init__.Qwen2MoeConfig static method) (easydel.__init__.Xerxes2Config static method) (easydel.__init__.XerxesConfig static method) (easydel.modules.arctic.__init__.ArcticConfig static method) (easydel.modules.arctic.arctic_configuration.ArcticConfig static method) (easydel.modules.cohere.__init__.CohereConfig static method) (easydel.modules.cohere.cohere_configuration.CohereConfig static method) (easydel.modules.cohere2.__init__.Cohere2Config static method) (easydel.modules.cohere2.cohere2_configuration.Cohere2Config static method) (easydel.modules.deepseek_v2.__init__.DeepseekV2Config static method) (easydel.modules.deepseek_v2.deepseek_configuration.DeepseekV2Config static method) (easydel.modules.exaone.__init__.ExaoneConfig static method) (easydel.modules.exaone.exaone_configuration.ExaoneConfig static method) (easydel.modules.gemma.__init__.GemmaConfig static method) (easydel.modules.gemma.gemma_configuration.GemmaConfig static method) (easydel.modules.gemma2.__init__.Gemma2Config static method) (easydel.modules.gemma2.gemma2_configuration.Gemma2Config static method) (easydel.modules.gemma3.__init__.Gemma3TextConfig static method) (easydel.modules.gemma3.gemma3_configuration.Gemma3TextConfig static method) (easydel.modules.grok_1.__init__.Grok1Config static method) (easydel.modules.grok_1.grok_1_configuration.Grok1Config static method) (easydel.modules.internlm2.__init__.InternLM2Config static method) (easydel.modules.internlm2.internlm2_configuration.InternLM2Config static method) (easydel.modules.llama.__init__.LlamaConfig static method) (easydel.modules.llama.llama_configuration.LlamaConfig static method) (easydel.modules.mistral.__init__.MistralConfig static method) (easydel.modules.mistral.mistral_configuration.MistralConfig static method) (easydel.modules.mixtral.__init__.MixtralConfig static method) (easydel.modules.mixtral.mixtral_configuration.MixtralConfig static method) (easydel.modules.openelm.__init__.OpenELMConfig static method) (easydel.modules.openelm.openelm_configuration.OpenELMConfig static method) (easydel.modules.qwen2.__init__.Qwen2Config static method) (easydel.modules.qwen2.qwen_configuration.Qwen2Config static method) (easydel.modules.qwen2_moe.__init__.Qwen2MoeConfig static method) (easydel.modules.qwen2_moe.configuration_qwen2_moe.Qwen2MoeConfig static method) (easydel.modules.xerxes.__init__.XerxesConfig static method) (easydel.modules.xerxes.xerxes_configuration.XerxesConfig static method) (easydel.modules.xerxes2.__init__.Xerxes2Config static method) (easydel.modules.xerxes2.xerxes2_configuration.Xerxes2Config static method)

rngs (easydel.__init__.ArcticForCausalLM attribute) (easydel.__init__.ArcticModel attribute) (easydel.__init__.AyaVisionForConditionalGeneration attribute) (easydel.__init__.CLIPForImageClassification attribute) (easydel.__init__.CLIPTextModel attribute) (easydel.__init__.CLIPTextModelWithProjection attribute) (easydel.__init__.CLIPVisionModel attribute) (easydel.__init__.CohereForSequenceClassification attribute) (easydel.__init__.GPT2LMHeadModel attribute) (easydel.__init__.GPT2Model attribute) (easydel.__init__.GPTJForCausalLM attribute) (easydel.__init__.GPTJModel attribute) (easydel.__init__.GPTNeoXForCausalLM attribute) (easydel.__init__.GPTNeoXModel attribute) (easydel.__init__.Grok1ForCausalLM attribute) (easydel.__init__.Grok1Model attribute) (easydel.__init__.InternLM2ForCausalLM attribute) (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.Llama4ForConditionalGeneration attribute) (easydel.__init__.MixtralForCausalLM attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.MixtralModel attribute) (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.MptModel attribute) (easydel.__init__.Olmo2ForCausalLM attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Olmo2Model attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OlmoModel attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.OpenELMModel attribute) (easydel.__init__.Phi3ForCausalLM attribute) (easydel.__init__.Phi3Model attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PhiModel attribute) (easydel.__init__.PhiMoeForCausalLM attribute) (easydel.__init__.PhiMoeModel attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.__init__.Qwen2ForCausalLM attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2Model attribute) (easydel.__init__.Qwen2MoeForCausalLM attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen2MoeModel attribute) (easydel.__init__.Qwen3ForCausalLM attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3Model attribute) (easydel.__init__.Qwen3MoeForCausalLM attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.__init__.Qwen3MoeModel attribute) (easydel.__init__.StableLmForCausalLM attribute) (easydel.__init__.StableLmModel attribute) (easydel.modules.arctic.__init__.ArcticForCausalLM attribute) (easydel.modules.arctic.__init__.ArcticForSequenceClassification attribute) (easydel.modules.arctic.__init__.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticAttention attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticDecoderLayer attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForCausalLM attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticForSequenceClassification attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMLP attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticModel attribute) (easydel.modules.arctic.modeling_arctic_flax.ArcticMoeBlock attribute) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration attribute) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration attribute), [1] (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionMultiModalProjector attribute) (easydel.modules.clip.__init__.CLIPForImageClassification attribute) (easydel.modules.clip.__init__.CLIPTextModel attribute) (easydel.modules.clip.__init__.CLIPTextModelWithProjection attribute) (easydel.modules.clip.__init__.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPAttention attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoder attribute) (easydel.modules.clip.modeling_clip_flax.CLIPEncoderLayer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPForImageClassification attribute) (easydel.modules.clip.modeling_clip_flax.CLIPMLP attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextModelWithProjection attribute) (easydel.modules.clip.modeling_clip_flax.CLIPTextTransformer attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionEmbeddings attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionModel attribute) (easydel.modules.clip.modeling_clip_flax.CLIPVisionTransformer attribute) (easydel.modules.cohere.__init__.CohereForSequenceClassification attribute) (easydel.modules.cohere.modeling_cohere_flax.CohereForSequenceClassification attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2Attention attribute) (easydel.modules.cohere2.modeling_cohere2_flax.Cohere2LayerNorm attribute) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration attribute) (easydel.modules.gpt2.__init__.GPT2LMHeadModel attribute) (easydel.modules.gpt2.__init__.GPT2Model attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Attention attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Block attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2LMHeadModel attribute), [1] (easydel.modules.gpt2.modeling_gpt2_flax.GPT2MLP attribute) (easydel.modules.gpt2.modeling_gpt2_flax.GPT2Model attribute) (easydel.modules.gpt_j.__init__.GPTJForCausalLM attribute) (easydel.modules.gpt_j.__init__.GPTJModel attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJAttention attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJBlock attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJForCausalLM attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJMLP attribute) (easydel.modules.gpt_j.modeling_gpt_j_flax.GPTJModel attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.__init__.GPTNeoXModel attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXAttention attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXBlock attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXForCausalLM attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXMlp attribute) (easydel.modules.gpt_neox.modeling_gpt_neox_flax.GPTNeoXModel attribute) (easydel.modules.grok_1.__init__.Grok1ForCausalLM attribute) (easydel.modules.grok_1.__init__.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Attention attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1BLockSparseMLP attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1DecoderLayer attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1ForCausalLM attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1Model attribute) (easydel.modules.grok_1.modeling_grok_1_flax.Grok1SparseMoeBlock attribute) (easydel.modules.internlm2.__init__.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForCausalLM attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration attribute) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration attribute), [1] (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration attribute) (easydel.modules.mixtral.__init__.MixtralForCausalLM attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForCausalLM attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralModel attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralSparseMoeBlock attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptBlock attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoModel attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoModel attribute) (easydel.modules.olmo2.__init__.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.__init__.Olmo2Model attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForCausalLM attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Model attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMDecoderLayer attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMFeedForwardNetwork attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.__init__.PhiModel attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiMLP attribute) (easydel.modules.phi.modeling_phi_flax.PhiModel attribute) (easydel.modules.phi3.__init__.Phi3ForCausalLM attribute) (easydel.modules.phi3.__init__.Phi3Model attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3ForCausalLM attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Model attribute) (easydel.modules.phimoe.__init__.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.__init__.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeForCausalLM attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeModel attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeSparseMoeBlock attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralBlock attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralTransformer attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) (easydel.modules.qwen2.__init__.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2Model attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForCausalLM attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Model attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForCausalLM attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeModel attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeSparseMoeBlock attribute) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration attribute) (easydel.modules.qwen3.__init__.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3Model attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForCausalLM attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Model attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForCausalLM attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeModel attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeSparseMoeBlock attribute) (easydel.modules.stablelm.__init__.StableLmForCausalLM attribute) (easydel.modules.stablelm.__init__.StableLmModel attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmForCausalLM attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmModel attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperModel attribute) RobertaAttention (class in easydel.modules.roberta.modeling_roberta_flax) RobertaClassificationHead (class in easydel.modules.roberta.modeling_roberta_flax) RobertaConfig (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.roberta_configuration) RobertaEmbeddings (class in easydel.modules.roberta.modeling_roberta_flax) RobertaEncoder (class in easydel.modules.roberta.modeling_roberta_flax) RobertaForCausalLM (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.modeling_roberta_flax) RobertaForMultipleChoice (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.modeling_roberta_flax) RobertaForQuestionAnswering (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.modeling_roberta_flax) RobertaForSequenceClassification (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.modeling_roberta_flax) RobertaForTokenClassification (class in easydel.__init__) (class in easydel.modules.roberta.__init__) (class in easydel.modules.roberta.modeling_roberta_flax) RobertaIntermediate (class in easydel.modules.roberta.modeling_roberta_flax) RobertaLayer (class in easydel.modules.roberta.modeling_roberta_flax) RobertaLMHead (class in easydel.modules.roberta.modeling_roberta_flax) RobertaModel (class in easydel.modules.roberta.modeling_roberta_flax) RobertaOutput (class in easydel.modules.roberta.modeling_roberta_flax) RobertaPooler (class in easydel.modules.roberta.modeling_roberta_flax) RobertaSelfAttention (class in easydel.modules.roberta.modeling_roberta_flax) RobertaSelfOutput (class in easydel.modules.roberta.modeling_roberta_flax) role (easydel.inference.openai_api_modules.ChatMessage attribute) (easydel.inference.openai_api_modules.DeltaMessage attribute) rope_deltas (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast attribute) rope_scaling (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) rope_theta (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) rope_type (easydel.layers.rotary_embedding.RopeConfig attribute), [1] rope_wraper() (in module easydel.layers.rotary_embedding) RopeConfig (class in easydel.layers.rotary_embedding) rot_pos_emb() (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VisionTransformerPretrainedModel method) rotary (easydel.__init__.FalconConfig property) (easydel.modules.falcon.__init__.FalconConfig property) (easydel.modules.falcon.falcon_configuration.FalconConfig property) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3Attention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) rotary_dim (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RotaryEmbedding attribute) rotary_emb_dim (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) RotaryEmbedding (class in easydel.layers.rotary_embedding) rotate_half() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) (in module easydel.modules.pixtral.modeling_pixtral_flax) (in module easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) router_logits (easydel.infra.modeling_outputs.DecoderLayerOutput attribute) (easydel.infra.modeling_outputs.MoeCausalLMOutput attribute) (easydel.infra.modeling_outputs.MoeModelOutput attribute) rpo_alpha (easydel.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] run() (easydel.inference.vsurge.utils.SafeThread method) run_all_tests() (in module easydel.kernels.collective_matmul) run_cli() (in module easydel.inference.vwhisper.__init__) run_server() (in module easydel.inference.vwhisper.__init__) (in module easydel.inference.vwhisper.server) running_scores (easydel.infra.mixins.generation.BeamSearchState attribute), [1] running_sequences (easydel.infra.mixins.generation.BeamSearchState attribute), [1] running_token (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.infra.mixins.generation.GreedyState attribute), [1] (easydel.infra.mixins.generation.SampleState attribute), [1] runtime3d_sharding() (in module easydel.infra.utils) runtime_dtype (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] runtime_softmax_dtype (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] rwkv_linear_attention() (in module easydel.modules.rwkv.modeling_rwkv_flax) RwkvCausalLMOutput (class in easydel.modules.rwkv.modeling_rwkv_flax) RwkvConfig (class in easydel.modules.rwkv.__init__) (class in easydel.modules.rwkv.rwkv_configuration) RwkvFeedForward (class in easydel.modules.rwkv.modeling_rwkv_flax) RwkvForCausalLM (class in easydel.modules.rwkv.__init__) (class in easydel.modules.rwkv.modeling_rwkv_flax) RwkvModel (class in easydel.modules.rwkv.__init__) (class in easydel.modules.rwkv.modeling_rwkv_flax) RwkvOutput (class in easydel.modules.rwkv.modeling_rwkv_flax) RwkvSelfAttention (class in easydel.modules.rwkv.modeling_rwkv_flax)

S

safe_autotune() (in module easydel.kernels.gpu_ops._utils) SafeThread (class in easydel.inference.vsurge.utils) SampleOutput (class in easydel.infra.modeling_outputs) samples_per_slot (easydel.__init__.oEngine property) (easydel.__init__.vEngine property) (easydel.inference.__init__.oEngine property) (easydel.inference.__init__.vEngine property) (easydel.inference.vsurge.__init__.oEngine property) (easydel.inference.vsurge.__init__.vEngine property) SampleState (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.utilities) (class in easydel.infra.mixins.generation) sampling_params (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] (easydel.layers.caching.paged_attention.__init__.ModelInputBatch attribute), [1] SamplingParams (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.utilities) (class in easydel.layers.caching.paged_attention.__init__) save_checkpoint() (easydel.utils.checkpoint_managers.__init__.CheckpointManager static method) (easydel.utils.checkpoint_managers.streamer.CheckpointManager static method) save_compiled_fn() (in module easydel.utils.__init__) (in module easydel.utils.compiling_utils) save_directory (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) save_inference() (easydel.__init__.vInference method) (easydel.inference.__init__.vInference method) (easydel.inference.vinference.__init__.vInference method) (easydel.inference.vinference.vinference.vInference method) save_information() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) save_optimizer_state (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) save_pickle() (easydel.utils.checkpoint_managers.__init__.CheckpointManager method) (easydel.utils.checkpoint_managers.streamer.CheckpointManager method) save_pretrained() (easydel.__init__.BaseTrainer method) (easydel.infra.mixins.__init__.EasyBridgeMixin method) (easydel.infra.mixins.bridge.EasyBridgeMixin method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) save_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) save_state_to_file() (easydel.utils.checkpoint_managers.__init__.CheckpointManager static method) (easydel.utils.checkpoint_managers.streamer.CheckpointManager static method) save_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) save_total_limit (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) SC (in module easydel.layers.attention) scale_embedding (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoder attribute) ScaledDotProductAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.scaled_dot_product) scaling_factor (easydel.layers.rotary_embedding.DeepseekScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.DynamicNTKScalingRotaryEmbedding attribute) (easydel.layers.rotary_embedding.Llama3RotaryEmbedding attribute) (easydel.layers.rotary_embedding.YaRNScalingRotaryEmbedding attribute) scaling_factors (easydel.layers.rotary_embedding.LinearScalingRotaryEmbedding attribute) scan_layers (easydel.__init__.InternLM2Model attribute) (easydel.modules.internlm2.__init__.InternLM2Model attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Model attribute) scan_ring_attention (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] schedule_decodes (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] schedule_prefill (easydel.layers.caching.paged_attention.__init__.NextIterationPlan attribute), [1] scheduler (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) score (easydel.__init__.InternLM2ForSequenceClassification attribute) (easydel.__init__.MixtralForSequenceClassification attribute) (easydel.__init__.Olmo2ForSequenceClassification attribute) (easydel.__init__.Qwen2ForSequenceClassification attribute) (easydel.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.__init__.Qwen3ForSequenceClassification attribute) (easydel.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.internlm2.__init__.InternLM2ForSequenceClassification attribute) (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2ForSequenceClassification attribute) (easydel.modules.mixtral.__init__.MixtralForSequenceClassification attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralForSequenceClassification attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2ForSequenceClassification attribute) (easydel.modules.qwen2.__init__.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2ForSequenceClassification attribute) (easydel.modules.qwen2_moe.__init__.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeForSequenceClassification attribute) (easydel.modules.qwen3.__init__.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3ForSequenceClassification attribute) (easydel.modules.qwen3_moe.__init__.Qwen3MoeForSequenceClassification attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeForSequenceClassification attribute) scores (easydel.infra.mixins.generation.BeamSearchState attribute), [1] (easydel.infra.modeling_outputs.BeamSearchOutput attribute) SDPA (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] segment_sum() (in module easydel.modules.mamba2.modeling_mamba2_flax) SegmentIds (class in easydel.kernels.tpu_ops.ring_attention_pallas._utils) segments (easydel.inference.vwhisper.server.TranscriptionResponse attribute) self_attn (easydel.modules.mixtral.modeling_mixtral_flax.MixtralDecoderLayer attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoDecoderLayer attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2DecoderLayer attribute) (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.phi.modeling_phi_flax.PhiDecoderLayer attribute) (easydel.modules.phi3.modeling_phi3_flax.Phi3DecoderLayer attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoeDecoderLayer attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2DecoderLayer attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeDecoderLayer attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3DecoderLayer attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeDecoderLayer attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) self_attn_layer_norm (easydel.modules.opt.modeling_opt_flax.OPTDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperDecoderLayer attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperEncoderLayer attribute) Seq2SeqLMOutput (class in easydel.infra.modeling_outputs) Seq2SeqModelOutput (class in easydel.infra.modeling_outputs) Seq2SeqQuestionAnsweringModelOutput (class in easydel.infra.modeling_outputs) Seq2SeqSequenceClassifierOutput (class in easydel.infra.modeling_outputs) seqlen_offset (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) sequence_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) sequence_axis_name (easydel.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] SEQUENCE_CLASSIFICATION (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] SEQUENCE_DIM_MAPPING (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) sequence_length (easydel.inference.vinference.metrics.ModelMetadata attribute), [1] (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) sequence_parallel_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) SEQUENCE_TO_SEQUENCE (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] SequenceClassifierOutput (class in easydel.infra.modeling_outputs) sequences (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.infra.mixins.generation.BeamSearchState attribute), [1] (easydel.infra.mixins.generation.GreedyState attribute), [1] (easydel.infra.mixins.generation.SampleState attribute), [1] (easydel.infra.modeling_outputs.BeamSearchOutput attribute) (easydel.infra.modeling_outputs.GreedySearchOutput attribute) (easydel.infra.modeling_outputs.SampleOutput attribute) SERVE (easydel.infra.base_config.EasyMethod attribute) set_attrs_carefully() (easydel.__init__.AttentionMetadata method) (easydel.layers.attention_operator.__init__.AttentionMetadata method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata method) set_attrs_smartly() (in module easydel.infra.base_config) set_decoder() (easydel.__init__.OPTForCausalLM method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) set_exception() (easydel.inference.vsurge.utils.AsyncMultifuture method) set_input_embeddings() (easydel.__init__.OPTForCausalLM method) (easydel.__init__.OPTModel method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.__init__.OPTModel method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTModel method) set_loggers_level() (in module easydel.utils.helpers) set_module_from_path() (in module easydel.utils.graph_utils) set_output_embeddings() (easydel.__init__.OPTForCausalLM method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) set_postfix() (easydel.trainers.trainer_protocol.BaseProgressBar method) (easydel.trainers.trainer_protocol.JSONProgressBar method) (easydel.trainers.trainer_protocol.NullProgressBar method) (easydel.trainers.trainer_protocol.RichProgressBar method) (easydel.trainers.trainer_protocol.TqdmProgressBar method) setdefault() (easydel.infra.modeling_outputs.ModelOutput method) SFTConfig (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.supervised_fine_tuning_trainer.__init__) (class in easydel.trainers.supervised_fine_tuning_trainer.sft_config) SFTTrainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.supervised_fine_tuning_trainer.__init__) (class in easydel.trainers.supervised_fine_tuning_trainer.sft_trainer) shard_attention_prod() (easydel.__init__.AttentionModule method) (easydel.layers.attention.AttentionModule method) shard_fns (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) shard_model() (easydel.__init__.EasyDeLBaseModule method) (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.base_state.EasyDeLState method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) shard_optimizer_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) shard_state() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) shard_with_shape() (easydel.__init__.EasyDeLState method) (easydel.infra.__init__.EasyDeLState method) (easydel.infra.base_state.EasyDeLState method) sharded_evaluation_step_function (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput attribute) sharded_training_step_function (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput attribute) shardings (easydel.__init__.EasyDeLState property) (easydel.infra.__init__.EasyDeLState property) (easydel.infra.base_state.EasyDeLState property) shift_and_pad() (in module easydel.trainers.utils) shift_tokens (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] shift_tokens_right() (in module easydel.modules.whisper.modeling_whisper_flax) short_factor (easydel.layers.rotary_embedding.Phi3LongRoPEScaledRotaryEmbedding attribute) (easydel.layers.rotary_embedding.RopeConfig attribute), [1] short_mscale (easydel.layers.rotary_embedding.RopeConfig attribute), [1] should_buffer_response() (easydel.__init__.vSurge method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) shuffle_train_dataset (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) SiglipAttention (class in easydel.modules.siglip.modeling_siglip_flax) SiglipConfig (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.configuration_siglip) SiglipEncoder (class in easydel.modules.siglip.modeling_siglip_flax) SiglipEncoderLayer (class in easydel.modules.siglip.modeling_siglip_flax) SiglipForImageClassification (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.modeling_siglip_flax)

SiglipMLP (class in easydel.modules.siglip.modeling_siglip_flax) SiglipModel (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.modeling_siglip_flax) SiglipMultiheadAttentionPoolingHead (class in easydel.modules.siglip.modeling_siglip_flax) SiglipOutput (class in easydel.modules.siglip.modeling_siglip_flax) SiglipTextConfig (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.configuration_siglip) SiglipTextEmbeddings (class in easydel.modules.siglip.modeling_siglip_flax) SiglipTextModel (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.modeling_siglip_flax) SiglipTextModelOutput (class in easydel.modules.siglip.modeling_siglip_flax) SiglipTextTransformer (class in easydel.modules.siglip.modeling_siglip_flax) SiglipVisionConfig (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.configuration_siglip) SiglipVisionEmbeddings (class in easydel.modules.siglip.modeling_siglip_flax) SiglipVisionModel (class in easydel.__init__) (class in easydel.modules.siglip.__init__) (class in easydel.modules.siglip.modeling_siglip_flax) SiglipVisionModelOutput (class in easydel.modules.siglip.modeling_siglip_flax) SiglipVisionTransformer (class in easydel.modules.siglip.modeling_siglip_flax) sigmoid_cross_entropy_with_logits() (in module easydel.infra.loss_utils) single_dequantize_nf4() (in module easydel.layers.quantization._nf4_quantizer) (in module easydel.layers.quantization.linear_nf4) single_quantize_and_pack_nf4() (in module easydel.layers.quantization._nf4_quantizer) (in module easydel.layers.quantization.linear_nf4) SingleStandRwkvBlock (class in easydel.modules.rwkv.modeling_rwkv_flax) sinusoidal_embedding_init() (in module easydel.modules.whisper.modeling_whisper_flax) size (easydel.__init__.EasyDeLState property) (easydel.infra.__init__.EasyDeLState property) (easydel.infra.base_state.EasyDeLState property) skip_apply_chat_template (easydel.__init__.GRPOConfig attribute) (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) slot (easydel.layers.caching.paged_attention.__init__.GenerationStepTask attribute), [1] (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment attribute), [1] SlotPageAssignment (class in easydel.layers.caching.paged_attention.__init__) smart_compile() (in module easydel.utils.compiling_utils) SMPMemoryMonitor (class in easydel.utils.analyze_memory) softmax_scale (easydel.__init__.AttentionMetadata attribute), [1] (easydel.kernels.flash_attention.AttentionConfig attribute) (easydel.layers.attention_operator.__init__.AttentionMetadata attribute), [1] (easydel.layers.attention_operator._attention_impl.AttentionMetadata attribute), [1] (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) sparse_module_type (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) sparsify_module (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) SpecialLossNormalizingFactor (class in easydel.infra.loss_utils) specs_to_name_sharding() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) (in module easydel.utils.traversals) SPEECH_SEQUENCE_TO_SEQUENCE (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] SPLASH (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] SplashAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.splash) split_lora_params() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (in module easydel.infra.utils) split_module() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) split_params() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) split_params_dict() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) srt (easydel.inference.vwhisper.server.ResponseFormat attribute) ssm_state_size (easydel.layers.caching.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData attribute) ssm_states (easydel.layers.caching.__init__.Mamba2CacheView attribute) (easydel.layers.caching.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.__init__.MambaCacheView attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView attribute) StableLmAttention (class in easydel.modules.stablelm.modeling_stablelm_flax) StableLmConfig (class in easydel.__init__) (class in easydel.modules.stablelm.__init__) (class in easydel.modules.stablelm.stablelm_configuration) StableLmDecoderLayer (class in easydel.modules.stablelm.modeling_stablelm_flax) StableLmForCausalLM (class in easydel.__init__) (class in easydel.modules.stablelm.__init__) (class in easydel.modules.stablelm.modeling_stablelm_flax) StableLmLayerNormPerHead (class in easydel.modules.stablelm.modeling_stablelm_flax) StableLmMLP (class in easydel.modules.stablelm.modeling_stablelm_flax) StableLmModel (class in easydel.__init__) (class in easydel.modules.stablelm.__init__) (class in easydel.modules.stablelm.modeling_stablelm_flax) start() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.__init__.vSurge method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor method) (easydel.utils.__init__.Timer method) (easydel.utils.analyze_memory.MemoryMonitorServer method) (easydel.utils.helpers.Timer method) start_client() (in module easydel.utils.analyze_memory) start_evaluation_hook() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) start_logits (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput attribute) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput attribute) start_monitoring() (easydel.utils.analyze_memory.MemoryMonitorClient method) (easydel.utils.analyze_memory.SMPMemoryMonitor method) start_server() (in module easydel.utils.analyze_memory) start_step() (easydel.trainers.trainer_protocol.StepMetrics method) start_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) (easydel.inference.vsurge.vsurge.vSurgeMetadata attribute) start_training_hook() (easydel.__init__.BaseTrainer method) (easydel.trainers.__init__.BaseTrainer method) (easydel.trainers.base_trainer.BaseTrainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) starts (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) state (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput attribute) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer.modeling_output.TrainerOutput attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerOutput attribute) state_apply_fn_kwarguments_to_model (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) state_named_sharding (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) state_partition_spec (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) state_shape (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) state_size (easydel.layers.caching.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData attribute) StateValidationResult (class in easydel.utils.traversals) static_arguments (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) step (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_state.EasyDeLState attribute), [1] step_partition_spec (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) step_start_point (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) StepMetrics (class in easydel.trainers.trainer_protocol) stop (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) stop() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.__init__.vSurge method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.__init__.vSurge method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method) (easydel.inference.vsurge.__init__.vSurge method) (easydel.inference.vsurge.vsurge.vSurge method) (easydel.utils.__init__.Timer method) (easydel.utils.helpers.Timer method) stop_monitoring() (easydel.utils.analyze_memory.MemoryMonitorClient method) (easydel.utils.analyze_memory.SMPMemoryMonitor method) StoreTupleAction (class in easydel.infra.etils) stream (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) streaming_chunks (easydel.__init__.vInferenceConfig attribute), [1] (easydel.inference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.__init__.vInferenceConfig attribute), [1] (easydel.inference.vinference.utilities.vInferenceConfig attribute), [1] string_key_to_int() (in module easydel.utils.traversals) string_to_bool() (in module easydel.utils.cli_helpers) sub_configs (easydel.__init__.AyaVisionConfig attribute) (easydel.__init__.CLIPConfig attribute) (easydel.__init__.Gemma3Config attribute) (easydel.__init__.Llama4Config attribute) (easydel.__init__.LlavaConfig attribute) (easydel.__init__.Qwen2VLConfig attribute) (easydel.__init__.SiglipConfig attribute) (easydel.modules.aya_vision.__init__.AyaVisionConfig attribute) (easydel.modules.aya_vision.aya_vision_configuration.AyaVisionConfig attribute) (easydel.modules.clip.__init__.CLIPConfig attribute) (easydel.modules.clip.clip_configuration.CLIPConfig attribute) (easydel.modules.gemma3.__init__.Gemma3Config attribute) (easydel.modules.gemma3.gemma3_configuration.Gemma3Config attribute) (easydel.modules.llama4.__init__.Llama4Config attribute) (easydel.modules.llama4.llama4_configuration.Llama4Config attribute) (easydel.modules.llava.__init__.LlavaConfig attribute) (easydel.modules.llava.llava_configuration.LlavaConfig attribute) (easydel.modules.qwen2_vl.__init__.Qwen2VLConfig attribute) (easydel.modules.qwen2_vl.qwen2_vl_configuration.Qwen2VLConfig attribute) (easydel.modules.siglip.__init__.SiglipConfig attribute) (easydel.modules.siglip.configuration_siglip.SiglipConfig attribute) submit_request() (easydel.__init__.oDriver method) (easydel.__init__.vDriver method) (easydel.inference.__init__.oDriver method) (easydel.inference.__init__.vDriver method) (easydel.inference.vsurge.__init__.oDriver method) (easydel.inference.vsurge.__init__.vDriver method) summary (easydel.inference.vinference.api_server.EndpointConfig attribute) (easydel.inference.vsurge.api_server.EndpointConfig attribute) supported_tasks (easydel.utils.readme_generator.ModelInfo attribute) suppress_tokens (easydel.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.logits_process.SuppressTokensLogitsProcessor attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] SuppressTokensAtBeginLogitsProcessor (class in easydel.inference.logits_process) SuppressTokensLogitsProcessor (class in easydel.inference.logits_process) sync_ref_model (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute)

T

tags (easydel.inference.vinference.api_server.EndpointConfig attribute) (easydel.inference.vsurge.api_server.EndpointConfig attribute) take_nearest_length() (in module easydel.inference.vsurge.utils) task (easydel.__init__.vWhisperInferenceConfig attribute) (easydel.inference.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig attribute) (easydel.inference.vwhisper.config.vWhisperInferenceConfig attribute) task_registry (easydel.infra.factory.Registry property) TaskType (class in easydel.__init__) (class in easydel.infra.factory) temperature (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.TemperatureLogitsWarper attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) (easydel.layers.caching.paged_attention.__init__.SamplingParams attribute), [1] TemperatureLogitsWarper (class in easydel.inference.logits_process) tensor_parallel_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) TensorParallelConfig (class in easydel.layers.linear) test_all_gather_matmul() (in module easydel.kernels.collective_matmul) test_bwd_call() (in module easydel.kernels.rms_norm) test_deepseek_yarn_scaling_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_dot_general_replacer() (in module easydel.kernels.matmul) test_dynamic_ntk_scaling_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_fwd_call() (in module easydel.kernels.rms_norm) test_get_rope() (in module easydel.layers.rotary_embedding_test) test_linear_scaling_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_llama3_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_phi3_long_rope_scaled_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_recurrent_gla() (in module easydel.layers.ops._gla) test_reduce_scatter_matmul() (in module easydel.kernels.collective_matmul) test_rotary_embedding() (in module easydel.layers.rotary_embedding_test) test_yarn_scaling_rotary_embedding() (in module easydel.layers.rotary_embedding_test) text (easydel.inference.openai_api_modules.CompletionResponseChoice attribute) (easydel.inference.openai_api_modules.CompletionStreamResponseChoice attribute) (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.vinference.PromptOutput attribute) (easydel.inference.vsurge.utils.ReturnSample attribute), [1] (easydel.inference.vwhisper.server.ResponseFormat attribute) (easydel.inference.vwhisper.server.TranscriptionResponse attribute) text_embeds (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.infra.modeling_outputs.CLIPTextModelOutput attribute) (easydel.infra.modeling_outputs.ImageClassifierOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput attribute) text_model_output (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) text_offset (easydel.inference.openai_api_modules.CompletionLogprobs attribute) text_tokens_to_string() (in module easydel.inference.vsurge.utils) timed() (easydel.utils.__init__.Timers method) (easydel.utils.helpers.Timers method) Timer (class in easydel.utils.__init__) (class in easydel.utils.helpers) timer (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) Timers (class in easydel.utils.__init__) (class in easydel.utils.helpers) timestamp (easydel.utils.analyze_memory.DeviceStats attribute), [1] to_dict() (easydel.__init__.AttentionMetadata method) (easydel.__init__.DPOConfig method) (easydel.__init__.EasyDeLBaseConfig method) (easydel.__init__.GRPOConfig method) (easydel.__init__.LossConfig method) (easydel.__init__.ORPOConfig method) (easydel.__init__.RewardConfig method) (easydel.__init__.SamplingParams method) (easydel.__init__.SFTConfig method) (easydel.__init__.TrainingArguments method) (easydel.__init__.vInferenceConfig method) (easydel.__init__.vInferencePreCompileConfig method) (easydel.__init__.vWhisperInferenceConfig method) (easydel.inference.__init__.SamplingParams method) (easydel.inference.__init__.vInferenceConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vWhisperInferenceConfig method) (easydel.inference.logits_process.EmptyProcessor method) (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor method) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor method) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor method) (easydel.inference.logits_process.MinLengthLogitsProcessor method) (easydel.inference.logits_process.MinPLogitsWarper method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor method) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor method) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensLogitsProcessor method) (easydel.inference.logits_process.TemperatureLogitsWarper method) (easydel.inference.logits_process.TopKLogitsWarper method) (easydel.inference.logits_process.TopPLogitsWarper method) (easydel.inference.utilities.SamplingParams method) (easydel.inference.vinference.__init__.SampleState method) (easydel.inference.vinference.__init__.vInferenceConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.api_server.EndpointConfig method) (easydel.inference.vinference.metrics.ModelMetadata method) (easydel.inference.vinference.utilities.SampleState method) (easydel.inference.vinference.utilities.vInferenceConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) (easydel.inference.vsurge.api_server.EndpointConfig method) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig method) (easydel.inference.vwhisper.config.vWhisperInferenceConfig method) (easydel.infra.__init__.EasyDeLBaseConfig method) (easydel.infra.__init__.LossConfig method) (easydel.infra.base_config.EasyDeLBaseConfig method) (easydel.infra.base_config.EasyMethod method) (easydel.infra.factory.ModuleRegistration method) (easydel.infra.loss_utils.LossConfig method) (easydel.infra.loss_utils.LossMetrics method) (easydel.infra.mixins.generation.BeamSearchState method) (easydel.infra.mixins.generation.GreedyState method) (easydel.infra.mixins.generation.SampleState method) (easydel.infra.modeling_outputs.AttentionLayerOutput method) (easydel.infra.modeling_outputs.BaseModelOutput method) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention method) (easydel.infra.modeling_outputs.BaseModelOutputWithPast method) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention method) (easydel.infra.modeling_outputs.BeamSearchOutput method) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions method) (easydel.infra.modeling_outputs.CLIPOutput method) (easydel.infra.modeling_outputs.CLIPTextModelOutput method) (easydel.infra.modeling_outputs.DecoderLayerOutput method) (easydel.infra.modeling_outputs.EncoderLayerOutput method) (easydel.infra.modeling_outputs.GreedySearchOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention method) (easydel.infra.modeling_outputs.MambaCausalLMOutput method) (easydel.infra.modeling_outputs.MambaOutput method) (easydel.infra.modeling_outputs.MaskedLMOutput method) (easydel.infra.modeling_outputs.MoeCausalLMOutput method) (easydel.infra.modeling_outputs.MoeModelOutput method) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput method) (easydel.infra.modeling_outputs.NextSentencePredictorOutput method) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.SampleOutput method) (easydel.infra.modeling_outputs.Seq2SeqLMOutput method) (easydel.infra.modeling_outputs.Seq2SeqModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput method) (easydel.infra.modeling_outputs.SequenceClassifierOutput method) (easydel.infra.modeling_outputs.TokenClassifierOutput method) (easydel.kernels.flash_attention.AttentionConfig method) (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes method) (easydel.layers.attention_operator.__init__.AttentionMetadata method) (easydel.layers.attention_operator.__init__.AttentionOutput method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata method) (easydel.layers.attention_operator._attention_impl.AttentionOutput method) (easydel.layers.caching.__init__.LightningCache method) (easydel.layers.caching.__init__.LightningCacheMetaData method) (easydel.layers.caching.__init__.LightningCacheView method) (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheMetaData method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.__init__.PagedAttentionCache method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.__init__.PagedAttentionMetadata method) (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.__init__.TransformerCacheMetaData method) (easydel.layers.caching.__init__.TransformerCacheView method) (easydel.layers.caching.__init__.TransformerMetadata method) (easydel.layers.caching._abstracts.BaseCacheMetadata method) (easydel.layers.caching._abstracts.BaseRunTimeMetadata method) (easydel.layers.caching.lightning.__init__.LightningCache method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData method) (easydel.layers.caching.lightning.__init__.LightningCacheView method) (easydel.layers.caching.lightning.lightning_cache.LightningCache method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages method) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch method) (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary method) (easydel.layers.caching.paged_attention.__init__.NextIterationPlan method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.__init__.SamplingParams method) (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData method) (easydel.layers.caching.transformer.__init__.TransformerCacheView method) (easydel.layers.caching.transformer.__init__.TransformerMetadata method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView method) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata method) (easydel.layers.linear.TensorParallelConfig method) (easydel.layers.rotary_embedding.RopeConfig method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast method) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast method) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast method) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput method) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput method) (easydel.trainers.__init__.DPOConfig method) (easydel.trainers.__init__.GRPOConfig method) (easydel.trainers.__init__.ORPOConfig method) (easydel.trainers.__init__.RewardConfig method) (easydel.trainers.__init__.SFTConfig method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig method) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig method) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig method) (easydel.trainers.reward_trainer.__init__.RewardConfig method) (easydel.trainers.reward_trainer.reward_config.RewardConfig method) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig method) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig method) (easydel.trainers.trainer.modeling_output.TrainerOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput method) (easydel.trainers.trainer_protocol.TrainerOutput method) (easydel.trainers.training_configurations.TrainingArguments method) (easydel.trainers.utils.DataCollatorForPreference method) (easydel.trainers.utils.DPODataCollatorWithPadding method) (easydel.trainers.utils.RewardDataCollatorWithPadding method) (easydel.utils.analyze_memory.DeviceStats method) (easydel.utils.traversals.StateValidationResult method) to_dtype() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) to_json() (easydel.__init__.AttentionMetadata method) (easydel.__init__.DPOConfig method) (easydel.__init__.GRPOConfig method) (easydel.__init__.LossConfig method) (easydel.__init__.ORPOConfig method) (easydel.__init__.RewardConfig method) (easydel.__init__.SamplingParams method) (easydel.__init__.SFTConfig method) (easydel.__init__.TrainingArguments method) (easydel.__init__.vInferenceConfig method) (easydel.__init__.vInferencePreCompileConfig method) (easydel.__init__.vWhisperInferenceConfig method) (easydel.inference.__init__.SamplingParams method) (easydel.inference.__init__.vInferenceConfig method) (easydel.inference.__init__.vInferencePreCompileConfig method) (easydel.inference.__init__.vWhisperInferenceConfig method) (easydel.inference.logits_process.EmptyProcessor method) (easydel.inference.logits_process.ForcedBOSTokenLogitsProcessor method) (easydel.inference.logits_process.ForcedEOSTokenLogitsProcessor method) (easydel.inference.logits_process.FrequencyPenaltyLogitsProcessor method) (easydel.inference.logits_process.MinLengthLogitsProcessor method) (easydel.inference.logits_process.MinPLogitsWarper method) (easydel.inference.logits_process.NoRepeatNGramLogitsProcessor method) (easydel.inference.logits_process.PresencePenaltyLogitsProcessor method) (easydel.inference.logits_process.RepetitionPenaltyLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensAtBeginLogitsProcessor method) (easydel.inference.logits_process.SuppressTokensLogitsProcessor method) (easydel.inference.logits_process.TemperatureLogitsWarper method) (easydel.inference.logits_process.TopKLogitsWarper method) (easydel.inference.logits_process.TopPLogitsWarper method) (easydel.inference.utilities.SamplingParams method) (easydel.inference.vinference.__init__.SampleState method) (easydel.inference.vinference.__init__.vInferenceConfig method) (easydel.inference.vinference.__init__.vInferencePreCompileConfig method) (easydel.inference.vinference.api_server.EndpointConfig method) (easydel.inference.vinference.metrics.ModelMetadata method) (easydel.inference.vinference.utilities.SampleState method) (easydel.inference.vinference.utilities.vInferenceConfig method) (easydel.inference.vinference.utilities.vInferencePreCompileConfig method) (easydel.inference.vsurge.api_server.EndpointConfig method) (easydel.inference.vwhisper.__init__.vWhisperInferenceConfig method) (easydel.inference.vwhisper.config.vWhisperInferenceConfig method) (easydel.infra.__init__.LossConfig method) (easydel.infra.base_config.EasyMethod method) (easydel.infra.factory.ModuleRegistration method) (easydel.infra.loss_utils.LossConfig method) (easydel.infra.loss_utils.LossMetrics method) (easydel.infra.mixins.generation.BeamSearchState method) (easydel.infra.mixins.generation.GreedyState method) (easydel.infra.mixins.generation.SampleState method) (easydel.infra.modeling_outputs.AttentionLayerOutput method) (easydel.infra.modeling_outputs.BaseModelOutput method) (easydel.infra.modeling_outputs.BaseModelOutputWithNoAttention method) (easydel.infra.modeling_outputs.BaseModelOutputWithPast method) (easydel.infra.modeling_outputs.BaseModelOutputWithPastAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPooling method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndCrossAttentions method) (easydel.infra.modeling_outputs.BaseModelOutputWithPoolingAndNoAttention method) (easydel.infra.modeling_outputs.BeamSearchOutput method) (easydel.infra.modeling_outputs.CausalLMOutputWithCrossAttentions method) (easydel.infra.modeling_outputs.CLIPOutput method) (easydel.infra.modeling_outputs.CLIPTextModelOutput method) (easydel.infra.modeling_outputs.DecoderLayerOutput method) (easydel.infra.modeling_outputs.EncoderLayerOutput method) (easydel.infra.modeling_outputs.GreedySearchOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutput method) (easydel.infra.modeling_outputs.ImageClassifierOutputWithNoAttention method) (easydel.infra.modeling_outputs.MambaCausalLMOutput method) (easydel.infra.modeling_outputs.MambaOutput method) (easydel.infra.modeling_outputs.MaskedLMOutput method) (easydel.infra.modeling_outputs.MoeCausalLMOutput method) (easydel.infra.modeling_outputs.MoeModelOutput method) (easydel.infra.modeling_outputs.MultipleChoiceModelOutput method) (easydel.infra.modeling_outputs.NextSentencePredictorOutput method) (easydel.infra.modeling_outputs.QuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.SampleOutput method) (easydel.infra.modeling_outputs.Seq2SeqLMOutput method) (easydel.infra.modeling_outputs.Seq2SeqModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqQuestionAnsweringModelOutput method) (easydel.infra.modeling_outputs.Seq2SeqSequenceClassifierOutput method) (easydel.infra.modeling_outputs.SequenceClassifierOutput method) (easydel.infra.modeling_outputs.TokenClassifierOutput method) (easydel.kernels.flash_attention.AttentionConfig method) (easydel.kernels.tpu_ops.ring_attention_pallas._utils.BlockSizes method) (easydel.layers.attention_operator.__init__.AttentionMetadata method) (easydel.layers.attention_operator.__init__.AttentionOutput method) (easydel.layers.attention_operator._attention_impl.AttentionMetadata method) (easydel.layers.attention_operator._attention_impl.AttentionOutput method) (easydel.layers.caching.__init__.LightningCache method) (easydel.layers.caching.__init__.LightningCacheMetaData method) (easydel.layers.caching.__init__.LightningCacheView method) (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheMetaData method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.__init__.PagedAttentionCache method) (easydel.layers.caching.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.__init__.PagedAttentionMetadata method) (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.__init__.TransformerCacheMetaData method) (easydel.layers.caching.__init__.TransformerCacheView method) (easydel.layers.caching.__init__.TransformerMetadata method) (easydel.layers.caching._abstracts.BaseCacheMetadata method) (easydel.layers.caching._abstracts.BaseRunTimeMetadata method) (easydel.layers.caching.lightning.__init__.LightningCache method) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData method) (easydel.layers.caching.lightning.__init__.LightningCacheView method) (easydel.layers.caching.lightning.lightning_cache.LightningCache method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData method) (easydel.layers.caching.lightning.lightning_cache.LightningCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheMetaData method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheMetaData method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheMetaData method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch method) (easydel.layers.caching.paged_attention.__init__.AllocatedPrefillPages method) (easydel.layers.caching.paged_attention.__init__.GenerationStepTask method) (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest method) (easydel.layers.caching.paged_attention.__init__.ModelOutputBatch method) (easydel.layers.caching.paged_attention.__init__.ModelOutputSummary method) (easydel.layers.caching.paged_attention.__init__.NextIterationPlan method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionMetadata method) (easydel.layers.caching.paged_attention.__init__.SamplingParams method) (easydel.layers.caching.paged_attention.__init__.SlotPageAssignment method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheMetaData method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionMetadata method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData method) (easydel.layers.caching.transformer.__init__.TransformerCacheView method) (easydel.layers.caching.transformer.__init__.TransformerMetadata method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData method) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView method) (easydel.layers.caching.transformer.transformer_cache.TransformerMetadata method) (easydel.layers.linear.TensorParallelConfig method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionCausalLMOutputWithPast method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3CausalLMOutputWithPast method) (easydel.modules.llama4.modeling_llama4_flax.Llama4CausalLMOutputWithPast method) (easydel.modules.llava.modeling_llava_flax.LlavaCausalLMOutputWithPast method) (easydel.modules.mamba.modeling_mamba_flax.MambaCausalLMOutput method) (easydel.modules.mamba.modeling_mamba_flax.MambaOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2CausalLMOutput method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2Output method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLCausalLMOutputWithPast method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvCausalLMOutput method) (easydel.modules.rwkv.modeling_rwkv_flax.RwkvOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipTextModelOutput method) (easydel.modules.siglip.modeling_siglip_flax.SiglipVisionModelOutput method) (easydel.trainers.__init__.DPOConfig method) (easydel.trainers.__init__.GRPOConfig method) (easydel.trainers.__init__.ORPOConfig method) (easydel.trainers.__init__.RewardConfig method) (easydel.trainers.__init__.SFTConfig method) (easydel.trainers.__init__.TrainingArguments method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig method) (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig method) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig method) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOConfig method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_config.ORPOConfig method) (easydel.trainers.reward_trainer.__init__.RewardConfig method) (easydel.trainers.reward_trainer.reward_config.RewardConfig method) (easydel.trainers.supervised_fine_tuning_trainer.__init__.SFTConfig method) (easydel.trainers.supervised_fine_tuning_trainer.sft_config.SFTConfig method) (easydel.trainers.trainer.modeling_output.TrainerOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureDataloaderOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureFunctionOutput method) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput method) (easydel.trainers.trainer_protocol.TrainerOutput method) (easydel.trainers.training_configurations.TrainingArguments method) (easydel.trainers.utils.DataCollatorForPreference method) (easydel.trainers.utils.DPODataCollatorWithPadding method) (easydel.trainers.utils.RewardDataCollatorWithPadding method) (easydel.utils.analyze_memory.DeviceStats method) (easydel.utils.traversals.StateValidationResult method)

to_linear() (easydel.layers.quantization.__init__.Linear8bit method) (easydel.layers.quantization.__init__.LinearNF4 method) (easydel.layers.quantization.linear_8bit.Linear8bit method) (easydel.layers.quantization.linear_nf4.LinearNF4 method) to_pure() (easydel.layers.caching.__init__.TransformerCache method) (easydel.layers.caching.transformer.__init__.TransformerCache method) (easydel.layers.caching.transformer.transformer_cache.TransformerCache method) to_state() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) to_torch() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) to_tuple() (easydel.infra.modeling_outputs.CLIPOutput method) (easydel.infra.modeling_outputs.ModelOutput method) (easydel.layers.rotary_embedding.RopeConfig method) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput method) token_embeddings (easydel.__init__.OpenELMModel attribute) (easydel.modules.openelm.__init__.OpenELMModel attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMModel attribute) token_ids (easydel.inference.vsurge.utils.ReturnSample attribute), [1] (easydel.layers.caching.paged_attention.__init__.ActiveSequenceBatch attribute), [1] (easydel.layers.caching.paged_attention.__init__.InitialSequenceRequest attribute), [1] token_logprobs (easydel.inference.openai_api_modules.CompletionLogprobs attribute) token_throughput (easydel.inference.vinference.metrics.vInferenceMetrics attribute) TokenClassifierOutput (class in easydel.infra.modeling_outputs) tokenize_and_pad() (in module easydel.inference.vsurge.utils) tokenize_row() (easydel.__init__.DPOTrainer static method) (easydel.__init__.ORPOTrainer method) (easydel.trainers.__init__.DPOTrainer static method) (easydel.trainers.__init__.ORPOTrainer method) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer static method) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer static method) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer method) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer method) tokenizer (easydel.__init__.vInference property) (easydel.inference.__init__.vInference property) (easydel.inference.vinference.__init__.vInference property) (easydel.inference.vinference.vinference.vInference property) (easydel.trainers.utils.RewardDataCollatorWithPadding attribute) tokens (easydel.inference.openai_api_modules.CompletionLogprobs attribute) tokens_per_second (easydel.inference.openai_api_modules.UsageInfo attribute) (easydel.inference.vinference.__init__.PromptOutput attribute) (easydel.inference.vinference.__init__.SampleState attribute), [1] (easydel.inference.vinference.utilities.SampleState attribute), [1] (easydel.inference.vinference.vinference.PromptOutput attribute) (easydel.inference.vsurge.utils.ReturnSample attribute), [1] tolist() (in module easydel.trainers.utils) tool_choice (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) ToolDefinition (class in easydel.inference.openai_api_modules) tools (easydel.__init__.DPOConfig attribute), [1] (easydel.__init__.GRPOConfig attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.GRPOConfig attribute) (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] (easydel.trainers.group_relative_policy_optimization.__init__.GRPOConfig attribute) (easydel.trainers.group_relative_policy_optimization.grpo_config.GRPOConfig attribute) top_k (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.TopKLogitsWarper attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) top_logprobs (easydel.inference.openai_api_modules.CompletionLogprobs attribute) top_p (easydel.__init__.SamplingParams attribute), [1] (easydel.__init__.vSurgeRequest attribute) (easydel.inference.__init__.SamplingParams attribute), [1] (easydel.inference.__init__.vSurgeRequest attribute) (easydel.inference.logits_process.TopPLogitsWarper attribute) (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) (easydel.inference.utilities.SamplingParams attribute), [1] (easydel.inference.vsurge.__init__.vSurgeRequest attribute) (easydel.inference.vsurge.utils.ActiveRequest attribute) (easydel.inference.vsurge.vsurge.vSurgeRequest attribute) (easydel.layers.caching.paged_attention.__init__.SamplingParams attribute), [1] TopKLogitsWarper (class in easydel.inference.logits_process) TopPLogitsWarper (class in easydel.inference.logits_process) torch_dict_to_easydel_params() (in module easydel.__init__) (in module easydel.utils.parameters_transformation) total_batch_size (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) total_generated_tokens (easydel.inference.vsurge.utils.ActiveRequest attribute) total_tokens (easydel.inference.openai_api_modules.UsageInfo attribute) TPU (easydel.__init__.EasyDeLBackends attribute), [1] (easydel.infra.etils.EasyDeLBackends attribute), [1] (easydel.kernels.flash_attention.Backend attribute) tpu_version_check() (in module easydel.layers.attention) TqdmProgressBar (class in easydel.trainers.trainer_protocol) trace_compilation() (easydel.infra.utils.CompilationTracker method) trace_functions() (in module easydel.infra.utils) TraceResult (class in easydel.infra.utils) track_compilation() (easydel.inference.vinference.metrics.vInferenceMetrics method) track_memory (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) TRAIN (easydel.infra.base_config.EasyMethod attribute) train() (easydel.__init__.Trainer method) (easydel.trainers.__init__.Trainer method) (easydel.trainers.trainer.__init__.Trainer method) (easydel.trainers.trainer.trainer.Trainer method) (easydel.trainers.trainer_protocol.BaseTrainerProtocol method) train_on_inputs (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) train_tracker (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) Trainer (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.trainer.__init__) (class in easydel.trainers.trainer.trainer) TrainerConfigureDataloaderOutput (class in easydel.trainers.trainer_protocol) TrainerConfigureFunctionOutput (class in easydel.trainers.trainer_protocol) TrainerConfigureModelOutput (class in easydel.trainers.trainer_protocol) TrainerOutput (class in easydel.trainers.trainer.modeling_output) (class in easydel.trainers.trainer_protocol) training_batch_size (easydel.__init__.BaseTrainer property) (easydel.trainers.__init__.BaseTrainer property) (easydel.trainers.base_trainer.BaseTrainer property) (easydel.trainers.trainer_protocol.BaseTrainerProtocol property) training_step() (in module easydel.trainers.direct_preference_optimization_trainer._fn) (in module easydel.trainers.reward_trainer._fn) (in module easydel.trainers.trainer._fn) training_time_limit (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) training_time_seconds (easydel.__init__.TrainingArguments property) (easydel.trainers.__init__.TrainingArguments property) (easydel.trainers.training_configurations.TrainingArguments property) TrainingArguments (class in easydel.__init__) (class in easydel.trainers.__init__) (class in easydel.trainers.training_configurations) TranscriptionResponse (class in easydel.inference.vwhisper.server) transfer_dequeue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) transfer_enqueue_time (easydel.inference.vsurge.utils.ActiveRequestMetadata attribute) transform_fn (easydel.__init__.EasyDeLBaseModule property) (easydel.infra.__init__.EasyDeLBaseModule property) (easydel.infra.base_module.EasyDeLBaseModule property) (easydel.infra.mixins.__init__.BaseModuleProtocol property) (easydel.infra.mixins.protocol.BaseModuleProtocol property) transformer (easydel.__init__.MptForCausalLM attribute) (easydel.__init__.OlmoForCausalLM attribute) (easydel.__init__.OpenELMForCausalLM attribute) (easydel.__init__.PhiForCausalLM attribute) (easydel.__init__.PixtralVisionModel attribute) (easydel.modules.mosaic_mpt.__init__.MptForCausalLM attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForCausalLM attribute) (easydel.modules.olmo.__init__.OlmoForSequenceClassification attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForCausalLM attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoForSequenceClassification attribute) (easydel.modules.openelm.__init__.OpenELMForCausalLM attribute) (easydel.modules.openelm.modeling_openelm_flax.OpenELMForCausalLM attribute) (easydel.modules.phi.__init__.PhiForCausalLM attribute) (easydel.modules.phi.modeling_phi_flax.PhiForCausalLM attribute) (easydel.modules.pixtral.__init__.PixtralVisionModel attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralVisionModel attribute) transformer_dim (easydel.modules.openelm.modeling_openelm_flax.OpenELMMultiHeadCausalAttention attribute) TransformerCache (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.transformer.__init__) (class in easydel.layers.caching.transformer.transformer_cache) TransformerCacheMetaData (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.transformer.__init__) (class in easydel.layers.caching.transformer.transformer_cache) TransformerCacheView (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.transformer.__init__) (class in easydel.layers.caching.transformer.transformer_cache) TransformerMetadata (class in easydel.layers.caching.__init__) (class in easydel.layers.caching.transformer.__init__) (class in easydel.layers.caching.transformer.transformer_cache) tree_apply() (in module easydel.utils.traversals) tree_path_to_string() (in module easydel.utils.traversals) TRITON (easydel.__init__.EasyDeLPlatforms attribute), [1] (easydel.infra.etils.EasyDeLPlatforms attribute), [1] (easydel.kernels.flash_attention.Platform attribute) triton_flash_attention() (in module easydel.kernels.gpu_ops.__init__) (in module easydel.kernels.gpu_ops.flash_attention_triton.__init__) truncate_right() (in module easydel.trainers.utils) truncation_mode (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) (easydel.trainers.utils.RewardDataCollatorWithPadding attribute) tx (easydel.__init__.EasyDeLState attribute), [1] (easydel.infra.__init__.EasyDeLState attribute), [1] (easydel.infra.base_state.EasyDeLState attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) (easydel.trainers.trainer_protocol.TrainerConfigureModelOutput attribute) tx_mu_dtype (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) type (easydel.inference.openai_api_modules.ToolDefinition attribute) (easydel.utils.readme_generator.ModelInfo attribute)

U

unflatten_dict() (in module easydel.utils.traversals) unpair_preference_dataset() (in module easydel.trainers.prompt_utils) unwrap_lora_to_layers() (easydel.__init__.EasyDeLBaseModule method) (easydel.infra.__init__.EasyDeLBaseModule method) (easydel.infra.base_module.EasyDeLBaseModule method) (easydel.infra.mixins.__init__.BaseModuleProtocol method) (easydel.infra.mixins.protocol.BaseModuleProtocol method) (in module easydel.infra.utils) up_proj (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptMLP attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoMLP attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2MLP attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralMLP attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2MLP attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeMLP attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3MLP attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeMLP attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmMLP attribute) update() (easydel.infra.modeling_outputs.ModelOutput method) (easydel.trainers.trainer_protocol.BaseProgressBar method) (easydel.trainers.trainer_protocol.JSONProgressBar method) (easydel.trainers.trainer_protocol.MetricsTracker method) (easydel.trainers.trainer_protocol.NullProgressBar method) (easydel.trainers.trainer_protocol.RichProgressBar method) (easydel.trainers.trainer_protocol.TqdmProgressBar method) update_causal_mask (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) update_conv_state() (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) update_inputs_for_generation() (easydel.__init__.AyaVisionForConditionalGeneration method) (easydel.__init__.Gemma3ForConditionalGeneration method) (easydel.__init__.Llama4ForConditionalGeneration method) (easydel.__init__.LlavaForConditionalGeneration method) (easydel.__init__.Mamba2ForCausalLM method) (easydel.__init__.MambaForCausalLM method) (easydel.__init__.OPTForCausalLM method) (easydel.__init__.Qwen2VLForConditionalGeneration method) (easydel.__init__.WhisperForConditionalGeneration method) (easydel.infra.mixins.__init__.EasyGenerationMixin method) (easydel.infra.mixins.generation.EasyGenerationMixin method) (easydel.modules.aya_vision.__init__.AyaVisionForConditionalGeneration method) (easydel.modules.aya_vision.modeling_aya_vision_flax.AyaVisionForConditionalGeneration method) (easydel.modules.gemma3.__init__.Gemma3ForConditionalGeneration method) (easydel.modules.gemma3.modeling_gemma3_flax.Gemma3ForConditionalGeneration method) (easydel.modules.llama4.__init__.Llama4ForConditionalGeneration method) (easydel.modules.llama4.modeling_llama4_flax.Llama4ForConditionalGeneration method) (easydel.modules.llava.__init__.LlavaForConditionalGeneration method) (easydel.modules.llava.modeling_llava_flax.LlavaForConditionalGeneration method) (easydel.modules.mamba.__init__.MambaForCausalLM method) (easydel.modules.mamba.modeling_mamba_flax.MambaForCausalLM method) (easydel.modules.mamba2.__init__.Mamba2ForCausalLM method) (easydel.modules.mamba2.modeling_mamba2_flax.Mamba2ForCausalLM method) (easydel.modules.opt.__init__.OPTForCausalLM method) (easydel.modules.opt.modeling_opt_flax.OPTForCausalLM method) (easydel.modules.qwen2_vl.__init__.Qwen2VLForConditionalGeneration method) (easydel.modules.qwen2_vl.modeling_qwen2_vl_flax.Qwen2VLForConditionalGeneration method) (easydel.modules.whisper.__init__.WhisperForConditionalGeneration method) (easydel.modules.whisper.modeling_whisper_flax.WhisperForConditionalGeneration method)

update_metrics() (in module easydel.trainers.training_utils) update_seq() (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) update_ssm_state() (easydel.layers.caching.__init__.Mamba2Cache method) (easydel.layers.caching.__init__.Mamba2CacheView method) (easydel.layers.caching.__init__.MambaCache method) (easydel.layers.caching.__init__.MambaCacheView method) (easydel.layers.caching.mamba.__init__.MambaCache method) (easydel.layers.caching.mamba.__init__.MambaCacheView method) (easydel.layers.caching.mamba.mamba_cache.MambaCache method) (easydel.layers.caching.mamba.mamba_cache.MambaCacheView method) (easydel.layers.caching.mamba2.__init__.Mamba2Cache method) (easydel.layers.caching.mamba2.__init__.Mamba2CacheView method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache method) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2CacheView method) update_state_respectfully() (in module easydel.trainers.training_utils) usage (easydel.inference.openai_api_modules.ChatCompletionResponse attribute) (easydel.inference.openai_api_modules.ChatCompletionStreamResponse attribute) (easydel.inference.openai_api_modules.CompletionResponse attribute) (easydel.inference.openai_api_modules.CompletionStreamResponse attribute) UsageInfo (class in easydel.inference.openai_api_modules) use_bias (easydel.layers.linear.ParallelLinear attribute) (easydel.modules.gpt2.modeling_gpt2_flax.Conv1D attribute) use_data_collactor (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) use_wandb (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) use_weighting (easydel.__init__.DPOConfig attribute), [1] (easydel.trainers.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOConfig attribute), [1] (easydel.trainers.direct_preference_optimization_trainer.dpo_config.DPOConfig attribute), [1] user (easydel.inference.openai_api_modules.ChatCompletionRequest attribute) (easydel.inference.openai_api_modules.CompletionRequest attribute) utilization_percent (easydel.utils.analyze_memory.DeviceStats attribute), [1] uuid4 (easydel.inference.vinference.vinference.vInferenceMetaData attribute)

V

v_proj (easydel.modules.mixtral.modeling_mixtral_flax.MixtralAttention attribute) (easydel.modules.olmo.modeling_olmo_flax.OlmoAttention attribute) (easydel.modules.olmo2.modeling_olmo2_flax.Olmo2Attention attribute) (easydel.modules.opt.modeling_opt_flax.OPTAttention attribute) (easydel.modules.phi.modeling_phi_flax.PhiAttention attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEAttention attribute) (easydel.modules.pixtral.modeling_pixtral_flax.PixtralAttention attribute) (easydel.modules.qwen2.modeling_qwen_flax.Qwen2Attention attribute) (easydel.modules.qwen2_moe.modeling_qwen2_moe_flax.Qwen2MoeAttention attribute) (easydel.modules.qwen3.modeling_qwen3_flax.Qwen3Attention attribute) (easydel.modules.qwen3_moe.modeling_qwen3_moe_flax.Qwen3MoeAttention attribute) (easydel.modules.stablelm.modeling_stablelm_flax.StableLmAttention attribute) (easydel.modules.whisper.modeling_whisper_flax.WhisperAttention attribute) validate_state() (in module easydel.utils.traversals) value (easydel.layers.caching.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheView attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheView attribute) value_dim (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) value_heads (easydel.layers.caching.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.lightning.__init__.LightningCacheMetaData attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCacheMetaData attribute) (easydel.layers.caching.transformer.__init__.TransformerCacheMetaData attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCacheMetaData attribute) value_pages (easydel.layers.caching.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView attribute), [1] values() (easydel.layers.rotary_embedding.RopeConfig method) VANILLA (easydel.__init__.AttentionMechanisms attribute), [1] (easydel.layers.attention.AttentionMechanisms attribute), [1] VanillaAttn (class in easydel.layers.attention_operator.__init__) (class in easydel.layers.attention_operator.modules.__init__) (class in easydel.layers.attention_operator.modules.vanilla) vDriver (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) vEngine (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) verbose (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) verbose_json (easydel.inference.vwhisper.server.ResponseFormat attribute) version (easydel.utils.readme_generator.ModelInfo attribute) views (easydel.layers.caching.__init__.LightningCache attribute) (easydel.layers.caching.__init__.Mamba2Cache attribute) (easydel.layers.caching.__init__.MambaCache attribute) (easydel.layers.caching.__init__.PagedAttentionCache attribute), [1] (easydel.layers.caching.__init__.TransformerCache attribute) (easydel.layers.caching._abstracts.BaseCache attribute) (easydel.layers.caching.lightning.__init__.LightningCache attribute) (easydel.layers.caching.lightning.lightning_cache.LightningCache attribute) (easydel.layers.caching.mamba.__init__.MambaCache attribute) (easydel.layers.caching.mamba.mamba_cache.MambaCache attribute) (easydel.layers.caching.mamba2.__init__.Mamba2Cache attribute) (easydel.layers.caching.mamba2.mamba2_cache.Mamba2Cache attribute) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCache attribute), [1] (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCache attribute), [1] (easydel.layers.caching.transformer.__init__.TransformerCache attribute) (easydel.layers.caching.transformer.transformer_cache.TransformerCache attribute) vInference (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.vinference) vInferenceApiServer (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.api_server)

vInferenceConfig (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.utilities) vInferenceMetaData (class in easydel.inference.vinference.vinference) vInferenceMetrics (class in easydel.inference.vinference.metrics) vInferencePreCompileConfig (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vinference.__init__) (class in easydel.inference.vinference.utilities) vision_apply_rotary_emb() (in module easydel.modules.llama4.modeling_llama4_flax) vision_batch_size (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] vision_channels (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] vision_height (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] vision_included (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] VISION_LM (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1] vision_model_output (easydel.infra.modeling_outputs.CLIPOutput attribute) (easydel.modules.siglip.modeling_siglip_flax.SiglipOutput attribute) vision_width (easydel.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.__init__.vInferencePreCompileConfig attribute), [1] (easydel.inference.vinference.utilities.vInferencePreCompileConfig attribute), [1] VisionAttention (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) VisionLlamaConfig (class in easydel.modules.llama.__init__) (class in easydel.modules.llama.llama_configuration) VisionMlp (class in easydel.modules.qwen2_vl.modeling_qwen2_vl_flax) vocab_axis (easydel.__init__.PartitionAxis attribute) (easydel.infra.__init__.PartitionAxis attribute) vSurge (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) (class in easydel.inference.vsurge.vsurge) vsurge_name (easydel.__init__.vSurge property) (easydel.inference.__init__.vSurge property) (easydel.inference.vsurge.__init__.vSurge property) (easydel.inference.vsurge.vsurge.vSurge property) vSurgeApiServer (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) (class in easydel.inference.vsurge.api_server) vSurgeMetadata (class in easydel.inference.vsurge.vsurge) vSurgeRequest (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vsurge.__init__) (class in easydel.inference.vsurge.vsurge) vtt (easydel.inference.vwhisper.server.ResponseFormat attribute) vWhisperInference (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vwhisper.__init__) (class in easydel.inference.vwhisper.core) vWhisperInferenceConfig (class in easydel.__init__) (class in easydel.inference.__init__) (class in easydel.inference.vwhisper.__init__) (class in easydel.inference.vwhisper.config)

W

w1 (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) w2 (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) w3 (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2MLP attribute) (easydel.modules.mixtral.modeling_mixtral_flax.MixtralBLockSparseTop2MLP attribute) (easydel.modules.phimoe.modeling_phimoe_flax.PhiMoEBlockSparseTop2MLP attribute) wait_and_get_loaded() (easydel.kernels.tpu_ops.paged_attention_pallas._forward_pallas.MultiPageAsyncCopyDescriptor method) wandb_entity (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) wandb_runtime (easydel.trainers.direct_preference_optimization_trainer.__init__.DPOTrainer attribute) (easydel.trainers.direct_preference_optimization_trainer.dpo_trainer.DPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.__init__.GRPOTrainer attribute) (easydel.trainers.group_relative_policy_optimization.grpo_trainer.GRPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.__init__.ORPOTrainer attribute) (easydel.trainers.odds_ratio_preference_optimization_trainer.orpo_trainer.ORPOTrainer attribute) (easydel.trainers.trainer_protocol.BaseTrainerProtocol attribute) warmup_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) weight_decay (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) weight_distribution_log_steps (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) weight_distribution_pattern (easydel.__init__.TrainingArguments attribute) (easydel.trainers.__init__.TrainingArguments attribute) (easydel.trainers.training_configurations.TrainingArguments attribute) weight_sum (easydel.infra.loss_utils.LossMetrics attribute), [1] WhisperAttention (class in easydel.modules.whisper.modeling_whisper_flax)

WhisperConfig (class in easydel.__init__) (class in easydel.modules.whisper.__init__) (class in easydel.modules.whisper.whisper_configuration) WhisperDecoder (class in easydel.modules.whisper.modeling_whisper_flax) WhisperDecoderLayer (class in easydel.modules.whisper.modeling_whisper_flax) WhisperEncoder (class in easydel.modules.whisper.modeling_whisper_flax) WhisperEncoderLayer (class in easydel.modules.whisper.modeling_whisper_flax) WhisperForAudioClassification (class in easydel.__init__) (class in easydel.modules.whisper.__init__) (class in easydel.modules.whisper.modeling_whisper_flax) WhisperForConditionalGeneration (class in easydel.__init__) (class in easydel.modules.whisper.__init__) (class in easydel.modules.whisper.modeling_whisper_flax) WhisperModel (class in easydel.inference.vwhisper.__init__) (class in easydel.inference.vwhisper.server) (class in easydel.modules.whisper.modeling_whisper_flax) WhisperTimeStampLogitsProcessor (class in easydel.__init__) (class in easydel.inference.logits_process) (class in easydel.modules.whisper.__init__) wo (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) wqkv (easydel.modules.internlm2.modeling_internlm2_flax.InternLM2Attention attribute) Wqkv (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptAttention attribute) write() (easydel.utils.__init__.Timers method) (easydel.utils.helpers.DummyStream method) (easydel.utils.helpers.Timers method) write_decodes_to_cache() (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) write_prefill_to_cache() (easydel.layers.caching.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.__init__.PagedAttentionCacheView method) (easydel.layers.caching.paged_attention.paged_attention_cache.PagedAttentionCacheView method) wte (easydel.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.__init__.MptModel attribute) (easydel.modules.mosaic_mpt.modeling_mosaic_flax.MptModel attribute)

X

Xerxes2Attention (class in easydel.modules.xerxes2.modeling_xerxes2_flax) Xerxes2Config (class in easydel.__init__) (class in easydel.modules.xerxes2.__init__) (class in easydel.modules.xerxes2.xerxes2_configuration) Xerxes2DecoderLayer (class in easydel.modules.xerxes2.modeling_xerxes2_flax) Xerxes2ForCausalLM (class in easydel.__init__) (class in easydel.modules.xerxes2.__init__) (class in easydel.modules.xerxes2.modeling_xerxes2_flax) Xerxes2MLP (class in easydel.modules.xerxes2.modeling_xerxes2_flax) Xerxes2Model (class in easydel.__init__) (class in easydel.modules.xerxes2.__init__) (class in easydel.modules.xerxes2.modeling_xerxes2_flax)

XerxesAttention (class in easydel.modules.xerxes.modeling_xerxes_flax) XerxesConfig (class in easydel.__init__) (class in easydel.modules.xerxes.__init__) (class in easydel.modules.xerxes.xerxes_configuration) XerxesDecoderLayer (class in easydel.modules.xerxes.modeling_xerxes_flax) XerxesForCausalLM (class in easydel.__init__) (class in easydel.modules.xerxes.__init__) (class in easydel.modules.xerxes.modeling_xerxes_flax) XerxesMLP (class in easydel.modules.xerxes.modeling_xerxes_flax) XerxesModel (class in easydel.__init__) (class in easydel.modules.xerxes.__init__) (class in easydel.modules.xerxes.modeling_xerxes_flax) XerxesSparseMoeBlock (class in easydel.modules.xerxes.modeling_xerxes_flax)

Y

yarn_find_correction_dim() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) yarn_find_correction_range() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) yarn_get_mscale() (in module easydel.layers.rotary_embedding) (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax)

yarn_linear_ramp_mask() (in module easydel.modules.deepseek_v2.modeling_deepseek_flax) (in module easydel.modules.deepseek_v3.modeling_deepseek_flax) YaRNScalingRotaryEmbedding (class in easydel.layers.rotary_embedding)

Z

z_loss (easydel.__init__.LossConfig attribute), [1] (easydel.infra.__init__.LossConfig attribute), [1] (easydel.infra.loss_utils.LossConfig attribute), [1] (easydel.infra.loss_utils.LossMetrics attribute), [1]

ZERO_SHOT_IMAGE_CLASSIFICATION (easydel.__init__.TaskType attribute), [1] (easydel.infra.factory.TaskType attribute), [1]