JuliaGPU/CUDA.jl

cuDNN: Provide wrappers for the declarative API

已关闭

#2,266 创建于 2024年2月14日

 (3 条评论) (0 个反应) (0 位负责人)Julia (274 个派生)batch import
enhancementhelp wanted

仓库指标

星标
 (1,408 个星标)
PR 合并指标
 (平均合并 5天 5小时) (30 天内合并 16 个 PR)

描述

As part of cuDNN 9.0, the old imperative API has been deprecated, while moving towards a new, graph-based declarative API. That means that our wrappers will have to be rewritten. As I'm not familiar with machine learning / NNlib.jl (which I guess these wrappers would need to be compatible with), I'm hoping that people from the ML community could be able to help with this.

cc @ToucheSir @CarloLucibello @mcabbott @DhairyaLGandhi

贡献者指南