Add docstrings for kimm.blocks.*

[james77777778]

Related to #44

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 94.85%. Comparing base (ac667c4) to head (8c60cce).

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #50      +/-   ##
==========================================
+ Coverage   94.84%   94.85%   +0.01%     
==========================================
  Files          39       39              
  Lines        4343     4353      +10     
==========================================
+ Hits         4119     4129      +10     
  Misses        224      224              

Flag	Coverage Δ
kimm	94.85% <100.00%> (+0.01%)	⬆️
kimm-jax	94.21% <100.00%> (+0.01%)	⬆️
kimm-numpy	64.32% <57.89%> (-0.06%)	⬇️
kimm-tensorflow	94.48% <100.00%> (+0.01%)	⬆️
kimm-torch	94.21% <100.00%> (+0.01%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

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