mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
8,899 Commits 506 Branches 263 Tags
Commit Graph

18 Commits

Author SHA1 Message Date
Matthew Johnson
24de3e992c small autodidax tweaks 2021-08-05 05:04:08 -07:00
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07:00
Matthew Johnson
fa274f3ace small tweaks, type annotations 2021-07-22 21:09:58 -07:00
George Necula
6a48c60a72 Rename master to main in embedded links. 
Tried to avoid the change on external links to repos that
have not yet renamed master.
 2021-06-18 10:00:01 +03:00
Matthew Johnson
d88acd8b8c autodidax: delete while_loop for now 2021-05-06 12:43:02 -07:00
Matthew Johnson
83cd42271b autodidax: add cond and start while_loop 2021-05-05 12:32:39 -07:00
jax authors
2fea627cbc Merge pull request #6401 from dan-zheng:fix-doc-typos 
PiperOrigin-RevId: 367755029
 2021-04-09 22:10:01 -07:00
Dan Zheng
0b41ff93ea DOC: fix typos. 2021-04-09 23:53:55 -04:00
Matthew Johnson
5f6bce4bfe add 'open in colab' button, add numpy<1.18 compat 
Co-authored-by: Edward Loper <edloper@google.com>
 2021-04-09 20:09:00 -07:00
James Knighton
8772b2c114 Allodidactax... 2021-03-29 04:57:02 -07:00
James Knighton
0d19b7c082 Mirror the minor spelling fixes over the {.ipynb, .md, .py}. 2021-03-27 21:47:07 -07:00
Matthew Johnson
5a055b1dad add WIP disclaimer to autodidax, fix some typos 2021-03-27 19:34:42 -07:00
Matthew Johnson
3595af2ed2 add linearize, vjp, grad. fix bugs. 2021-03-26 21:27:37 -07:00
Matthew Johnson
3457696e80 add linearize code (needs text) 2021-03-12 14:49:13 -08:00
Matthew Johnson
6b5133d71f autodidax: jit, multi-output, pytrees, DeviceArrays 2021-03-05 19:38:52 -08:00
Matthew Johnson
e4f79fb13e add license 2021-02-24 20:57:15 -08:00
Matthew Johnson
9789677e85 reviewer comments, delint 2021-02-24 20:26:48 -08:00
Matthew Johnson
0fcecb2bd2 Add first two sections of Autodidax. 2021-02-24 01:03:52 -08:00