LICENSE
README.md
pyproject.toml
veomni/__init__.py
veomni/_version.py
veomni.egg-info/PKG-INFO
veomni.egg-info/SOURCES.txt
veomni.egg-info/dependency_links.txt
veomni.egg-info/requires.txt
veomni.egg-info/top_level.txt
veomni/arguments/__init__.py
veomni/arguments/arguments_types.py
veomni/arguments/parser.py
veomni/checkpoint/__init__.py
veomni/checkpoint/checkpointer.py
veomni/checkpoint/dcp_checkpointer.py
veomni/checkpoint/dcp_consolidation.py
veomni/data/__init__.py
veomni/data/chat_template.py
veomni/data/data_collator.py
veomni/data/data_loader.py
veomni/data/data_transform.py
veomni/data/dataset.py
veomni/data/dummy_dataset.py
veomni/data/dynamic_batching.py
veomni/data/diffusion/data_loader.py
veomni/data/diffusion/dataset.py
veomni/data/diffusion/preprocess/origin_dataset.py
veomni/data/multimodal/__init__.py
veomni/data/multimodal/audio_utils.py
veomni/data/multimodal/image_utils.py
veomni/data/multimodal/preprocess.py
veomni/data/multimodal/video_utils.py
veomni/data/multimodal/dit/data_transform.py
veomni/data/multimodal/dit/preprocess.py
veomni/distributed/__init__.py
veomni/distributed/async_offload.py
veomni/distributed/checkpoint.py
veomni/distributed/clip_grad_norm.py
veomni/distributed/offloading.py
veomni/distributed/parallel_plan.py
veomni/distributed/parallel_state.py
veomni/distributed/torch_compile.py
veomni/distributed/torch_parallelize.py
veomni/distributed/utils.py
veomni/distributed/context_parallel/__init__.py
veomni/distributed/context_parallel/dsa_cp.py
veomni/distributed/context_parallel/sharding.py
veomni/distributed/fsdp2/__init__.py
veomni/distributed/fsdp2/clip_grad_norm.py
veomni/distributed/moe/__init__.py
veomni/distributed/moe/comm.py
veomni/distributed/moe/moe_layer.py
veomni/distributed/moe/moe_utils.py
veomni/distributed/sequence_parallel/__init__.py
veomni/distributed/sequence_parallel/async_ulysses.py
veomni/distributed/sequence_parallel/async_ulysses_dit.py
veomni/distributed/sequence_parallel/backward.py
veomni/distributed/sequence_parallel/comm.py
veomni/distributed/sequence_parallel/data.py
veomni/distributed/sequence_parallel/loss.py
veomni/distributed/sequence_parallel/op_wrappers.py
veomni/distributed/sequence_parallel/ulysses.py
veomni/distributed/sequence_parallel/utils.py
veomni/lora/__init__.py
veomni/lora/config.py
veomni/lora/layers.py
veomni/lora/mapping.py
veomni/lora/model.py
veomni/lora/moe_layers.py
veomni/lora/state_dict.py
veomni/lora/target_mapping.py
veomni/lora/weight_loading.py
veomni/lora/ops/__init__.py
veomni/lora/ops/moe_group_gemm.py
veomni/lora/ops/npu_moe_group_gemm.py
veomni/models/__init__.py
veomni/models/_moe_fused_weight_map.py
veomni/models/auto.py
veomni/models/checkpoint_tensor_loading.py
veomni/models/loader.py
veomni/models/module_utils.py
veomni/models/diffusers/__init__.py
veomni/models/diffusers/ltx2_3/__init__.py
veomni/models/diffusers/ltx2_3/ltx_condition/__init__.py
veomni/models/diffusers/ltx2_3/ltx_condition/captioning.py
veomni/models/diffusers/ltx2_3/ltx_condition/configuration_ltx2_3_condition.py
veomni/models/diffusers/ltx2_3/ltx_condition/modeling_ltx2_3_condition.py
veomni/models/diffusers/ltx2_3/ltx_condition/preprocess_dataset.py
veomni/models/diffusers/ltx2_3/ltx_core/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/types.py
veomni/models/diffusers/ltx2_3/ltx_core/utils.py
veomni/models/diffusers/ltx2_3/ltx_core/components/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/components/patchifiers.py
veomni/models/diffusers/ltx2_3/ltx_core/components/protocols.py
veomni/models/diffusers/ltx2_3/ltx_core/guidance/perturbations.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/fuse_loras.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/helpers.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/module_ops.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/primitives.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/registry.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/sd_ops.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/sft_loader.py
veomni/models/diffusers/ltx2_3/ltx_core/loader/single_gpu_model_builder.py
veomni/models/diffusers/ltx2_3/ltx_core/model/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/model/model_protocol.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/attention.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/audio_vae.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/causal_conv_2d.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/causality_axis.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/downsample.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/model_configurator.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/ops.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/resnet.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/upsample.py
veomni/models/diffusers/ltx2_3/ltx_core/model/audio_vae/vocoder.py
veomni/models/diffusers/ltx2_3/ltx_core/model/common/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/model/common/normalization.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/adaln.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/attention.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/feed_forward.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/gelu_approx.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/modality.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/model.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/model_configurator.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/ops.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/rope.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/text_projection.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/timestep_embedding.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/transformer.py
veomni/models/diffusers/ltx2_3/ltx_core/model/transformer/transformer_args.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/convolution.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/enums.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/model_configurator.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/normalization.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/ops.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/resnet.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/sampling.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/tiling.py
veomni/models/diffusers/ltx2_3/ltx_core/model/video_vae/video_vae.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/__init__.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/config.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/embeddings_connector.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/embeddings_processor.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/feature_extractor.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/tokenizer.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/encoders/base_encoder.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/encoders/encoder_configurator.py
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/encoders/prompts/gemma_i2v_system_prompt.txt
veomni/models/diffusers/ltx2_3/ltx_core/text_encoders/gemma/encoders/prompts/gemma_t2v_system_prompt.txt
veomni/models/diffusers/ltx2_3/ltx_transformer/__init__.py
veomni/models/diffusers/ltx2_3/ltx_transformer/configuration_ltx2_3_transformer.py
veomni/models/diffusers/ltx2_3/ltx_transformer/modeling_ltx2_3_transformer.py
veomni/models/diffusers/minimax_h3/__init__.py
veomni/models/diffusers/minimax_h3/inference.py
veomni/models/diffusers/minimax_h3/minimax_h3_condition/__init__.py
veomni/models/diffusers/minimax_h3/minimax_h3_condition/configuration_minimax_h3_condition.py
veomni/models/diffusers/minimax_h3/minimax_h3_condition/modeling_minimax_h3_condition.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/core.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/flow_match_scheduler.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/minimax_h3_audio_vae.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/minimax_h3_dit.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/minimax_h3_text_encoder.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/minimax_h3_video_vae.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/offline_loader.py
veomni/models/diffusers/minimax_h3/minimax_h3_core/packed_sequence.py
veomni/models/diffusers/minimax_h3/minimax_h3_transformer/__init__.py
veomni/models/diffusers/minimax_h3/minimax_h3_transformer/configuration_minimax_h3_transformer.py
veomni/models/diffusers/minimax_h3/minimax_h3_transformer/modeling_minimax_h3_transformer.py
veomni/models/diffusers/qwen_image/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_condition/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_condition/configuration_qwen_image_condition.py
veomni/models/diffusers/qwen_image/qwen_image_condition/modeling_qwen_image_condition.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/configuration_qwen_image_transformer.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/modeling_qwen_image_transformer.py
veomni/models/diffusers/wan_t2v/__init__.py
veomni/models/diffusers/wan_t2v/wan_condition/__init__.py
veomni/models/diffusers/wan_t2v/wan_condition/configuration_wan_condition.py
veomni/models/diffusers/wan_t2v/wan_condition/modeling_wan_condition.py
veomni/models/diffusers/wan_t2v/wan_transformer/__init__.py
veomni/models/diffusers/wan_t2v/wan_transformer/configuration_wan_transformer.py
veomni/models/diffusers/wan_t2v/wan_transformer/modeling_wan_transformer.py
veomni/models/transformers/__init__.py
veomni/models/transformers/attention_utils.py
veomni/models/transformers/masking_utils.py
veomni/models/transformers/deepseek_v3/__init__.py
veomni/models/transformers/deepseek_v3/checkpoint_tensor_converter.py
veomni/models/transformers/deepseek_v3/deepseek_v3_gpu_patch_gen_config.py
veomni/models/transformers/deepseek_v3/deepseek_v3_npu_patch_gen_config.py
veomni/models/transformers/deepseek_v3/device_patch.py
veomni/models/transformers/deepseek_v3/parallel_plan.py
veomni/models/transformers/deepseek_v3/generated/patched_modeling_deepseek_v3_gpu.py
veomni/models/transformers/deepseek_v3/generated/patched_modeling_deepseek_v3_npu.py
veomni/models/transformers/deepseek_v4/__init__.py
veomni/models/transformers/deepseek_v4/checkpoint_tensor_converter.py
veomni/models/transformers/deepseek_v4/configuration_deepseek_v4.py
veomni/models/transformers/deepseek_v4/deepseek_v4_gpu_patch_gen_config.py
veomni/models/transformers/deepseek_v4/deepseek_v4_npu_patch_gen_config.py
veomni/models/transformers/deepseek_v4/device_patch.py
veomni/models/transformers/deepseek_v4/packed_utils.py
veomni/models/transformers/deepseek_v4/parallel_plan.py
veomni/models/transformers/deepseek_v4/generated/__init__.py
veomni/models/transformers/deepseek_v4/generated/patched_modeling_deepseek_v4_gpu.py
veomni/models/transformers/deepseek_v4/generated/patched_modeling_deepseek_v4_npu.py
veomni/models/transformers/flux/__init__.py
veomni/models/transformers/flux/config_flux.py
veomni/models/transformers/flux/encode_flux.py
veomni/models/transformers/flux/modeling_flux.py
veomni/models/transformers/flux/utils_flux.py
veomni/models/transformers/gemma3/__init__.py
veomni/models/transformers/gemma3/gemma3_gpu_patch_gen_config.py
veomni/models/transformers/gemma3/generated/patched_modeling_gemma3_gpu.py
veomni/models/transformers/glm_moe_dsa/__init__.py
veomni/models/transformers/glm_moe_dsa/glm_moe_dsa_gpu_patch_gen_config.py
veomni/models/transformers/glm_moe_dsa/glm_moe_dsa_npu_patch_gen_config.py
veomni/models/transformers/glm_moe_dsa/generated/patched_modeling_glm_moe_dsa_gpu.py
veomni/models/transformers/glm_moe_dsa/generated/patched_modeling_glm_moe_dsa_npu.py
veomni/models/transformers/gpt_oss/__init__.py
veomni/models/transformers/gpt_oss/gpt_oss_gpu_patch_gen_config.py
veomni/models/transformers/gpt_oss/parallel_plan.py
veomni/models/transformers/gpt_oss/generated/patched_modeling_gpt_oss_gpu.py
veomni/models/transformers/llama/__init__.py
veomni/models/transformers/llama/llama_gpu_patch_gen_config.py
veomni/models/transformers/llama/generated/patched_modeling_llama_gpu.py
veomni/models/transformers/movqgan/__init__.py
veomni/models/transformers/movqgan/configuration_movqgan.py
veomni/models/transformers/movqgan/modeling_movqgan.py
veomni/models/transformers/movqgan/processing_movqgan.py
veomni/models/transformers/qwen2/__init__.py
veomni/models/transformers/qwen2/qwen2_gpu_patch_gen_config.py
veomni/models/transformers/qwen2/generated/patched_modeling_qwen2_gpu.py
veomni/models/transformers/qwen2_5_omni/__init__.py
veomni/models/transformers/qwen2_5_omni/configuration_qwen2_5_omni.py
veomni/models/transformers/qwen2_5_omni/processing_qwen2_5_omni.py
veomni/models/transformers/qwen2_5_omni/qwen2_5_omni_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_5_omni/generated/patched_modeling_qwen2_5_omni_gpu.py
veomni/models/transformers/qwen2_5vl/__init__.py
veomni/models/transformers/qwen2_5vl/qwen2_5_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_5vl/generated/patched_modeling_qwen2_5_vl_gpu.py
veomni/models/transformers/qwen2_vl/__init__.py
veomni/models/transformers/qwen2_vl/qwen2_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_vl/generated/patched_modeling_qwen2_vl_gpu.py
veomni/models/transformers/qwen3/__init__.py
veomni/models/transformers/qwen3/qwen3_gpu_patch_gen_config.py
veomni/models/transformers/qwen3/qwen3_npu_patch_gen_config.py
veomni/models/transformers/qwen3/generated/patched_modeling_qwen3_gpu.py
veomni/models/transformers/qwen3/generated/patched_modeling_qwen3_npu.py
veomni/models/transformers/qwen3_5/__init__.py
veomni/models/transformers/qwen3_5/qwen3_5_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_5/qwen3_5_npu_patch_gen_config.py
veomni/models/transformers/qwen3_5/generated/patched_modeling_qwen3_5_gpu.py
veomni/models/transformers/qwen3_5/generated/patched_modeling_qwen3_5_npu.py
veomni/models/transformers/qwen3_5_moe/__init__.py
veomni/models/transformers/qwen3_5_moe/parallel_plan.py
veomni/models/transformers/qwen3_5_moe/qwen3_5_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_5_moe/qwen3_5_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_5_moe/generated/patched_modeling_qwen3_5_moe_gpu.py
veomni/models/transformers/qwen3_5_moe/generated/patched_modeling_qwen3_5_moe_npu.py
veomni/models/transformers/qwen3_moe/__init__.py
veomni/models/transformers/qwen3_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_moe/parallel_plan.py
veomni/models/transformers/qwen3_moe/qwen3_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_moe/qwen3_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_moe/generated/patched_modeling_qwen3_moe_gpu.py
veomni/models/transformers/qwen3_moe/generated/patched_modeling_qwen3_moe_npu.py
veomni/models/transformers/qwen3_omni_moe/__init__.py
veomni/models/transformers/qwen3_omni_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_omni_moe/configuration_qwen3_omni_moe.py
veomni/models/transformers/qwen3_omni_moe/parallel_plan.py
veomni/models/transformers/qwen3_omni_moe/processing_qwen3_omni_moe.py
veomni/models/transformers/qwen3_omni_moe/qwen3_omni_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_omni_moe/qwen3_omni_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_omni_moe/generated/patched_modeling_qwen3_omni_moe_gpu.py
veomni/models/transformers/qwen3_omni_moe/generated/patched_modeling_qwen3_omni_moe_npu.py
veomni/models/transformers/qwen3_vl/__init__.py
veomni/models/transformers/qwen3_vl/qwen3_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_vl/qwen3_vl_npu_patch_gen_config.py
veomni/models/transformers/qwen3_vl/generated/patched_modeling_qwen3_vl_gpu.py
veomni/models/transformers/qwen3_vl/generated/patched_modeling_qwen3_vl_npu.py
veomni/models/transformers/qwen3_vl_moe/__init__.py
veomni/models/transformers/qwen3_vl_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_vl_moe/parallel_plan.py
veomni/models/transformers/qwen3_vl_moe/qwen3_vl_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_vl_moe/qwen3_vl_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_vl_moe/generated/patched_modeling_qwen3_vl_moe_gpu.py
veomni/models/transformers/qwen3_vl_moe/generated/patched_modeling_qwen3_vl_moe_npu.py
veomni/models/transformers/seed_oss/__init__.py
veomni/models/transformers/seed_oss/seed_oss_gpu_patch_gen_config.py
veomni/models/transformers/seed_oss/seed_oss_npu_patch_gen_config.py
veomni/models/transformers/seed_oss/generated/patched_modeling_seed_oss_gpu.py
veomni/models/transformers/seed_oss/generated/patched_modeling_seed_oss_npu.py
veomni/models/transformers/wan/__init__.py
veomni/models/transformers/wan/config_wan.py
veomni/models/transformers/wan/device_patch.py
veomni/models/transformers/wan/modeling_wan.py
veomni/models/transformers/wan/npu_patch.py
veomni/ops/__init__.py
veomni/ops/dispatch.py
veomni/ops/kernel_registry.py
veomni/ops/batch_invariant_ops/__init__.py
veomni/ops/batch_invariant_ops/batch_invariant_ops.py
veomni/ops/config/__init__.py
veomni/ops/config/registry.py
veomni/ops/config/singleton.py
veomni/ops/kernels/__init__.py
veomni/ops/kernels/attention/__init__.py
veomni/ops/kernels/attention/flash.py
veomni/ops/kernels/attention/flex.py
veomni/ops/kernels/attention/ulysses.py
veomni/ops/kernels/attention/magi/__init__.py
veomni/ops/kernels/attention/magi/_fa4_cuda.py
veomni/ops/kernels/attention/magi/mask.py
veomni/ops/kernels/cross_entropy/__init__.py
veomni/ops/kernels/cross_entropy/chunk_logprobs.py
veomni/ops/kernels/cross_entropy/chunk_loss.py
veomni/ops/kernels/cross_entropy/chunk_topk_distill.py
veomni/ops/kernels/cross_entropy/eager.py
veomni/ops/kernels/cross_entropy/liger.py
veomni/ops/kernels/deepseek_sparse_attention/__init__.py
veomni/ops/kernels/deepseek_sparse_attention/flashmla_cudnn.py
veomni/ops/kernels/deepseek_v4/__init__.py
veomni/ops/kernels/deepseek_v4/precision_aligned_ops.py
veomni/ops/kernels/deepseek_v4/quant.py
veomni/ops/kernels/deepseek_v4/tilelang_indexer.py
veomni/ops/kernels/deepseek_v4/tilelang_indexer_bwd.py
veomni/ops/kernels/deepseek_v4/tilelang_indexer_fwd.py
veomni/ops/kernels/deepseek_v4/tilelang_sparse_mla.py
veomni/ops/kernels/deepseek_v4/tilelang_sparse_mla_bwd.py
veomni/ops/kernels/deepseek_v4/tilelang_sparse_mla_fwd.py
veomni/ops/kernels/deepseek_v4/tilelang_sparse_mla_target.py
veomni/ops/kernels/gated_delta_rule/__init__.py
veomni/ops/kernels/gated_delta_rule/npu_ascendc_gated_delta_rule.py
veomni/ops/kernels/gated_delta_rule/npu_causal_conv1d.py
veomni/ops/kernels/gated_delta_rule/npu_rms_norm_gated.py
veomni/ops/kernels/gated_delta_rule/_ascend/__init__.py
veomni/ops/kernels/gated_delta_rule/_ascend/causal_conv1d.py
veomni/ops/kernels/gated_delta_rule/_ascend/chunk_gated_delta_rule_mm.py
veomni/ops/kernels/gated_delta_rule/_ascend/flash_gated_delta_rule.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/__init__.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/chunk_delta_h.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/chunk_o.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/chunk_scaled_dot_kkt.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/convolution.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/cumsum.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/solve_tril.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/utils.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton/wy_fast.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton_core/__init__.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton_core/chunk_scaled_dot_kkt.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton_core/l2norm.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton_core/solve_tril.py
veomni/ops/kernels/gated_delta_rule/_ascend/triton_core/utils.py
veomni/ops/kernels/load_balancing_loss/__init__.py
veomni/ops/kernels/load_balancing_loss/eager.py
veomni/ops/kernels/load_balancing_loss/triton.py
veomni/ops/kernels/mhc/__init__.py
veomni/ops/kernels/mhc/tile_kernels.py
veomni/ops/kernels/moe/__init__.py
veomni/ops/kernels/moe/_scatter.py
veomni/ops/kernels/moe/group_gemm.py
veomni/ops/kernels/moe/mlu_group_gemm.py
veomni/ops/kernels/moe/npu_group_gemm.py
veomni/ops/kernels/moe/quack_gemm.py
veomni/ops/kernels/moe/quack_gemm_interleave_gate_up.py
veomni/ops/kernels/moe/_kernels/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/group_gemm.py
veomni/ops/kernels/moe/_kernels/kernel/moe.py
veomni/ops/kernels/moe/_kernels/kernel/npu_clamped_swiglu.py
veomni/ops/kernels/moe/_kernels/kernel/npu_group_gemm.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/activation.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/memory.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/utils.py
veomni/ops/kernels/moe/_kernels/utils/__init__.py
veomni/ops/kernels/moe/_kernels/utils/benchmark_utils.py
veomni/ops/kernels/moe/_kernels/utils/config.py
veomni/ops/kernels/moe/_kernels/utils/device.py
veomni/ops/kernels/moe/_kernels/utils/envvars.py
veomni/ops/kernels/moe/_kernels/utils/kernel.py
veomni/ops/kernels/moe/_kernels/utils/path.py
veomni/ops/kernels/moe/_kernels/utils/pretuned.py
veomni/ops/kernels/rms_norm/__init__.py
veomni/ops/kernels/rms_norm/npu.py
veomni/ops/kernels/rms_norm/triton_batch_invariant.py
veomni/ops/kernels/rotary/__init__.py
veomni/ops/kernels/rotary/npu.py
veomni/ops/kernels/rotary/triton_deepseek_v4.py
veomni/ops/kernels/rotary/triton_deterministic.py
veomni/ops/kernels/rotary/triton_wan.py
veomni/ops/kernels/swiglu/__init__.py
veomni/ops/liger/__init__.py
veomni/ops/platform/__init__.py
veomni/ops/platform/npu/__init__.py
veomni/ops/platform/npu/hccl_premul_sum.py
veomni/ops/qat/__init__.py
veomni/ops/qat/fp4_blockwise.py
veomni/ops/qat/fp8_blockwise.py
veomni/optim/__init__.py
veomni/optim/lr_scheduler.py
veomni/optim/muon.py
veomni/optim/optimizer.py
veomni/patchgen/__init__.py
veomni/patchgen/check_patchgen.py
veomni/patchgen/codegen.py
veomni/patchgen/patch_spec.py
veomni/patchgen/run_codegen.py
veomni/schedulers/flow_match.py
veomni/trainer/__init__.py
veomni/trainer/base.py
veomni/trainer/base_rl_trainer.py
veomni/trainer/dit_trainer.py
veomni/trainer/text_dpo_trainer.py
veomni/trainer/text_trainer.py
veomni/trainer/vlm_trainer.py
veomni/trainer/callbacks/__init__.py
veomni/trainer/callbacks/base.py
veomni/trainer/callbacks/channel_loss_callback.py
veomni/trainer/callbacks/checkpoint_callback.py
veomni/trainer/callbacks/evaluate_callback.py
veomni/trainer/callbacks/trace_callback.py
veomni/utils/__init__.py
veomni/utils/checkpoint_utils.py
veomni/utils/constants.py
veomni/utils/count_flops.py
veomni/utils/device.py
veomni/utils/dist_utils.py
veomni/utils/dit_utils.py
veomni/utils/env.py
veomni/utils/fs.py
veomni/utils/hdfs_io.py
veomni/utils/helper.py
veomni/utils/import_utils.py
veomni/utils/logging.py
veomni/utils/loss_utils.py
veomni/utils/model_outputs.py
veomni/utils/model_utils.py
veomni/utils/moe_monitor.py
veomni/utils/moe_router_replay.py
veomni/utils/multisource_utils.py
veomni/utils/recompute_utils.py
veomni/utils/registry.py
veomni/utils/save_safetensor_utils.py
veomni/utils/seqlen_pos_transform_utils.py
veomni/utils/data_balance/balance_sorting_algo.py
veomni/utils/data_balance/data_balance.py