jax

1aa217b8 - Add a fallback to `get_hlo_text()` to go via old code if AttributeError is hit.