Formatting updates and relax parameter type

[avik-pal]

TODO:

• Fixes Lux does not handle sensitivity algorithms with ReverseDiff SciML/SciMLSensitivity.jl#609
• Add a reversediff.jl test
• Pull Documentation Updates from Add Lux.@layerdef for easy custom model creation #46 [WIP] Allow named layers and some Misc Fixes #44

[codecov]

Codecov Report

Merging #48 (195041b) into main (95d27d0) will increase coverage by 3.69%.
The diff coverage is 39.34%.

@@            Coverage Diff             @@
##             main      #48      +/-   ##
==========================================
+ Coverage   45.03%   48.73%   +3.69%     
==========================================
  Files          13       13              
  Lines         946      946              
==========================================
+ Hits          426      461      +35     
+ Misses        520      485      -35     

Impacted Files	Coverage Δ
src/adapt.jl	0.00% <0.00%> (ø)
src/autodiff.jl	46.66% <0.00%> (+4.44%)	⬆️
src/layers/display.jl	0.00% <0.00%> (ø)
src/layers/dropout.jl	100.00% <ø> (ø)
src/layers/normalize.jl	34.28% <0.00%> (ø)
src/layers/recurrent.jl	100.00% <ø> (ø)
src/transform.jl	0.00% <ø> (ø)
src/utils.jl	51.80% <20.00%> (+4.81%)	⬆️
src/nnlib.jl	61.68% <25.00%> (ø)
src/layers/basic.jl	53.88% <48.48%> (+15.02%)	⬆️
... and 5 more

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 95d27d0...195041b. Read the comment docs.