TF_JAX_tutorials
TF_JAX_tutorials copied to clipboard
AakashKumarNain
All about the fundamental blocks of TF and JAX!
Results
1
TF_JAX_tutorials issues
Sort by
recently updated
recently updated
newest added
Hi, I noticed that you are reusing the same key in e.g. cell 23 of [`main/src/notebooks/jax_tutorials/chapter_5_vmap_pmap.ipynb`](https://github.com/AakashKumarNain/TF_JAX_tutorials/blob/main/src/notebooks/jax_tutorials/chapter_5_vmap_pmap.ipynb) ```python key, subkey = random.split(key) rotate = random.randint(key, shape=[batch_size], minval=0, maxval=2) ``` Looks like...
Metadata
262
Stars
23
Forks
Watchers
Owner
Metadata
All about the fundamental blocks of TF and JAX!