JuliaGPU/CUDA.jl

cuDNN: Provide wrappers for the declarative API

Open

#2.266 aberto em 14 de fev. de 2024

Ver no GitHub
 (3 comments) (0 reactions) (0 assignees)Julia (274 forks)batch import
enhancementhelp wanted

Métricas do repositório

Stars
 (1.408 stars)
Métricas de merge de PR
 (Mesclagem média 5d 5h) (16 fundiu PRs em 30d)

Description

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

Guia do colaborador