Hacktoberfest 2026:维护者为十月标记出来的 issue,仍然开放、适合新手。 浏览 Hacktoberfest issue

[JAX] score_mod attention to support "enforce_precompiled" flag in cuDNN graph deserialization

未关闭
#3,046 0 条评论 0 个 reaction 已指派 1 人 在 GitHub 查看

维护者通常 2 天内回复

@vcherepanov-nv 已经在做这个了。

开始于 2026年5月26日。

评估

这个 Issue 还没有评估数据。

描述

attention jax

Once https://github.com/NVIDIA/cudnn-frontend/issues/254 is completed, we should support this in TE's cuDNN usage with score_mod to give clearer error messages instead of potentially incorrect values or less clear errors in cases where the cuDNN Python FE and cuDNN C++ FE requires the cuDNN C++ FE to recompile the kernels.

主要语言
Python
星标
3.6k
派生
851
平均合并
4 天 15 小时
30 天内合并 PR
51

环境准备

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

NVIDIA/TransformerEngine 的其他 Issue

查看 NVIDIA/TransformerEngine 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。