Skip to content

[Fused MoE] Use jax.nn.sigmoid#1980

Open
catswe wants to merge 1 commit intovllm-project:mainfrom
catswe:pallas-sigmoid
Open

[Fused MoE] Use jax.nn.sigmoid#1980
catswe wants to merge 1 commit intovllm-project:mainfrom
catswe:pallas-sigmoid

Conversation

@catswe
Copy link
Copy Markdown
Contributor

@catswe catswe commented Mar 20, 2026

Description

Pallas supports it now after the Jax version upgrade. I need a ready label

Tests

Please describe how you tested this change, and include any instructions and/or
commands to reproduce.

Checklist

Before submitting this PR, please make sure:

  • I have performed a self-review of my code.
  • I have necessary comments in my code, particularly in hard-to-understand areas.
  • I have made or will make corresponding changes to any relevant documentation.

Signed-off-by: catswe <212922539+catswe@users.noreply.github.com>
@kyuyeunk kyuyeunk added the ready ONLY add when PR is ready to merge/full CI is needed label Mar 20, 2026
Copy link
Copy Markdown
Collaborator

@kyuyeunk kyuyeunk left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm. @bythew3i feel free to take a look and review.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants