jax

be2a6beb - Raise error when converting dot_general with unsupported precision in jax2tf