-
jberchtold-nvidia authored
[JAX] Fix test_layer to support fused attention and adjust test encoder tolerance to account for minor diff (#2563) Fix failing unit tests Signed-off-by:Jeremy Berchtold <jberchtold@nvidia.com>
404a3ee0
[JAX] Fix test_layer to support fused attention and adjust test encoder tolerance to account for minor diff (#2563)
Fix failing unit tests
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>