A lesson in numerical stability

[wouterwln]

Softmaxing apparently helps a lot with numerical stability

[github-actions]

🤖 Code Formatting

Your PR still has some code formatting issues. I've updated PR #485 with the necessary formatting changes.

You can merge that PR into this branch to fix the code style check.

Alternatively, you can run make format locally and push the changes yourself.

[codecov]

Codecov Report

Attention: Patch coverage is 92.06349% with 5 lines in your changes missing coverage. Please review.

Project coverage is 75.50%. Comparing base (f117db4) to head (62682b0).
Report is 19 commits behind head on main.

Files with missing lines	Patch %	Lines
src/rules/discrete_transition/marginals.jl	57.14%	3 Missing ⚠️
...s/discrete_transition/predefined/structured_vmp.jl	86.66%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #484      +/-   ##
==========================================
+ Coverage   75.49%   75.50%   +0.01%     
==========================================
  Files         200      200              
  Lines        5965     5986      +21     
==========================================
+ Hits         4503     4520      +17     
- Misses       1462     1466       +4     

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

🚀 New features to boost your workflow:

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

[bvdmitri]

This is a very nice catch :D