Skip to content

[JAX] Bugfix for softmax primitives accepting invalid input sharding #3161

[JAX] Bugfix for softmax primitives accepting invalid input sharding

[JAX] Bugfix for softmax primitives accepting invalid input sharding #3161