Skip to content

[JAX] Fix test_layer to support fused attention and adjust test encoder tolerance to account for minor diff #7951

[JAX] Fix test_layer to support fused attention and adjust test encoder tolerance to account for minor diff

[JAX] Fix test_layer to support fused attention and adjust test encoder tolerance to account for minor diff #7951