mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1423 from gnecula/colabs

Browse Source 
Added How_JAX_primitives_work colab
This commit is contained in:
George Necula 2019-10-07 14:59:37 +02:00 committed by GitHub
commit a5063ee68b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1660 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -10,3 +10,6 @@ build/
dist/
.mypy_cache/
docs/notebooks/.ipynb_checkpoints/
docs/_autosummary
.idea

6
docs/index.rst
View File

@ -13,7 +13,13 @@ For an introduction to JAX, start at the
notebooks/quickstart
notebooks/autodiff_cookbook
.. toctree::
:maxdepth: 1
:caption: Advanced Tutorials
notebooks/Common_Gotchas_in_JAX
notebooks/How_JAX_primitives_work
.. toctree::
:maxdepth: 1

1651
docs/notebooks/How_JAX_primitives_work.ipynb Normal file
View File

File diff suppressed because it is too large Load Diff