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

364 Commits

Author SHA1 Message Date
Jake VanderPlas
aa1452375b Register beta args deprecation 
PiperOrigin-RevId: 642427224
 2024-06-11 16:19:14 -07:00
jax authors
5cf52b8215 Merge pull request #21725 from rajasekharporeddy:testbranch3 
PiperOrigin-RevId: 642332950
 2024-06-11 11:25:23 -07:00
jax authors
8199267a4f Merge pull request #21762 from rajasekharporeddy:testbranch2 
PiperOrigin-RevId: 642310572
 2024-06-11 10:19:00 -07:00
jax authors
27140fe6de Merge pull request #21772 from jakevdp:beta-dep 
PiperOrigin-RevId: 642275316
 2024-06-11 08:15:58 -07:00
rajasekharporeddy
2aa2a398f1 Update code examples 2024-06-11 17:47:36 +05:30
Jake VanderPlas
afe088f876 Simplify definition of jax.scipy.special.kl_div 2024-06-10 11:36:35 -07:00
Jake VanderPlas
990b475b77 jax.scipy.special.beta: deprecate x,y in favor of a,b 2024-06-10 09:01:39 -07:00
jax authors
af95803d00 Merge pull request #21759 from rajasekharporeddy:testbranch1 
PiperOrigin-RevId: 641831969
 2024-06-10 02:29:12 -07:00
rajasekharporeddy
479145b1a6 Add example code snippets to jax.scipy.signal.convolve2d, correlate and correlate2d docs 2024-06-10 14:27:41 +05:30
rajasekharporeddy
775c6f8727 Fix Typos in docs and one error message 2024-06-10 11:38:01 +05:30
sdupourque
751d59ce67 increase default precision for hyp1f1 2024-06-08 00:38:51 +02:00
rajasekharporeddy
7989c70572 Add example code snippets to jax.scipy.linalg.expm and jax.scipy.linalg.polar docs 2024-06-08 03:30:12 +05:30
rajasekharporeddy
aee310b54d Fix some doc typos 2024-06-06 15:19:53 +05:30
jax authors
bf59a67bf0 Merge pull request #21657 from rajasekharporeddy:test_branch7 
PiperOrigin-RevId: 640652642
 2024-06-05 14:35:35 -07:00
rajasekharporeddy
be41f309ed Add code examples to jax.scipy.fft.idct and jax.scipy.fft.idctn docs 2024-06-06 01:45:07 +05:30
jax authors
913ff50000 Merge pull request #21418 from rajasekharporeddy:test_branch6 
PiperOrigin-RevId: 640617948
 2024-06-05 12:53:17 -07:00
jax authors
ba496a4c53 Merge pull request #21624 from alvarosg:master 
PiperOrigin-RevId: 640217359
 2024-06-04 11:15:45 -07:00
rajasekharporeddy
bb6cb2c0de Add code examples to jax.scipy.fft.dct doc 2024-06-04 22:10:27 +05:30
rajasekharporeddy
0df26689bd Add code example to jax.scipy.fft.dctn doc 2024-06-04 21:41:46 +05:30
Alvaro
4a5b693a7e Fix to scipy.ndimage.map_coordinates documentation. 
This correctly documents the current behavior of `constant`, and `wrap` modes.
 2024-06-04 15:47:33 +00:00
jax authors
82ad1daad5 Merge pull request #21193 from sh0416:main 
PiperOrigin-RevId: 637938498
 2024-05-28 10:12:29 -07:00
Seonghyeon
818e7d92a4 Fix rel_entr behavior at boundary value 2024-05-28 13:17:28 +00:00
Sergei Lebedev
f5617d7323 Removed noop # type: ignore comments 
mypy should now flag these by default.
 2024-05-19 21:01:29 +01:00
Sergei Lebedev
c3bc88d5e4 Bumped mypy to 1.10.0 and ruff to 0.4.4 2024-05-16 23:16:32 +01:00
Jake VanderPlas
c3c239325f kl_div: fix incorrect formula in doc 2024-05-14 09:25:04 -07:00
Jake VanderPlas
ff67e51e7e Remove last scipy imports 2024-05-03 10:20:05 -07:00
jax authors
3952b7e78a Merge pull request #21042 from jakevdp:scipy-docs 
PiperOrigin-RevId: 630212828
 2024-05-02 16:36:46 -07:00
Jake VanderPlas
27c0c41b44 Remove remaining top-level scipy imports 2024-05-02 14:37:44 -07:00
Jake VanderPlas
18e4cfa911 DOC: Improve remaining jax.scipy docstrings 2024-05-02 10:29:00 -07:00
jax authors
d02ec9f90d Merge pull request #21018 from jakevdp:scipy-linalg-docs 
PiperOrigin-RevId: 629893239
 2024-05-01 17:57:29 -07:00
Jake VanderPlas
d51ccdf628 DOC: Improve docstrings for jax.scipy.linalg 2024-05-01 17:36:24 -07:00
Jake VanderPlas
18703d9385 DOC: Improve docstrings for jax.scipy.signal 2024-05-01 14:26:00 -07:00
Jake VanderPlas
1936317b8b jax.scipy.stats: manually document functions to avoid scipy import 2024-05-01 09:26:26 -07:00
jax authors
1b5c49e752 Merge pull request #20822 from jakevdp:scipy-special-doc 
PiperOrigin-RevId: 629129623
 2024-04-29 11:27:23 -07:00
Jake VanderPlas
74f1d8897c DOC: add manual documentation to jax.scipy.special functions. 
This lets us give more implementation-specific information, and
lets us avoid a needless dependency on scipy.
 2024-04-29 10:58:07 -07:00
Jake VanderPlas
75921162ab jax.scipy.fft: manually document functions to avoid scipy import 2024-04-29 09:23:43 -07:00
jax authors
8b1418244b Merge pull request #20885 from rajasekharporeddy:test_branch4 
PiperOrigin-RevId: 627486343
 2024-04-23 13:29:40 -07:00
rajasekharporeddy
c536eea1e5 Fix jax.scipy.stats.beta.logpdf to emulate scipy.stats.beta.logpdf 2024-04-24 01:24:09 +05:30
jax authors
ba57ce3bd1 Merge pull request #20891 from rajasekharporeddy:test_branch1 
PiperOrigin-RevId: 627472771
 2024-04-23 12:39:18 -07:00
rajasekharporeddy
95ed0538fd Fix jax.scipy.stats.poisson.logpmf to emulate scipy.stats.poisson.logpmf for non-integer values of k 2024-04-24 00:29:52 +05:30
Jake VanderPlas
568db105ea Add jax.scipy.special.gammasgn 2024-04-18 16:14:55 -07:00
rajasekharporeddy
660d612e6a DOC: Fix docstring typos in scipy special functions 2024-04-16 16:05:39 +05:30
Jake VanderPlas
97b15cc64b BUG: fix sign of beta() 2024-04-08 11:06:08 -07:00
jax authors
66427f8800 Merge pull request #20528 from rajasekharporeddy:test_branch1 
PiperOrigin-RevId: 622894800
 2024-04-08 10:54:26 -07:00
Meekail Zain
8b7aae586b Update jnp.clip to Array API 2023 standard 2024-04-04 22:55:10 +00:00
rajasekharporeddy
4f3d27bde8 Update jax.scipy.special.ndtri to return NaN for the values outside of the range [0, 1] 2024-04-03 11:45:21 +05:30
Jake VanderPlas
9e01afe7af Add jax.numpy.trapezoid 
This function has been added to NumPy in version 2.0, as a replacement
for the already deprecated trapz function.
 2024-04-01 13:05:20 -07:00
rajasekharporeddy
4d6a53fb63 Add Hilbert matrix to jax.scipy.linalg 2024-03-20 22:55:03 +05:30
Jake VanderPlas
f35538be6a DOC: fix two minor doc issues 2024-03-05 15:22:42 -08:00
Jake VanderPlas
85f205bdc7 typing: fix incorrect tuple annotations 2024-02-26 10:53:19 -08:00