Robert Shaw 683e3cb9c4 [ Misc ] fbgemm checkpoints (#6559)

2024-07-20 09:36:57 -07:00

..

configs

Unmark fused_moe config json file as executable (#5960)

2024-06-28 06:36:12 -07:00

__init__.py

[ Misc ] Refactor MoE to isolate Fp8 From Mixtral (#5970)

2024-07-02 21:54:35 +00:00

fused_moe.py

[ Misc ] Apply MoE Refactor to Deepseekv2 To Support Fp8 (#6417)

2024-07-13 20:03:58 -07:00

layer.py

[ Misc ] fbgemm checkpoints (#6559)

2024-07-20 09:36:57 -07:00

moe_pallas.py

[Hardware][TPU] Support MoE with Pallas GMM kernel (#6457)

2024-07-16 09:56:28 -07:00