Run float32 jobs separately

[michaelosthege]

This change will save a lot of CI resources.

For tests that should run with either floatX setting, aesara.change_flags can be used as seen in test_types.py.

Initially I wanted to parametrize the float32 job across Ubuntu & Windows, but the different shells require different syntax in the YAML. The best way to unify that would probably to outsource the pytest commands and test-subset into a non-YAML file, but I don't have time for that right now.

CI time reduced to ~3h45m 🎉

Closes #5626

[codecov]

Codecov Report

Merging #5630 (5677646) into main (b895e40) will decrease coverage by 1.37%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main    #5630      +/-   ##
==========================================
- Coverage   87.63%   86.26%   -1.38%     
==========================================
  Files          76       76              
  Lines       13717    13723       +6     
==========================================
- Hits        12021    11838     -183     
- Misses       1696     1885     +189     

Impacted Files	Coverage Δ
pymc/distributions/simulator.py	87.58% <100.00%> (+0.08%)	⬆️
pymc/sampling_jax.py	0.00% <0.00%> (-96.97%)	⬇️
pymc/distributions/logprob.py	96.33% <0.00%> (+0.17%)	⬆️
pymc/parallel_sampling.py	87.70% <0.00%> (+0.99%)	⬆️

[ricardoV94]

I pushed a change that should fix the SMC tests

[michaelosthege]

The coverage check for 6af5104 was good and in the last commit 5677646 I don't see how it could affect sampling_jax.py coverage.

Let's ship it :)