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-oss: Agentic RL Environments (MiMo RL release). keras core.ops.softmax default axis is inconsistent with other libraries I'm porting some code that uses tf.nn.softmax / torch.nn.functional.softmax /…
The task
keras_core.ops.softmax default axis is inconsistent with other libraries 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. In…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.