JuliaGPU/CUDA.jl

cuDNN: Provide wrappers for the declarative API

クローズ

#2,266 opened on 2024/02/14

 (3 件のコメント) (0 件のリアクション) (0 人の担当者)Julia (274 件のフォーク)batch import
enhancementhelp wanted

Repository metrics

Stars
 (1,408 個のスター)
PR merge metrics
 (平均マージ 5d 5h) (30d で 16 merged PRs)

説明

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

コントリビューターガイド