Exchange rng in init of RQSCouplingBlocks

src/rqspline_coupling.jl

@@ -112,9 +112,8 @@ Construct and instance of `RQSplineCouplingBlock`, while initializing the parame
 compute device specified in `compute_unit`. (Defaults to CPU)
 """
 function RQSplineCouplingBlock(mask::Vector{Bool}, nn::Chain, compute_unit::AbstractComputeUnit=CPUnit())
-    # Set fixed rng to be consistent across julia versions, alternative: rng = Random.default_rng()
-    rng = TaskLocalRNG()
-    Random.seed!(rng, 0)
+    # Set fixed rng to be consistent across julia versions, alternative: rng = Random.default_rng(); Random.seed!(rng,0)
+    rng = MersenneTwister(1234)
 
     lux_compute_unit = compute_unit isa CPUnit ? cpu_device() : gpu_device()