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

貢獻者指南