Challenge: PRNG and Probability Distribution
Explore pseudo-random number generation and probability distributions in JAX by creating PRNG keys and implementing batch normalization. Understand how to transform neural network outputs into a standard normal distribution for improved modeling.
We'll cover the following...
We'll cover the following...
Exercise 1: PRNG
The key component of developing any random variables setup is having a good grip on PRNG. Begin by creating a PRNG with a seed of, say, 100, and create a subkey ...