Warning
Package has been moved to a subdirectory in Lux https://github.com/LuxDL/Lux.jl/tree/main/lib/
Backend for Lux.jl.
This is a developer-facing project and most users should not depend on it directly. As such, we don't have tutorials for this package. Instead, we recommend you check out the Lux tutorials.
What's the distinction from NNlib.jl?
This is currently a place to hold more specialized kernels and layer implementations for Lux.jl. Anyone is free to move these to NNlib.jl (this package is MIT licensed), but I probably don't have the time to do so myself. But incase you do, open an issue here and let me know I will delete the code from this package.
groupnorm with statistics tracking support has been removed.
Support for CUDA has been moved to a weak dependency. If you want to use CUDA, you need
to install and load LuxCUDA as using LuxCUDA or import LuxCUDA.