Skip to content

[Feature] Integrate fused moe kernel #507

@JamesBrianD

Description

@JamesBrianD

Checklist

Motivation

The tpu-inference released a new fast moe kernel. We should integrate it. See Fused MOE Pallas Kernel.

Related resources

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions