"""Backward compatibility shim — import from shepherd_score.alignment._jax instead."""
from shepherd_score.alignment._jax import *  # noqa: F401, F403
from shepherd_score.alignment._jax import (   # explicit for IDE/type checkers
    optimize_ROCS_overlay_jax,
    optimize_ROCS_esp_overlay_jax,
    optimize_esp_combo_score_overlay_jax,
    optimize_pharm_overlay_jax,
    optimize_pharm_overlay_jax_vectorized,
    convert_to_jnp_array,
    _make_jit_val_grad_pharm_vectorized,
)
