How to initialize Optax Optimizer for two neural networks.

[raj-brown]

Dear All- I have a very simple question. I have two neural networks of type MLP and I want to initialize optimizer via optax. When I have one neural network I do like pyt1 = eqx.filter(bn, eqx.is_array) opt_state = optimizer.init(pyt) Is there anyway I can concatenate parameters from both neural networks.

Thank you very much! Raj