Index of /FlagScale_to_hygon/third_party/Megatron-LM/megatron/core/transformer/moe/

../
__pycache__/                                       10-Sep-2025 08:16                   -
README.md                                          10-Sep-2025 08:16               32802
__init__.py                                        10-Sep-2025 08:16                   0
experts.py                                         10-Sep-2025 08:16               47083
fused_a2a.py                                       10-Sep-2025 08:16                6604
grouped_gemm_util.py                               10-Sep-2025 08:16                 591
legacy_a2a_token_dispatcher.py                     10-Sep-2025 08:16               14908
moe_layer.py                                       10-Sep-2025 08:16                8118
moe_utils.py                                       10-Sep-2025 08:16               36053
router.py                                          10-Sep-2025 08:16               18873
shared_experts.py                                  10-Sep-2025 08:16               11397
token_dispatcher.py                                10-Sep-2025 08:16               45119
upcycling_utils.py                                 10-Sep-2025 08:16                8438