HF RL Explorer

Feature request: keras.ops.unravel index

Feature request: keras.ops.unravel index: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Feature request: keras.ops.unravel index I'm writing backend-agnostic code on top of keras.ops so the same script runs on JAX, TensorFlow and PyTorch backends. A pretty common pattern I…

The task

Feature request: keras.ops.unravel_index I'm writing backend-agnostic code on top of keras.ops so the same script runs on JAX, TensorFlow and PyTorch backends. A pretty common pattern I have is: flatten a tensor, find the index of the max (or top-k) with something like…

Part of XiaomiMiMo/MiMo-V2.6-RL-oss.