mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 23:56:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
11,532 Commits 508 Branches 263 Tags
Commit Graph

981 Commits

Author SHA1 Message Date
Jake VanderPlas
681240244c DOC: explicitly enable dollarmath extension 2022-05-12 10:37:34 -07:00
jax authors
d092d6305f Merge pull request #10668 from sharadmv:custom-interpreter-update 
PiperOrigin-RevId: 448058058
 2022-05-11 12:31:13 -07:00
jax authors
43467f9873 Merge pull request #10657 from sharadmv:effects-design-note 
PiperOrigin-RevId: 448051701
 2022-05-11 12:03:05 -07:00
Sharad Vikram
2bce098f15 Add sequencing effects design note 2022-05-11 11:19:58 -07:00
Sharad Vikram
aca9dc6c6f Update custom interpreter tutorial 2022-05-11 11:11:47 -07:00
Onno Eberhard
7276731e8c
Small fix in FAQ documentation 2022-05-10 13:04:49 +02:00
Jake VanderPlas
29efc30c63 DOC: update myst-nb to v0.15.0 2022-05-09 14:32:48 -07:00
jax authors
95b8e49e27 Merge pull request #10636 from jakevdp:jupytext-doc 
PiperOrigin-RevId: 447515735
 2022-05-09 11:07:25 -07:00
Jake VanderPlas
9679ca826a DOC: clarify jupytext instructions 2022-05-09 09:53:33 -07:00
ngam
7132fbaca1 minor edits to clarify pjit docs 2022-05-09 16:14:45 +00:00
Jake VanderPlas
c4836aa507 DOC: add FAQ entry on jit-compiling methods 2022-05-06 12:39:12 -07:00
Jake VanderPlas
dd71e2920b DOC: remove mentions of units 2022-05-04 16:08:44 -07:00
jax authors
ceb3222bab Merge pull request #10568 from jakevdp:sparse-doc 
PiperOrigin-RevId: 446573409
 2022-05-04 16:03:30 -07:00
Jake VanderPlas
9e371c7fcc [sparse] incremental improvement to docs 2022-05-04 12:25:16 -07:00
Alex Riley
372371cec6 Add jax.scipy.linalg.funm 2022-05-02 21:46:41 +01:00
Jake VanderPlas
1134ff5871 CI: update jupytext to 0.13.8 2022-05-02 11:22:31 -07:00
jax authors
fb4731d40e Merge pull request #10502 from jakevdp:fix-myst 
PiperOrigin-RevId: 445530774
 2022-04-29 16:26:51 -07:00
Jake VanderPlas
472f72a83f docs: pin myst-nb to 0.13.2 2022-04-29 16:01:41 -07:00
jax authors
227e525de2 Merge pull request #10458 from carlosgmartin:random_orthogonal_unitary 
PiperOrigin-RevId: 445522278
 2022-04-29 15:40:16 -07:00
Kehang Han
4717a6b6d2 updates maps.mesh() context manager in jax_101/pjit documentation 
PiperOrigin-RevId: 445501580
 2022-04-29 14:06:11 -07:00
Carlos Martin
b276c31b75 Added random.orthogonal. 2022-04-29 14:20:50 -04:00
Jake VanderPlas
47ca71d502 DOC: link to install instructions in HTML docs 2022-04-22 12:35:31 -07:00
Peter Hawkins
1af5c888bd Remove CpuDevice from documentation index. 
The corresponding class no longer exists.
 2022-04-16 10:03:47 -04:00
jax authors
806ad7349e Merge pull request #10314 from google:Mesh 
PiperOrigin-RevId: 442098232
 2022-04-15 14:42:18 -07:00
yashkatariya
9719cc89d3 Add a docstring for maps.Mesh 2022-04-15 13:46:59 -07:00
Jake VanderPlas
c5728cb5d8 DOC: add section on input validation in custom pytrees 2022-04-13 13:50:05 -07:00
jax authors
191c83816c Merge pull request #10226 from ljjsalt:add-polydiv 
PiperOrigin-RevId: 441548874
 2022-04-13 12:27:22 -07:00
Jiajie Li
128e51c638 Add polydiv to jax.numpy 
Fix code style, fix tests

Add warning when use polydiv with trim_leading_zeros

Update warning for polydiv

Co-authored-by: Jake Vanderplas <jakevdp@gmail.com>

Enable type check in _CompileAndCheck

Fix cutoff

Fix cut-off in polydiv

Add trim_zeros_tol, remove redundant code in polydiv

Remove unused import

Fix trim_zero_tol usage in polydiv
 2022-04-13 18:31:27 +00:00
Peter Hawkins
af733ee0c0 Remove mention of six package from jaxlib build instructions. 
I verified that six is no longer needed.
 2022-04-13 14:09:01 -04:00
Alex Riley
869596fc2c Add jax.scipy.linalg.rsf2csf 2022-04-06 21:06:23 +01:00
jax authors
cdf4177f92 Merge pull request #10126 from jakevdp:tree-multimap 
PiperOrigin-RevId: 438956536
 2022-04-01 18:52:50 -07:00
Jake VanderPlas
c61a18b346 DOC: switch from tree_multimap to tree_map in docs 2022-04-01 14:52:16 -07:00
Fabian Pedregosa
4fd466be1f
remove $ from command line commands 
A few commands in this file were prefixed with $ which results in an invalid command when copied with the sphinx "copy" button.
 2022-04-01 15:50:29 -04:00
jax authors
e766b96063 Merge pull request #10058 from yotarok:istft 
PiperOrigin-RevId: 438832534
 2022-04-01 08:43:27 -07:00
Yotaro Kubo
a7fd751acf Add istft to jax.scipy.signal. 2022-04-01 14:28:53 +09:00
Jake VanderPlas
72f14a5d64 Sharp bits: add miscellaneous divergences from numpy 2022-03-31 12:15:06 -07:00
jax authors
9da5f4e793 Merge pull request #10074 from jakevdp:jit-partial-doc 
PiperOrigin-RevId: 438193490
 2022-03-29 20:02:17 -07:00
Jake VanderPlas
f4b64f48f4 doc: add examples of using partial with jit 2022-03-29 15:43:58 -07:00
jax authors
b31cf89e48 Merge pull request #10072 from jakevdp:fromiter 
PiperOrigin-RevId: 438141629
 2022-03-29 15:28:36 -07:00
jax authors
23c783af88 Merge pull request #10068 from hawkinsp:docs 
PiperOrigin-RevId: 438131830
 2022-03-29 14:49:47 -07:00
Jake VanderPlas
fbfc3d8edf Better error messages for jnp.fromiter and jnp.fromfile 2022-03-29 14:30:32 -07:00
Jake VanderPlas
093b7032a8 Implement jnp.from* array creation functions 2022-03-29 10:52:47 -07:00
Peter Hawkins
150235ca5b Add note about enabling developer mode on Windows. 
Issue 4843.
 2022-03-29 12:32:42 -04:00
Roy Frostig
e0ddd43933 add custom_vjp upgrade flag to jax module and fix doc rendering for upgrade flags 2022-03-28 20:36:35 -07:00
Hyeontaek Lim
87671f4d85 Add a documentation for the JAX transfer guard 2022-03-25 19:50:12 +00:00
jax authors
f57e78e240 Merge pull request #9766 from ayaka14732:patch-1 
PiperOrigin-RevId: 437297267
 2022-03-25 11:39:41 -07:00
jax authors
440cd7b9df Merge pull request #9906 from jakevdp:gamma-log-space 
PiperOrigin-RevId: 436255847
 2022-03-21 11:11:50 -07:00
Jake VanderPlas
32c718db11 quickstart: fix outdated TPU reference 2022-03-21 09:41:10 -07:00
Jake VanderPlas
69969ef803 add random.loggamma and improve dirichlet & beta implementation 2022-03-21 08:33:11 -07:00
Matthew Johnson
43036e1316 improvements to autodidax reduce_sum 
* generalize reduce_sum to handle multiple axes
* add reduce_sum transpose rule

also fix bug in AD jaxpr formation related to deduplicating consts
 2022-03-15 20:22:55 -07:00