flatten_parameters ================== Auto-generated from flax_nnx_code_defs .. raw:: html
torch.nn.modules.rnn.RNNBase.flatten_parameters
mlx.core.flatten
nnx.graph.SplitContext.flatten