@seanmor5
When I try the
Nx.Random.uniform_split(new_key, split: {})
function in page 130 I got an undefined function error
** (UndefinedFunctionError) function Nx.Random.uniform_split/2
# ...
I corrected the code to
Nx.Random.unifrom_split(new_key, 0, 1, shape: {})
And everything worked.