NNlibCUDA.jl

CUDA integration for the NNlib API
Author FluxML
Popularity
5 Stars
Updated Last
1 Year Ago
Started In
March 2021

NNlibCUDA

CUDA compatibility for NNlib.jl.

Julia gpu kernels are in src/, while wrappers around cudnn are in src/cudnn/.