Lux
Accelerator Support
Building Blocks
Domain Specific Modeling
Testing Functionality
Appearance
Scientific Computing
SciML.ai
Machine Learning
Structure Preserving
GeometricML.jl
Physical Systems
MeshGraphNets.jl
Graph Neural Nets
LuxCUDA is meant to be used as a trigger package for all CUDA dependencies in Lux. Users requiring CUDA support should install LuxCUDA and load it alongside Lux.
LuxCUDA
CUDA
LuxCUDA.functional
functional()
Check if LuxCUDA is functional.
source