Skip to content

jax.nn.softmax: fix fill value when where is specified#16209

Merged
copybara-service[bot] merged 1 commit intojax-ml:mainfrom jakevdp:softmax-whereJun 1, 2023
+16-8

Commits

Commits on Jun 1, 2023