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

92 Commits

Author SHA1 Message Date
Jake VanderPlas
29a8cce66c jax.numpy: require boolean dtype for where argument 2024-12-05 09:27:19 -08:00
Jake VanderPlas
e9864c69da Make logaddexp and logaddexp2 into ufuncs 2024-11-18 09:27:36 -08:00
Jake VanderPlas
f652b6ad6a Set __module__ attribute for objects in jax.numpy 2024-11-15 06:03:54 -08:00
Jake VanderPlas
4a3e1155b9 cleanup: delete unused argument from internal reduction helper 2024-11-14 13:07:15 -08:00
jax authors
842d93e131 Merge pull request #24810 from jakevdp:bitwise-reduce 
PiperOrigin-RevId: 696313033
 2024-11-13 16:42:33 -08:00
Jake VanderPlas
6e1aa3c1e7 Specialize ufunc.reduce for monoidal binary ufuncs. 2024-11-12 10:10:18 -08:00
Jake VanderPlas
44c6883cee Fix debug_nans false positive in jnp.quantile 2024-11-05 15:36:14 -08:00
Jake VanderPlas
02daf75f97 Add new jnp.cumulative_prod function. 
This follows the API of the similar function added in NumPy 2.1.0
 2024-10-25 13:45:54 -07:00
Jake VanderPlas
0276514a10 Improve docs for jnp.average 2024-10-08 09:23:16 -07:00
Peter Hawkins
562e9e8dff Fix an incorrect output for jnp.cumsum. 
If dtype=bool but a non-bool input is passed, we should test for
non-equality with zero rather than performing a cast to integer.
 2024-09-24 14:46:44 +00:00
Michael Hudgins
d4d1518c3d Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax 
PiperOrigin-RevId: 676843138
 2024-09-20 07:52:33 -07:00
Jake VanderPlas
e7d3785b18 Refactor & document cumulative reductions 2024-09-04 12:20:24 -07:00
Jake VanderPlas
a3d6cf007e First pass at ufunc interfaces for several jax.numpy functions 2024-08-30 11:53:02 -07:00
jax authors
be13d4055e Merge pull request #23236 from jakevdp:fix-mean-norm 
PiperOrigin-RevId: 667634188
 2024-08-26 10:38:01 -07:00
Jake VanderPlas
4b1c9f483c jnp.mean: fix normalizer for large arrays 2024-08-26 09:04:15 -07:00
Jake VanderPlas
9090b8a4f9 Better docs for jnp quantile & percentile 2024-08-23 13:38:20 -07:00
Jake VanderPlas
2c221f2d5a Register several jax.numpy argument name deprecations 2024-08-22 09:41:53 -07:00
rajasekharporeddy
a00c94c5a2 Improved docs for jnp.median and nanmedian 2024-07-16 17:27:46 +05:30
rajasekharporeddy
9b8ed82355 Improved docs for jnp.nanvar and jnp.nanstd 2024-07-12 06:31:23 +05:30
jax authors
4f8cff5934 Merge pull request #22369 from rajasekharporeddy:testbranch2 
PiperOrigin-RevId: 651504020
 2024-07-11 12:50:23 -07:00
jax authors
33e56d4261 Merge pull request #22392 from rajasekharporeddy:testbranch3 
PiperOrigin-RevId: 651464579
 2024-07-11 10:51:26 -07:00
rajasekharporeddy
bf739371b4 Improved docs for jnp.nansum and nanprod 2024-07-11 23:01:11 +05:30
rajasekharporeddy
1a34ded56f Improved docs for jnp.nanmean 2024-07-11 22:38:30 +05:30
rajasekharporeddy
3d8f702ff6 Improved docs for jnp.nanmin and jnp.nanmax 2024-07-11 10:20:53 +05:30
rajasekharporeddy
cb134cca61 Improved docs for jnp.ptp and jnp.count_nonzero 2024-07-09 12:15:27 +05:30
Jake VanderPlas
2a6eabba1f Fix debug_nans false positive in jnp.var 2024-07-08 04:33:23 -07:00
jax authors
68e9683040 Merge pull request #22167 from rajasekharporeddy:testbranch1 
PiperOrigin-RevId: 648535148
 2024-07-01 17:33:37 -07:00
rajasekharporeddy
6713e9dc8e Improve docs for jnp.var and std 2024-07-01 19:42:08 +05:30
rajasekharporeddy
9623dcfa95 Update jnp.mean doc 2024-07-01 12:07:52 +05:30
jax authors
43c9fe89d0 Merge pull request #22105 from rajasekharporeddy:testbranch4 
PiperOrigin-RevId: 647090732
 2024-06-26 15:05:06 -07:00
jax authors
2985b623bc Merge pull request #22117 from hawkinsp:pyver2 
PiperOrigin-RevId: 647079147
 2024-06-26 14:29:04 -07:00
Peter Hawkins
7f4ef63cd8 Run pyupgrade --py310-plus. 
Also apply manual fixes to import sorting and unused imports.
 2024-06-26 16:10:18 -04:00
rajasekharporeddy
04a2e4216d Update docs of jnp.all and jnp.any 2024-06-27 00:38:46 +05:30
rajasekharporeddy
2702f2113a Update jnp.prod doc 2024-06-27 00:36:02 +05:30
rajasekharporeddy
7281c39add Update jnp.min/amin docs 2024-06-26 23:12:47 +05:30
jax authors
7b73e8ecf7 Merge pull request #22083 from rajasekharporeddy:testbranch2 
PiperOrigin-RevId: 646550598
 2024-06-25 11:36:26 -07:00
rajasekharporeddy
ee0bf776fe Update jnp.max/amax docs 2024-06-25 23:07:59 +05:30
rajasekharporeddy
b975ca1932 Better doc for jnp.sum 2024-06-25 22:25:23 +05:30
jax authors
bab7f40dec Merge pull request #21262 from vfdev-5:depr-change-ddof-to-correction-21088 
PiperOrigin-RevId: 636949170
 2024-05-24 09:47:27 -07:00
vfdev-5
55f8284e27 Added correction arg in jnp.var and jnp.std 
Description:
- Added correction arg in jnp.var and jnp.std
- Addresses https://github.com/google/jax/issues/21088
- Updated signatures in init.pyi
- Updated tests
 2024-05-24 16:16:12 +00:00
vfdev-5
3c201e0b8c jnp.var returns nan if N-ddof <= 0 
Description:
- Updated jnp.var function to explicitly return np.nan if normalizer is non-positive
- Added a test for jnp.var and jnp.std

Fixed #21330
 2024-05-24 13:45:59 +00:00
Jake VanderPlas
bfbde5efd5 jnp.quantile & friends: properly deprecate interpolation 2024-05-16 15:13:25 -07:00
Meekail Zain
34c5163fd2 Refactored common upcast for integral-type accumulators 2024-05-06 15:13:10 +00:00
Meekail Zain
ceeb975735 Add new cumulative_sum function to numpy and array_api 2024-04-16 19:57:55 +00:00
Hongyu Chiu
c295b1655c Fix axis=None and keepdims=True for jnp.quantile and jnp.median 
Fix `axis=None` and `keepdims=True` for `jnp.quantile` and `jnp.median`

Remove `print`

Update tests
 2024-03-15 09:05:02 +08:00
Jake VanderPlas
43a9faa06a Rename _wraps to implements 2024-01-24 14:14:19 -08:00
Jake VanderPlas
9b46e2d6a3 Support float8 in reduce_min & reduce_max 2023-12-18 13:37:45 -08:00
Sergei Lebedev
f936613b06 Upgrade remaining sources to Python 3.9 
This PR is a follow up to #18881.

The changes were generated by adding

    from __future__ import annotations

to the files which did not already have them and running

    pyupgrade --py39-plus --keep-percent-format {jax,tests,jaxlib,examples,benchmarks}/**/*.py
 2023-12-13 10:29:45 +00:00
Jake VanderPlas
e7932858d4 Fix dtype-related docs for cumsum/cumprod 2023-11-06 14:31:25 -08:00
Jake VanderPlas
6d6c12a14d jnp.nanmean: avoid integer overflow for large arrays 2023-11-03 13:15:18 -07:00