search

renecotyfanboy / jaxspec

jaxspec is an X-ray spectra Bayesian analysis package, relying on JAX to enable just in time compilation
https://jaxspec.readthedocs.io/en/latest/
MIT License
29 stars 2 forks source link

Multiproc for the posterior predictive #204

Open renecotyfanboy opened 2 weeks ago

renecotyfanboy commented 2 weeks ago

Posterior predictive steps is longer than actual inference using XIFU data, this should be parallelized using sharded stuff