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

766 Commits

Author SHA1 Message Date
Jake VanderPlas
0c91be7b46 CI: temporarily pin numpy to <1.21 2021-06-22 11:15:16 -07:00
Loïc Estève
6192516dfd
Use scikit-learn rather than sklearn in docs/requirements.txt 2021-06-21 12:03:44 +02: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
Jake VanderPlas
b88bf37698 DOC: fix typos in FAQ (np -> jnp) 2021-06-15 09:38:05 -07:00
tlu7
095e6507b9 Support value computation of associated Legendre functions. 
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
 2021-06-14 14:51:37 -07:00
Jake VanderPlas
9be4a3e83f Fix typo/formatting issue in DeviceArray docs 2021-06-07 13:11:37 -07:00
jax authors
69c707639d Merge pull request #6891 from skye:peter_comments 
PiperOrigin-RevId: 377900628
 2021-06-07 06:30:34 -07:00
jax authors
1912bd97f4 Merge pull request #6673 from hypercubestart:master 
PiperOrigin-RevId: 377417011
 2021-06-03 18:29:31 -07:00
Skye Wanderman-Milne
4a85877159 Clean up multi-process guide language. 
Many thanks to hawkinsp@ for these suggestions.
 2021-06-03 14:19:22 -07:00
George Necula
3e9b13b011 Expanded the type promotion documentation with a confusing case 
I filed this as a bug, but I am assuming that it is not easy to fix, so
I also change the documentation.

Bug: 6874
 2021-06-03 09:45:01 +03:00
tlu7
a02bf59233 Adds associated Legendre functions of the first kind. 
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
 2021-06-02 11:37:37 -07:00
Peter Hawkins
46cc654537 Move jax.abstract_arrays to jax._src.abstract_arrays. 
PiperOrigin-RevId: 377044255
 2021-06-02 06:25:22 -07:00
Phúc Lê Khắc
f30a36dbbc Typo. 
Update Common_Gotchas_in_JAX.md
 2021-05-29 17:22:40 +01:00
jax authors
7150a10047 Merge pull request #6835 from jakevdp:sharp-bits 
PiperOrigin-RevId: 375988125
 2021-05-26 11:21:44 -07:00
Jake VanderPlas
d844609c6d DOC: add section to Sharp Bits discussing implicit list conversions 2021-05-26 09:03:42 -07:00
Jake VanderPlas
80a310f8d8 DOC: add note about array views in numpy docs 2021-05-25 16:54:28 -07:00
Jake VanderPlas
b04d0c75a9 sync common gotchas notebook 2021-05-21 12:24:28 -07:00
James Bradbury
8c0aa3880b add note on autodiff of OOB indexing 2021-05-21 12:23:33 -07:00
Jake VanderPlas
b624731203 Fix broken link in custom derivatives notebook 2021-05-21 11:32:21 -07:00
Jake VanderPlas
ff7c2afcaf DOC: directly silence GPU warning 2021-05-21 10:03:50 -07:00
Lukas Geiger
b4b02cb7c3 Add jnp.resize 2021-05-21 09:25:49 +01:00
jax authors
1e9c7e4995 Merge pull request #6796 from skye:multi_process_guide 
PiperOrigin-RevId: 375002136
 2021-05-20 20:07:36 -07:00
Skye Wanderman-Milne
40cc1ec230 Add multi-process guide to docs. 
This is the first section of the internal doc I drafted. I still need
to add the "Common multi-process idioms" and "Multi-host workflows"
sections.
 2021-05-20 18:38:41 -07:00
Lukas Geiger
a5a1f62c5c Improve docs for jnp.round and export jnp.round_ 2021-05-21 01:43:56 +01:00
Paul Nguyen
bbcaec4a3a Initial implementation of jax.numpy.poly 
This is an initial jax.numpy.poly implementation. It is tested by testPoly in the tests/lax_numpy_test.py test file.
 2021-05-20 13:51:14 -05:00
Bogdan Damoc
db39a67cca Update documentation for custom_jvp handling of nondiff_argnums as arguments of _fwd and _bwd rules. 2021-05-12 16:44:52 +01:00
Andrew Liu
8d54a53b53 Update convolutions.md 
Update convolutions.ipynb

Update convolutions.ipynb

Update convolutions.md
 2021-05-11 17:00:09 -07:00
Peter Hawkins
d005e38f78 Promote the x.at[idx].set(y) operators as the preferred way to do indexed updates. 
Mark the index_update() etc. operators as deprecated in the documentation.

Add new .divide and .power operators. Fixes #2694.
Add .multiply as an alias for .mul. To be more numpy-like we should probably prefer the longer names.
 2021-05-10 20:32:00 -04:00
Jake VanderPlas
26f74e64a6 DOC: add documentation of DeviceArray object properties & methods 2021-05-10 12:37:59 -07:00
jax authors
a946824732 Merge pull request #6682 from bdamoc:updated_doc 
PiperOrigin-RevId: 372583046
 2021-05-07 10:32:12 -07:00
Bogdan Damoc
33ca40a9db Update code example in custom VJP documentation. 2021-05-07 16:34:25 +01:00
Parker Schuh
92246017d2
Revert "Use convert_element_type instead of device_put_raw." 2021-05-06 20:19:23 -07:00
jax authors
d0aa875b4d Merge pull request #6400 from pschuh:convert_element 
PiperOrigin-RevId: 372431314
 2021-05-06 14:57:12 -07:00
jax authors
4bbb24d582 Merge pull request #6453 from google:autodidax 
PiperOrigin-RevId: 372422829
 2021-05-06 14:14:19 -07:00
Parker Schuh
9d3e535ad2
Merge branch 'master' into convert_element 2021-05-06 13:18:01 -07:00
Matthew Johnson
d88acd8b8c autodidax: delete while_loop for now 2021-05-06 12:43:02 -07:00
Adam Paszke
ee93ee221c Add a docstring for pjit 
Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>
PiperOrigin-RevId: 372393461
 2021-05-06 12:00:49 -07:00
Matthew Johnson
83cd42271b autodidax: add cond and start while_loop 2021-05-05 12:32:39 -07:00
jax authors
0bf732be15 Merge pull request #6656 from skye:profiling_docs 
PiperOrigin-RevId: 372174753
 2021-05-05 11:49:32 -07:00
Skye Wanderman-Milne
22ae98ac50 Update profiling instructions. 2021-05-05 11:40:22 -07:00
Jake VanderPlas
382506ba1a DOC: add bicgstab to doc page 2021-05-04 12:24:50 -07:00
Matthew Johnson
7ec0b40173 Roll-forward of #6584, which broke internal tests. 
PiperOrigin-RevId: 371839298
 2021-05-03 21:41:23 -07:00
jax authors
ffecef5555 Merge pull request #6611 from njunge94:pip-install-fork 
PiperOrigin-RevId: 371724836
 2021-05-03 10:26:33 -07:00
jax authors
75b00a1235 Copybara import of the project: 
--
3c400a3e588abf9e2259119c50343cba6f3477f1 by Matthew Johnson <mattjj@google.com>:

add 'inline' option to xla_call for jaxpr inlining

--
fe297e39ca37896b75d7943b9b77c0b53fad13ee by Matthew Johnson <mattjj@google.com>:

add 'inline' to jit docstring

--
ff6866c4b3757cde66fe659c2f27d8aeff024e8f by Matthew Johnson <mattjj@google.com>:

new_sublevel in jax2tf

PiperOrigin-RevId: 371542778
 2021-05-01 22:18:39 -07:00
jax authors
e8f209c775 Merge pull request #6584 from google:jit-inline-2 
PiperOrigin-RevId: 371541392
 2021-05-01 21:53:34 -07:00
Ian Horn
74b6762745 Fixed error in autodidax broadasting batching 2021-05-01 19:07:58 -07:00
Nicholas Junge
f559a9bfb4 Fix pip install instruction for forks in contributing section 2021-05-01 19:32:19 +02:00
Nicholas Junge
e985e86808 Implement jnp.r_ and jnp.c_ 2021-05-01 01:05:22 +02:00
Lukas Geiger
a613ce12a3 Export jnp.broadcast_shapes as user facing function 2021-04-30 19:32:51 +02:00
Roy Frostig
b8f9dd6269 unify tree_map and tree_multimap 2021-04-28 19:59:31 -07:00