mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove author info in notebook (old, redundant)

Browse Source
This commit is contained in:
Matthew Johnson 2019-08-14 07:13:42 -07:00 committed by GitHub
parent 4559d36d17
commit 2dec779b49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
notebooks/XLA_in_Python.ipynb
View File

@ -37,10 +37,6 @@
"\n",
"_Anselm Levskaya_ \n",
"\n",
"_The Python XLA client was designed by Roy Frostig._\n",
"\n",
"_JAX was written by Dougal Maclaurin, Peter Hawkins, Matthew Johnson, Roy Frostig, Alex Wiltschko, Chris Leary._ \n",
"\n",
"XLA is the compiler that JAX uses, and the compiler that TF uses for TPUs and will soon use for all devices, so it's worth some study. However, it's not exactly easy to play with XLA computations directly using the raw C++ interface. JAX exposes the underlying XLA computation builder API through a python wrapper, and makes interacting with the XLA compute model accessible for messing around and prototyping.\n",
"\n",
"XLA computations are built as computation graphs in HLO IR, which is then lowered to LLO that is device specific (CPU, GPU, TPU, etc.). \n",