tl.mofa: add train_kwargs argument

[ilia-kats]

This allows more fine-grained control of MOFA training.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 26.80%. Comparing base (907302a) to head (a533638).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #152      +/-   ##
==========================================
+ Coverage   26.73%   26.80%   +0.06%     
==========================================
  Files          16       16              
  Lines        2289     2291       +2     
==========================================
+ Hits          612      614       +2     
  Misses       1677     1677              

Files with missing lines	Coverage Δ
src/muon/tl.py	41.15% <100.00%> (+0.24%)	⬆️

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.