NNlibCUDA
CUDA compatibility for NNlib.jl.
Julia gpu kernels are in src/, while wrappers around cudnn are in src/cudnn/.
- Popularity
- 5 Stars
- Updated Last
- 2 Years Ago
- Started In
- March 2021
Required Packages
-
AbstractFFTs
-
Adapt
-
BFloat16s
-
CEnum
-
ChainRulesCore
-
Compat
-
CUDA
-
DataStructures
-
DocStringExtensions
-
ExprTools
-
GPUArrays
-
GPUCompiler
-
JLLWrappers
-
LLVM
-
LogExpFunctions
-
MacroTools
-
Memoize
-
NNlib
-
OpenSpecFun_jll
-
OrderedCollections
-
Preferences
-
Random123
-
RandomNumbers
-
Reexport
-
Requires
-
Scratch
-
SpecialFunctions
-
TimerOutputs
Used By Packages
-
Adversarial
-
AdversarialPrediction
-
AlphaZero
-
Ant
-
ARMANets
-
AtomicGraphNets
-
AugmentedGaussianProcesses
-
AutoLandmarking
-
ChemistryFeaturization
-
DeepForest
-
DeepQLearning
-
DiffEqFlux
-
DistributedFlux
-
DistributedReinforcementLearning
-
FastRecurFlux
-
FeedbackNets
-
Flux
-
Flux3D
-
FluxExtra
-
FluxJS
-
FluxTikz
-
FluxTraining
-
FluxUtils
-
GalacticOptim
-
GenerativeModels
-
GenFlux
-
GeometricFlux
-
GlobalApproximationValueIteration
-
GPFlux
-
GreenFlux
-
JsonGrinder
-
Kinetic
-
KitML
-
LighthouseFlux
-
Metalhead
-
Mill
-
MLJFlux
-
NaiveGAflux
-
NaiveNASflux
-
NeuralArithmetic
-
NeuralNetDiffEq
-
NeuralPDE
View all packages