Usage Notes

CPU vs GPU

The package can run on CPU or GPU via JAX. For reproducible quick checks and CI runs, CPU mode is recommended.

export JAX_PLATFORM_NAME=cpu

Input schema for PDet_O4.pdet

The expected physical parameters are:

  • m1_source

  • m2_source

  • redshift

  • a1

  • a2

  • right_ascension

  • sin_declination

  • polarization

  • cos_inclination

  • cos_tilt1

  • cos_tilt2

  • phi1

  • phi2

If columns are missing, helper logic may fill them from the injection model, but passing all columns explicitly is recommended for clarity, control and prediction quality.