Infra

• easydel.infra.__init__
  • EasyDeLBaseConfig
  • EasyDeLBaseConfigDict
  • EasyDeLBaseModule
  • EasyDeLState
  • LossConfig
• 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
  • FlaxBaseModelOutput
  • FlaxBaseModelOutputWithNoAttention
  • FlaxBaseModelOutputWithPast
  • FlaxBaseModelOutputWithPastAndCrossAttentions
  • FlaxBaseModelOutputWithPooling
  • FlaxBaseModelOutputWithPoolingAndCrossAttentions
  • FlaxBaseModelOutputWithPoolingAndNoAttention
  • FlaxBeamSearchOutput
  • FlaxCLIPOutput
  • FlaxCLIPTextModelOutput
  • FlaxCausalLMOutput
  • FlaxCausalLMOutputWithCrossAttentions
  • FlaxGreedySearchOutput
  • FlaxImageClassifierOutput
  • FlaxImageClassifierOutputWithNoAttention
  • FlaxMaskedLMOutput
  • FlaxMultipleChoiceModelOutput
  • FlaxNextSentencePredictorOutput
  • FlaxQuestionAnsweringModelOutput
  • FlaxSampleOutput
  • FlaxSeq2SeqLMOutput
  • FlaxSeq2SeqModelOutput
  • FlaxSeq2SeqQuestionAnsweringModelOutput
  • FlaxSeq2SeqSequenceClassifierOutput
  • FlaxSequenceClassifierOutput
  • FlaxTokenClassifierOutput
  • MambaCausalLMOutput
  • MambaOutput
  • ModelOutput
  • MoeCausalLMOutput
  • MoeModelOutput
• easydel.infra.utils
  • CompilationTracker
  • FunctionTracer
  • ModuleCaches
  • OverWriteWithGradient
  • TraceResult
  • add_start_docstrings()
  • apply_lora_to_layers()
  • apply_sparsity_to_params()
  • auto_remat()
  • block_wise_ffn()
  • canonicalize_dtype()
  • control_mlp_sharding()
  • count_flop_jaxpr()
  • extract_static_parameters()
  • get_dot_general_by_bits()
  • get_gradient_checkpoint_policy()
  • is_flatten()
  • merge_lora_params()
  • quantize_linear_layers()
  • quick_gelu()
  • split_lora_params()
  • trace_functions()
  • unwrap_lora_to_layers()