Error with control flow.scan and plated uniform random variables
Error with control flow.scan and plated uniform random variables: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Hi -- I'm facing this weird issue where I'm unable to run a model with control flow.scan and uniformly distributed random variables in it. The following is a…
The task
Hi -- I'm facing this weird issue where I'm unable to run a model with control_flow.scan and uniformly distributed random variables in it. The following is a simplified reproducer of what I'm trying to do Attempting to sample from the above model raises an error However, if I…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.