Print module - implement module display for remaining modules (part2)

[antimora]

Pull Request Template

Checklist

• Confirmed that run-checks all script has been executed.
• Made sure the book is up to date with changes in this PR.

Related Issues/PRs

#1357

Changes

Added the module displays for the following modules:

1. gelu
2. loss
   • mse
   • huber
   • binary_cross_entropy
   • cross_entropy
3. dropout
4. leaky_relu
5. pool
   • adaptive_avg_pool1d
   • adaptive_avg_pool2d
   • avg_pool1d
   • avg_pool2d
   • max_pool1d
   • max_pool2d
6. norm
   • layer
   • rms
   • instance
   • group
   • batch
7. prelu
8. pos_encoding
9. relu
10. tanh
11. unfold
12. embedding
13. attention
    • mha
14. swiglu
15. conv
    • conv_transpose1d
    • conv_transpose2d
    • conv1d
    • conv2d
16. linear
17. rnn
    • lstm
    • gru
18. transformer
    • pwff
    • encoder
    • decoder
19. rope_encoding

Testing

Added rendering unit tests for each module.

[codecov]

Codecov Report

Attention: Patch coverage is 99.47575% with 4 lines in your changes missing coverage. Please review.

Project coverage is 85.04%. Comparing base (1ae1c03) to head (6045411).

Files	Patch %	Lines
crates/burn-core/src/nn/unfold.rs	92.85%	2 Missing ⚠️
crates/burn-core/src/nn/loss/cross_entropy.rs	96.29%	1 Missing ⚠️
crates/burn-train/src/learner/train_val.rs	0.00%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1933      +/-   ##
==========================================
+ Coverage   84.56%   85.04%   +0.47%     
==========================================
  Files         793      793              
  Lines       93859    94590     +731     
==========================================
+ Hits        79370    80442    +1072     
+ Misses      14489    14148     -341     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[nathanielsimard]

Only one comment, approving in advance! 🙏

[nathanielsimard]

We should put the parameters public.

[antimora]

Fixed. I made other modules' attributes public for consistency since it's been requested frequently.

[laggui]

LGTM! Thanks for also fixing the modules parameters and adding docstrings 🙏