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

Fix #396 by updating the comment in README.md

Browse Source
This commit is contained in:
Bharat Raghunathan 2019-03-02 23:49:35 +05:30 committed by GitHub
parent dddb221a77
commit 0c7aca1b31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -210,7 +210,7 @@ In [4]: x = random.normal(key, (5000, 5000))
In [5]: print(np.dot(x, x.T) / 2) # fast!
[[ 2.52727051e+03 8.15895557e+00 -8.53276134e-01 ..., # ...
In [6]: print(np.dot(x, x.T) / 2) # even faster!
In [6]: print(np.dot(x, x.T) / 2) # even faster! (JIT-compiled code is cached and reused in the second call)
[[ 2.52727051e+03 8.15895557e+00 -8.53276134e-01 ..., # ...
```