Allow compile_kwargs in sample_smc

[jessegrabowski]

Description

Allow compile_kwargs to be included among kernel_kwargs, so users can compile logp functions to numba/jax/torch. e.g. pm.sample_smc(..., compile_kwargs={'mode':'NUMBA'})

Necessary because the multiprocessors don't respect pytensor.config.change_flags context. But also we allow this everywhere else, so why not here too?

Related Issue

• Closes #
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc--7702.org.readthedocs.build/en/7702/

[codecov]

Codecov Report

Attention: Patch coverage is 90.00000% with 1 line in your changes missing coverage. Please review.

Project coverage is 92.66%. Comparing base (d1aff0b) to head (5f630f5).
Report is 5 commits behind head on main.

Files with missing lines	Patch %	Lines
pymc/smc/kernels.py	85.71%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #7702      +/-   ##
==========================================
- Coverage   92.67%   92.66%   -0.01%     
==========================================
  Files         107      107              
  Lines       18319    18325       +6     
==========================================
+ Hits        16977    16981       +4     
- Misses       1342     1344       +2     

Files with missing lines	Coverage Δ
pymc/smc/sampling.py	99.30% <100.00%> (+0.01%)	⬆️
pymc/smc/kernels.py	97.82% <85.71%> (-0.35%)	⬇️

... and 1 file with indirect coverage changes

[jessegrabowski]

Here's a gist with timings, to show this is doing the right thing:

https://gist.github.com/jessegrabowski/2ec8b44c3d21b4304e059e49d82b13c7