def __call__(self, x: Array, **kwargs) -> jnp.ndarray:
        """
        Forward pass.

        Parameters
        ----------
        x: Array
            Inputs.

        jnp.ndarray
            Model outputs.
        """
        x = self.dfe_subnet(x)
        x = self.output_subnet(x)
        return x

        Forward pass.

        Parameters
        ----------
        x: Array
            Inputs.

        jnp.ndarray
            Model outputs.
        
fortuna/model/lenet.py