https://github.com/fluxml/nnlibcuda.jl

CUDA integration for the NNlib API
https://github.com/fluxml/nnlibcuda.jl

Last synced: 11 months ago
JSON representation

CUDA integration for the NNlib API

• Host: GitHub
• URL: https://github.com/fluxml/nnlibcuda.jl
• Owner: FluxML
• License: other
• Created: 2021-03-15T08:53:27.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2023-02-05T21:52:08.000Z (over 3 years ago)
• Last Synced: 2025-02-23T16:38:40.381Z (over 1 year ago)
• Language: Julia
• Size: 157 KB
• Stars: 14
• Watchers: 10
• Forks: 13
• Open Issues: 7
• Metadata Files:
  • Readme: README.md
  • License: LICENSE.md

Awesome Lists containing this project

README

          
# NNlibCUDA.jl

This is a glue package which extends functions from [NNlib.jl](https://github.com/FluxML/NNlib.jl) to work with [CUDA.jl](https://github.com/JuliaGPU/CUDA.jl). It should be loaded automatically when using [Flux.jl](https://github.com/FluxML/Flux.jl), but not when using NNlib.jl by itself.

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