Skip to content

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

[JAX] Bugfix for softmax primitives accepting invalid input sharding …

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