Infra

• easydel.infra.__init__
  • EasyDeLBaseConfig
  • EasyDeLBaseConfigDict
  • EasyDeLBaseModule
  • EasyDeLState
  • LossConfig
  • PartitionAxis
  • PyTree
  • auto_pytree()
• easydel.infra.base_config
  • EasyDeLBaseConfig
  • EasyDeLBaseConfigDict
  • EasyMethod
  • set_attrs_smartly()
• easydel.infra.base_module
  • EasyDeLBaseModule
• easydel.infra.base_state
  • EasyDeLState
• easydel.infra.errors
  • EasyDeLBlockWiseFFNError
  • EasyDeLBreakRequest
  • EasyDeLComputeError
  • EasyDeLNotImplementedFeatureError
  • EasyDeLProcessError
  • EasyDeLRuntimeError
  • EasyDeLSyntaxRuntimeError
  • EasyDeLTimerError
• easydel.infra.etils
  • EasyDeLBackends
  • EasyDeLGradientCheckPointers
  • EasyDeLOptimizers
  • EasyDeLPlatforms
  • EasyDeLQuantizationMethods
  • EasyDeLSchedulers
  • StoreTupleAction
  • define_flags_with_default()
• easydel.infra.factory
  • ConfigType
  • ModuleRegistration
  • Registry
  • TaskType
  • register_config()
  • register_module()
• easydel.infra.loss_utils
  • ForCausalLMLoss()
  • ForQuestionAnsweringLoss()
  • ForSequenceClassificationLoss()
  • ForTokenClassification()
  • LossConfig
  • LossMetrics
  • SpecialLossNormalizingFactor
  • auxiliary_load_balancing_loss_func()
  • compute_weighted_cross_entropy()
  • compute_weighted_cross_entropy_and_accuracy()
  • convert_special_loss_normalizing_factor_to_enum()
  • cross_entropy_loss_and_accuracy()
  • dynamic_cross_entropy_loss()
  • fixed_cross_entropy()
  • get_loss_normalizing_factor_and_weights()
  • onehot()
  • sigmoid_cross_entropy_with_logits()
• easydel.infra.loss_utils_test
• Mixins
  • easydel.infra.mixins.__init__
  • easydel.infra.mixins.bridge
  • easydel.infra.mixins.generation
  • easydel.infra.mixins.protocol
• easydel.infra.modeling_outputs
  • AttentionLayerOutput
  • BaseModelOutput
  • BaseModelOutputWithNoAttention
  • BaseModelOutputWithPast
  • BaseModelOutputWithPastAndCrossAttentions
  • BaseModelOutputWithPooling
  • BaseModelOutputWithPoolingAndCrossAttentions
  • BaseModelOutputWithPoolingAndNoAttention
  • BeamSearchOutput
  • CLIPOutput
  • CLIPTextModelOutput
  • CausalLMOutput
  • CausalLMOutputWithCrossAttentions
  • DecoderLayerOutput
  • EncoderLayerOutput
  • GreedySearchOutput
  • ImageClassifierOutput
  • ImageClassifierOutputWithNoAttention
  • MambaCausalLMOutput
  • MambaOutput
  • MaskedLMOutput
  • ModelOutput
  • MoeCausalLMOutput
  • MoeModelOutput
  • MultipleChoiceModelOutput
  • NextSentencePredictorOutput
  • QuestionAnsweringModelOutput
  • SampleOutput
  • Seq2SeqLMOutput
  • Seq2SeqModelOutput
  • Seq2SeqQuestionAnsweringModelOutput
  • Seq2SeqSequenceClassifierOutput
  • SequenceClassifierOutput
  • TokenClassifierOutput
• easydel.infra.utils
  • CompilationTracker
  • FunctionTracer
  • ModuleCaches
  • OverWriteWithGradient
  • TraceResult
  • add_start_docstrings()
  • apply_lora_to_layers()
  • apply_sparsity_to_params()
  • attention3d_sharding()
  • attention4d_sharding()
  • auto_remat()
  • block_wise_ffn()
  • canonicalize_dtype()
  • count_flop_jaxpr()
  • extract_static_parameters()
  • get_dot_general_by_bits()
  • get_gradient_checkpoint_policy()
  • is_flatten()
  • merge_lora_params()
  • mlp3d_sharding()
  • quantize_linear_layers()
  • quick_gelu()
  • runtime3d_sharding()
  • split_lora_params()
  • trace_functions()
  • unwrap_lora_to_layers()