Skip to content
This repository was archived by the owner on Jan 12, 2026. It is now read-only.

FluxML/NNlibCUDA.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

167 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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/.

About

CUDA integration for the NNlib API

Resources

License

Code of conduct

Stars

Watchers

Forks

Sponsor this project

 

Packages

 
 
 

Contributors

Languages