NNlibCUDA.jl

CUDA integration for the NNlib API
Author FluxML
Popularity
13 Stars
Updated Last
12 Months Ago
Started In
March 2021

NNlibCUDA.jl

This is a glue package which extends functions from NNlib.jl to work with CUDA.jl. It should be loaded automatically when using Flux.jl, but not when using NNlib.jl by itself.

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