# 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.

```bash
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.