Obtaining the name, values pairs in Equinox

[sachith-gunasekara]

Pytorch has this default function that yields the parameter name-value pairs in a module including all its submodules called torch.nn.Module.named_parameters().

This function can be used in a loop as follows: for pn, p in self.named_parameters()

Is it possible to achieve something like this in JAX/Equinox?