jax

c9f9f28b - [shape_poly] Fix handling of dot_general with different lhs_dtype and rhs_dtype

[shape_poly] Fix handling of dot_general with different lhs_dtype and rhs_dtype Add primitives tests for the case of dot_general with different lhs_dtype and rhs_dtype. Then fix the lowering to work with dynamic shapes.