Pseudo-Random Number Generation
Explore how pseudo-random number generation works in JAX, focusing on stateless PRNGs like Threefry. Understand the advantages of JAX's approach including parallelization, smaller state vectors, and improved reproducibility for deep learning and numerical computing tasks.
Random variable
Consider a function:
No matter how many times we call this function, it will always give the same output for a given input.
On the other hand, consider some random/stochastic ...