Update mlp.py

added rng_uniform