v0.5.4

What's Changed

Breaking changes

• Tweaks to finalize cryojax.ndimage namespace by @michael-0brien in #525
• Change image_model.transform -> image_model.image_transform for clarity by @michael-0brien in #532
• Change image_model.volume_parametrization to image_model.volume for simplicity by @michael-0brien in #533

Deprecations / API updates

• Replace BasicImageConfig coordinate/frequency grid properties with get_coordinate_grid/get_frequency_grid by @michael-0brien in #531
• Simplifications and upgrades in cryojax.dataset: RelionParticleStackDataset -> RelionParticleDataset and remove typed dicts ParticleStackInfo and ParticleParameterInfo by @michael-0brien in #534
• Mark cryojax.jax_util.run_grid_search for removal in favor of the library brutax by @michael-0brien in #542
• Mark cryojax.dataset for removal in favor of the library cryospax

Features

• Option to precompute grids via BasicImageConfig(..., precompute_mode=...) by @michael-0brien in #535
• Move make_linear_operator from cryojax.experimental to cryojax.jax_util by @michael-0brien in #538
• CRYOJAX_ENABLE_CHECKS environmental variable controls equinox.error_if run-time checks by @michael-0brien in #530
• Explicitly determine rotation convention: make_image_model(..., rotation_convention=...) by @michael-0brien in #548
• Image transforms for translation and rotation: cryojax.ndimage.PhaseShiftFFT and cryojax.ndimage.RotateFFT by @DSilva27 in #527

Performance improvements

• Speedups for when CRYOJAX_ENABLE_CHECKS=false: lazy evaluation of boolean arrays by @michael-0brien in #545

Other Changes

• Add benchmarks folder by @michael-0brien in #547 (Thanks @geoffwoollard!)

Full Changelog: v0.5.3...v0.5.4