Lux.jl Documentation

Appearance


Trusted by

Scientific Computing

SciML.ai

Machine Learning

Structure Preserving

GeometricML.jl

Machine Learning

Physical Systems

MeshGraphNets.jl

Graph Neural Nets

LuxCUDA ​

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.

Index ​

API Reference ​

# LuxCUDA.functional — Method.
julia
functional()

Check if LuxCUDA is functional.

source