avik-pal/ExplicitFluxLayers.jl

This is stars and forks stats for /avik-pal/ExplicitFluxLayers.jl repository. As of 05 May, 2024 this repository has 350 stars and 30 forks.

The 🔥 Deep Learning Framework Installation ] add Lux Getting Started using Lux, Random, Optimisers, Zygote # using LuxCUDA, LuxAMDGPU # Optional packages for GPU support # Seeding rng = Random.default_rng() Random.seed!(rng, 0) # Construct the layer model = Chain(BatchNorm(128), Dense(128, 256, tanh), BatchNorm(256), Chain(Dense(256, 1, tanh), Dense(1, 10))) # Get the device determined by Lux device = gpu_device() # Parameter and State Variables ps, st = Lux.setup(rng, model) .|>...