I want mlx.optimizers.MultiOptimizer(optimizers, filters= ) as a stateful optimizer wrapper that lets me…

I want mlx.optimizers.MultiOptimizer(optimizers, filters= ) as a stateful optimizer wrapper that lets me…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). I want mlx.optimizers.MultiOptimizer(optimizers, filters= ) as a stateful optimizer wrapper that lets me route different…

The task

I want mlx.optimizers.MultiOptimizer(optimizers, filters=[]) as a stateful optimizer wrapper that lets me route different parameters in one parameter tree to different optimizer instances. The constructor should take a list of optimizer objects and a list of predicate…

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