Add model.logp_elemwiset

[ricardoV94]

Closes #5236
Closes #5230

[codecov]

Codecov Report

Merging #5245 (046fc7a) into main (7191e61) will increase coverage by 0.14%.
The diff coverage is 87.50%.

@@            Coverage Diff             @@
##             main    #5245      +/-   ##
==========================================
+ Coverage   78.98%   79.13%   +0.14%     
==========================================
  Files          88       88              
  Lines       14231    14337     +106     
==========================================
+ Hits        11240    11345     +105     
- Misses       2991     2992       +1     

Impacted Files	Coverage Δ
pymc/sampling_jax.py	0.00% <0.00%> (ø)
pymc/model.py	83.26% <85.41%> (-0.05%)	⬇️
pymc/distributions/logprob.py	96.49% <92.10%> (+1.44%)	⬆️
pymc/backends/arviz.py	89.51% <100.00%> (-0.05%)	⬇️
pymc/backends/report.py	89.51% <0.00%> (-2.10%)	⬇️
pymc/distributions/dist_math.py	86.78% <0.00%> (-0.92%)	⬇️
pymc/initial_point.py	100.00% <0.00%> (ø)
pymc/distributions/bound.py	100.00% <0.00%> (ø)
pymc/distributions/discrete.py	98.46% <0.00%> (+0.10%)	⬆️
... and 3 more

[OriolAbril]

I only checked the arviz and test parts which look good, I can't really review the other files