HF RL Explorer

keras core.ops.softmax default axis is inconsistent with other libraries

keras core.ops.softmax default axis is inconsistent with other libraries: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). I'm porting some code that uses tf.nn.softmax / torch.nn.functional.softmax / scipy.special.softmax over to keras core.ops.softmax , and I'm…

The task

I'm porting some code that uses `tf.nn.softmax` / `torch.nn.functional.softmax` / `scipy.special.softmax` over to `keras_core.ops.softmax`, and I'm getting different numbers when I don't pass an explicit `axis`.

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.