[Model] use FusedMoE layer in Jamba

[avshalomman]

This PR replaces the Jamba-specific MoE layer impl with the standard vLLM FusedMoE layer.

[github-actions]

👋 Hi! Thank you for contributing to the vLLM project.
Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run fastcheck CI which consists a small and essential subset of CI tests to quickly catch errors. You can run other CI tests on top of default ones by unblocking the steps in your fast-check build on Buildkite UI.

Once the PR is approved and ready to go, please make sure to run full CI as it is required to merge (or just use auto-merge).

To run full CI, you can do one of these:

• Comment /ready on the PR
• Add ready label to the PR
• Enable auto-merge.

🚀

[avshalomman]

/ready

[robertgshaw2-redhat]

Thanks for making this change! Did not realize Jamba had FusedMoE during the refactor

[avshalomman]

I see that some tests in the extended suite are failing but they don't seem to be related

[mgoin]

LGTM but is there an active test in CI to ensure this works?

[avshalomman]

LGTM but is there an active test in CI to ensure this works?

There are 10 general tests for the Jamba model, which pass after the change to the FusedMoE layer

[avshalomman]

/ready

[mgoin]

Thanks for the answers, LGTM

[avshalomman]

Thanks for the answers, LGTM

Thanks! How can I re-run the failing tests? These seem like transient, unrelated errors