key, subkey = random.split(key)
print(random.normal(subkey, shape=(3,)))  # New values

        __
        __