def METHOD_NAME():
    ma = nk.nn.blocks.MLP(hidden_dims_alpha=(4, 5))
    x = np.zeros((16,))
    pars = ma.init(nk.jax.PRNGKey(), x)
    assert pars["params"]["Dense_0"]["kernel"].shape == (16, 4 * 16)
    assert pars["params"]["Dense_1"]["kernel"].shape == (4 * 16, 5 * 16)