JuliaGPU/CUDA.jl

cuDNN: Provide wrappers for the declarative API

Open

#2,266 建立於 2024年2月14日

在 GitHub 查看
 (3 留言) (0 反應) (0 負責人)Julia (274 fork)batch import
enhancementhelp wanted

倉庫指標

Star
 (1,408 star)
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

貢獻者指南