Replacement for PyTorch's nn.Parameter in Flax

[tanaymeh]

Hello everyone!

I was wondering if there is something similar to PyTorch's nn.Parameter class in Flax?

[marcvanzee]

Module parameters in Flax are created using self.param in a Module. See the Module parameters section of our "Getting started" tutorial.

[tanaymeh]

Thanks a lot!