Cleaner and Better Documentation

[avik-pal]

TODOs:

• Use @doc doc for docstrings containing LaTeX
• Shift to MKDocs Material theme (it is just prettier)
  • Can we make the functions more prominently highlighted?
  • Code highlighting?
• Restructure Parts of the Documentation
  • Users don't know what to do to integrate Lux with their framework (See DiffEqFlux Layers don't satisfy Lux API SciML/DiffEqFlux.jl#727). Make the AbstractLayer API more prominent
  • Update Style Guide to document deviations from SciMLStyle
• Add docs for packages in lib/
• Turn off draft mode before merging

[codecov]

Codecov Report

Merging #56 (a6e0141) into main (ea5e73d) will increase coverage by 0.05%.
The diff coverage is n/a.

@@            Coverage Diff             @@
##             main      #56      +/-   ##
==========================================
+ Coverage   48.37%   48.42%   +0.05%     
==========================================
  Files          13       13              
  Lines         955      954       -1     
==========================================
  Hits          462      462              
+ Misses        493      492       -1     

Impacted Files	Coverage Δ
src/Lux.jl	100.00% <ø> (ø)
src/core.jl	60.46% <ø> (ø)
src/layers/basic.jl	54.40% <ø> (+0.28%)	⬆️
src/layers/conv.jl	47.11% <ø> (ø)
src/layers/dropout.jl	100.00% <ø> (ø)
src/layers/normalize.jl	33.64% <ø> (ø)
src/layers/recurrent.jl	100.00% <ø> (ø)
src/nnlib.jl	61.68% <ø> (ø)
src/utils.jl	51.80% <ø> (ø)

---

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 ea5e73d...a6e0141. Read the comment docs.

[YichengDWu]

My understanding is that if my custom layer does not have a state then I don't need to implement this function? Since it is calling initialstates(::AbstractRNG, ::Any) = NamedTuple()