refactor: move @public declaration for LuxCore imports to end of file

Co-authored-by: avik-pal <[email protected]>

Author: Copilot

src/Lux.jl

@@ -45,21 +45,6 @@ import LuxCore:
     preserves_state_type
 
 @reexport using LuxCore, LuxLib, MLDataDevices, WeightInitializers
-
-@public (
-    initialparameters,
-    initialstates,
-    parameterlength,
-    statelength,
-    outputsize,
-    update_state,
-    trainmode,
-    testmode,
-    setup,
-    apply,
-    replicate,
-    preserves_state_type,
-)
 using NNlib:
     NNlib,
     DenseConvDims,
@@ -208,6 +193,20 @@ export LuxOps
 @public Experimental
 @public set_dispatch_doctor_preferences!
 @public Serialization
+@public (
+    initialparameters,
+    initialstates,
+    parameterlength,
+    statelength,
+    outputsize,
+    update_state,
+    trainmode,
+    testmode,
+    setup,
+    apply,
+    replicate,
+    preserves_state_type,
+)
 
 # NNlib.jl reexports
 ## Functional API for common layers. Recommended to use the LuxLib versions