mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
7,020 Commits 504 Branches 263 Tags
Commit Graph

7020 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jake VanderPlas
a641452bc1 DOC: update doc build requirements 2021-02-23 10:31:26 -08:00
jax authors
037cc2b821 Merge pull request #5796 from lucasb-eyer:patch-4 
PiperOrigin-RevId: 358962056
 2021-02-22 20:25:25 -08:00
jax authors
35c41542f8 Merge pull request #5812 from jakevdp:fix-docstrings 
PiperOrigin-RevId: 358957905
 2021-02-22 19:48:23 -08:00
jax authors
9df3454ee0 Merge pull request #5731 from terhorst:master 
PiperOrigin-RevId: 358951861
 2021-02-22 18:53:38 -08:00
jax authors
60bff69ee6 Merge pull request #4323 from tudorcebere:issue2347 
PiperOrigin-RevId: 358938423
 2021-02-22 17:19:05 -08:00
jax authors
1e3b3bbbdb Merge pull request #5809 from jakevdp:pytree-doc-md 
PiperOrigin-RevId: 358935958
 2021-02-22 17:03:24 -08:00
Jake VanderPlas
d8074aca10 DOC: fix update_numpydoc to prevent deleting docstring content 2021-02-22 16:45:40 -08:00
jax authors
7afce974aa Merge pull request #5747 from theosanderson:patch-1 
PiperOrigin-RevId: 358895383
 2021-02-22 13:49:33 -08:00
jax authors
9dacc5d623 Merge pull request #5794 from lucasb-eyer:patch-3 
PiperOrigin-RevId: 358847580
 2021-02-22 10:25:12 -08:00
jax authors
2bf97298be Merge pull request #5765 from jakevdp:cleanup-notebooks 
PiperOrigin-RevId: 358847147
 2021-02-22 10:21:42 -08:00
jax authors
234990e060 Merge pull request #5299 from sunilkpai:feature/bicgstab 
PiperOrigin-RevId: 358845323
 2021-02-22 10:12:02 -08:00
jax authors
25c03f79f2 Merge pull request #5807 from jakevdp:fix-xla-notebook 
PiperOrigin-RevId: 358844761
 2021-02-22 10:11:47 -08:00
jax authors
c92d87ee29 Merge pull request #5808 from jakevdp:nn-doc-link 
PiperOrigin-RevId: 358844692
 2021-02-22 10:11:30 -08:00
Jake VanderPlas
afaec32969 fix conflicts 2021-02-22 10:08:33 -08:00
jax authors
be6950e78b Merge pull request #5718 from jakevdp:github-role 
PiperOrigin-RevId: 358844691
 2021-02-22 10:07:59 -08:00
Jake VanderPlas
c9fd50ccf3 DOC: execute XLA_in_Python notebook 2021-02-22 10:04:42 -08:00
Jake VanderPlas
586b3bb6a9 DOC: regularize notebook metadata 2021-02-22 10:04:42 -08:00
Jake VanderPlas
35983ebba6 Fix broken example in pytrees.rst & migrate to myst-md 2021-02-22 10:03:32 -08:00
jax authors
f32788f161 Merge pull request #5797 from jamesr66a:thinking_typo 
PiperOrigin-RevId: 358840386
 2021-02-22 09:51:22 -08:00
Jake VanderPlas
4b21f3ef7a Add :jax-issue: sphinx role for use in CHANGELOG. 2021-02-22 09:34:24 -08:00
Jake VanderPlas
4c83723d11 DOC: link to RTD version of neural network notebook 2021-02-22 09:19:41 -08:00
Jake VanderPlas
d9af9ee487 Fix stray header in XLA_In_Python notebook 2021-02-22 09:05:48 -08:00
James Reed
c19486c900 Fix typo in thinking_in_jax notebook 2021-02-20 13:24:26 -08:00
Lucas Beyer
5fc1730eec Also print function ID in compile reason logs. 2021-02-20 20:22:52 +01:00
Lucas Beyer
074abba682
Fix error message string interpolation 2021-02-20 18:18:32 +01:00
jax authors
7e040c0a89 Merge pull request #5792 from nfiedel:master 
PiperOrigin-RevId: 358542761
 2021-02-19 22:16:36 -08:00
jax authors
e8ab58ce03 Merge pull request #5792 from nfiedel:master 
PiperOrigin-RevId: 358540730
 2021-02-19 22:00:13 -08:00
Noah Fiedel
bb5ae556ad
Update thinking_in_jax.md 2021-02-19 21:59:47 -08:00
jax authors
42df4b7657 Merge pull request #5793 from yisitu:patch-1 
PiperOrigin-RevId: 358522036
 2021-02-19 18:32:35 -08:00
Situ Yi
77905ef5fd
Update build.py with Cloud TPU support 
Update build.py to have Cloud TPU support by default.
b/180656002
 2021-02-19 12:57:57 -08:00
Noah Fiedel
c1bba43fe0
Merge pull request #1 from nfiedel/nfiedel-jax-patch-1 
Update thinking_in_jax.ipynb
 2021-02-19 12:39:38 -08:00
Noah Fiedel
80a8204fdb
Update thinking_in_jax.ipynb 
Fix typo in notebook url.
 2021-02-19 12:37:04 -08:00
jax authors
15a2f509b8 Merge pull request #5791 from hawkinsp:jax2tf2 
PiperOrigin-RevId: 358459171
 2021-02-19 12:23:50 -08:00
Peter Hawkins
6e48050730 Disable some jax2tf tests that fail on GPU. 
Fix TF/JAX array interoperability test on GPU.
 2021-02-19 15:01:55 -05:00
sunilkpai
d35ae4c9bf removing x64 enable for testing 
removing commented-out x64 line
 2021-02-19 11:17:27 -08:00
jax authors
c5bfdccd64 Merge pull request #5787 from hawkinsp:xla 
PiperOrigin-RevId: 358426001
 2021-02-19 09:55:55 -08:00
Peter Hawkins
b131038094 Update XLA. 2021-02-19 09:55:44 -05:00
sunilkpai
997ad31670 added bicgstab to new jax repo 
fixed some bugs in the bicgstab method and adjusted tolerance for scipy comparison

fixed flake8

added some tests for gradients, fixed symmetry checks, modified lax.cond -> jnp.where

comment out gmres grad check, to be addressed on future PR

increasing tolerance for bicgstab grad test

change to order 1 checks for bicgstab (gmres still fails in order 1) for internal CI check

remove grad checks for now

changing tolerance to pass numpy comparison test
 2021-02-18 18:01:28 -08:00
jax authors
5bbb449ae5 Merge pull request #5777 from google:issue5776 
PiperOrigin-RevId: 358211915
 2021-02-18 10:37:02 -08:00
Matthew Johnson
5a97eab2a0 improve error messages for grad(..., has_aux=True) 
fixes #5776
 2021-02-18 09:46:16 -08:00
Jonathan Terhorst
4c202ad222 implement np.polyint (#70) 2021-02-18 11:08:41 -05:00
jax authors
8ad5118228 Merge pull request #5762 from hawkinsp:padbatch 
PiperOrigin-RevId: 358070219
 2021-02-17 17:32:46 -08:00
jax authors
347481b784 Merge pull request #5768 from jakevdp:fix-take 
PiperOrigin-RevId: 358049345
 2021-02-17 15:42:45 -08:00
Jake VanderPlas
28a19ff9c1 BUG: support negative indices in jnp.take 2021-02-17 15:18:19 -08:00
jax authors
ba269b407f Merge pull request #5748 from jakevdp:doc-fixes 
PiperOrigin-RevId: 358019598
 2021-02-17 13:26:44 -08:00
jax authors
97071f52b1 Merge pull request #5761 from hawkinsp:packbits 
PiperOrigin-RevId: 357987148
 2021-02-17 11:01:22 -08:00
jax authors
ad5f473f7d Merge pull request #5740 from jakevdp:fix-dirichlet 
PiperOrigin-RevId: 357981739
 2021-02-17 10:38:47 -08:00
Peter Hawkins
0dd1b5516d Implement lax.pad batching rule for batched padding values. 2021-02-17 13:30:20 -05:00
Peter Hawkins
67dd1e6de9 Use lax.pad in jnp.packbits. 
jnp.pad() is overkill for this situation.
 2021-02-17 12:54:49 -05:00
Jake VanderPlas
e66ae17e05 BUG: fix jax.scipy.stats.dirichlet implementation & add tests 2021-02-17 09:40:18 -08:00