├── .gitignore ├── LICENSE ├── README.md ├── src └── notebooks │ ├── jax_tutorials │ ├── chapter_1_devicearray.ipynb │ ├── chapter_2_pure_functions.ipynb │ ├── chapter_3_prng.ipynb │ ├── chapter_4_just_in_time_compilation.ipynb │ ├── chapter_5_vmap_pmap.ipynb │ ├── chapter_6_autodiff.ipynb │ └── chapter_7_pytrees.ipynb │ └── tensorflow_tutorials │ ├── chapter_1_tensors.ipynb │ ├── chapter_2_variables.ipynb │ └── chapter_3_AD_and_grads.ipynb └── static └── combined_logo.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/README.md -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_1_devicearray.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_1_devicearray.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_2_pure_functions.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_2_pure_functions.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_3_prng.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_3_prng.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_4_just_in_time_compilation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_4_just_in_time_compilation.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_5_vmap_pmap.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_5_vmap_pmap.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_6_autodiff.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_6_autodiff.ipynb -------------------------------------------------------------------------------- /src/notebooks/jax_tutorials/chapter_7_pytrees.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/jax_tutorials/chapter_7_pytrees.ipynb -------------------------------------------------------------------------------- /src/notebooks/tensorflow_tutorials/chapter_1_tensors.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/tensorflow_tutorials/chapter_1_tensors.ipynb -------------------------------------------------------------------------------- /src/notebooks/tensorflow_tutorials/chapter_2_variables.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/tensorflow_tutorials/chapter_2_variables.ipynb -------------------------------------------------------------------------------- /src/notebooks/tensorflow_tutorials/chapter_3_AD_and_grads.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/src/notebooks/tensorflow_tutorials/chapter_3_AD_and_grads.ipynb -------------------------------------------------------------------------------- /static/combined_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AakashKumarNain/TF_JAX_tutorials/HEAD/static/combined_logo.png --------------------------------------------------------------------------------