hwosim

hwosim is a JAX-based toolkit for running end-to-end Habitable Worlds Observatory (HWO) mission simulations. It ties together the full suite of HWO direct imaging simulation tools:

• yippy — Coronagraph performance modeling
• orbix — Orbital dynamics and target scheduling
• coronagraphoto — Image simulation for coronagraphic observations
• coronalyze — Post-processing and SNR analysis
• All the other ones I need to still write

Key Features

• Mission Simulation: End-to-end simulation pipelines from target selection through observation and analysis
• Yield Estimation: Detection completeness and exoplanet yield calculations
• Integration Orchestration: Unified interface across the HWO simulation suite
• JAX-Native: Fully JIT-compilable, differentiable, and GPU-accelerated

Installation

With uv (recommended):

uv pip install hwosim

Or with pip:

pip install hwosim

(Note: You may need to install JAX separately to match your specific hardware acceleration requirements.)

Quick Start

import hwosim

Documentation

Full documentation is available at hwosim.readthedocs.io.