João Sousa Pinto

@GiulioRomualdi That makes sense. I actually only had issues with the bound vectors; I somehow assumed that the same issue would exist with the matrices, so I made sure they...

@a-kenji That didn't work for me. In general, this repo would benefit from having a `rust-toolchain.toml ` file (https://rust-lang.github.io/rustup/overrides.html#the-toolchain-file).

I just did some moments ago.

It looks like `tan` only became part of the VHLO dialect on v1.4.0 ([source](https://github.com/openxla/stablehlo/blob/main/stablehlo/dialect/VhloDialect.td)). This seems to be causing a unit test failure, due to an export failure. I might...

> One difficulty with this feature request is that after a function is compiled into HLO, information about the original python function boundaries is lost. There are several ways of...

For example, this can be used (although it might be not the most compact representation): ``` with open("output.hlo", "w") as f: ir.operation.print( enable_debug_info=True, pretty_debug_info=True, use_local_scope=True, file=f, ) ```

Related discussion: https://github.com/jax-ml/jax/issues/23251

@jakevdp @gnecula @justinjfu Any thoughts?

Sure, I'll give it a go.

Created a pull request: https://github.com/google/jax/pull/23261/files Feel free to review @hawkinsp