Jake VanderPlas
a1140e9246
Better docs for jnp.cross
2024-10-21 05:59:22 -07:00
Jake VanderPlas
0a85ba5f82
Better documentation for jnp.load
2024-10-19 06:20:20 -07:00
Yash Katariya
57a95a77ff
[sharding_in_types] Support jnp.array with sharding_in_types. When the input array has a sharding, propagate it through without dropping the sharding.
...
PiperOrigin-RevId: 687089357
2024-10-17 16:51:41 -07:00
jax authors
2e5920db76
Merge pull request #24326 from rajasekharporeddy:testbranch1
...
PiperOrigin-RevId: 686807960
2024-10-17 00:59:32 -07:00
rajasekharporeddy
aaefa82230
Better docs for jax.numpy:equal and not_equal
2024-10-17 07:59:04 +05:30
jax authors
f332dd561a
Merge pull request #24351 from jakevdp:sinc-doc
...
PiperOrigin-RevId: 686715954
2024-10-16 18:40:56 -07:00
Jake VanderPlas
e1f280c843
CI: enable additional ruff formatting checks
2024-10-16 16:09:54 -07:00
Jake VanderPlas
6ca7228d0d
Improve docs for jnp.sinc
2024-10-16 16:08:39 -07:00
jax authors
ebac2e4421
Merge pull request #24323 from selamw1:tile_doc
...
PiperOrigin-RevId: 686637204
2024-10-16 14:11:44 -07:00
selamw1
24b6f50938
tile_docstring_added
2024-10-15 15:34:01 -07:00
Jake VanderPlas
63f4299e0e
Improve array setitem error
2024-10-15 11:39:31 -07:00
jax authors
94152e8dfc
Merge pull request #24305 from jakevdp:signbit-doc
...
PiperOrigin-RevId: 686174793
2024-10-15 11:21:07 -07:00
Jake VanderPlas
dd4a0408a4
Improve docs for jnp.invert and related functions
2024-10-15 08:57:19 -07:00
Jake VanderPlas
a096986844
Better docs for jnp.signbit
2024-10-14 19:51:42 -07:00
Jake VanderPlas
0c307fe706
Better docs for jnp.apply_along_axis & apply_over_axes
2024-10-14 14:48:33 -07:00
Jake VanderPlas
9b9fc5afae
Improve documentation for jnp.histogram & related APIs
2024-10-14 06:44:54 -07:00
Peter Hawkins
94abaf430e
Add lax.FftType.
...
We had never provided a public name for the enum of FFT types; instead it was only known by a semi-private name (jax.lib.xla_client.FftType). Add a public name (jax.lax.FftType) and deprecate the private one.
We define a new FftType IntEnum rather than trying to expose the one in xla_client. The xla_client definition was useful when building classic HLO, but we no longer do that so there's no reason we need to couple our type to XLA's type.
PiperOrigin-RevId: 684447186
2024-10-10 08:07:35 -07:00
jax authors
db71965c56
Merge pull request #24167 from rajasekharporeddy:testbranch1
...
PiperOrigin-RevId: 684085868
2024-10-09 10:23:44 -07:00
Jake VanderPlas
2f798902b4
Better documentation for jnp.nan_to_num
2024-10-09 06:36:45 -07:00
rajasekharporeddy
ed028be7fb
Better docs for jnp.left_shift
2024-10-09 12:09:33 +05:30
Jake VanderPlas
0276514a10
Improve docs for jnp.average
2024-10-08 09:23:16 -07:00
jax authors
a0ab2a79fb
Merge pull request #24122 from rajasekharporeddy:testbranch2
...
PiperOrigin-RevId: 683167116
2024-10-07 06:58:07 -07:00
jax authors
4cb3a6d30d
Merge pull request #24118 from jakevdp:fft-docs
...
PiperOrigin-RevId: 683158665
2024-10-07 06:29:35 -07:00
rajasekharporeddy
9832a11c50
Better doc for jnp.hypot
2024-10-06 21:58:03 +05:30
Jake VanderPlas
45f0e9ad68
Simplify definition of jnp.isscalar
...
The new semantics are to return True for any array-like object with zero dimensions.
Previously we only returned True for zero-dimensional array-like objects with a weak type. This ends up being more confusing/suprising than it needs to be, and the weak type dependence is rarely useful in practice.
PiperOrigin-RevId: 682656411
2024-10-05 07:12:20 -07:00
Tom Natan
ed5ba633d4
Reverts 6cf09f8c24c67ff650b95d174501fff3cb59db0d
...
PiperOrigin-RevId: 682440543
2024-10-04 13:56:27 -07:00
jax authors
46b7bfae91
Merge pull request #24113 from rajasekharporeddy:testbranch1
...
PiperOrigin-RevId: 682344324
2024-10-04 09:15:44 -07:00
Jake VanderPlas
162322fc70
Better docs for fftshift & ifftshift
2024-10-04 06:12:02 -07:00
Jake VanderPlas
0a8f46a6ca
Fix some typos in lax_numpy.py
2024-10-04 03:55:19 -07:00
rajasekharporeddy
321b9bc3f4
Better doc for jnp.heaviside
2024-10-04 14:23:13 +05:30
rajasekharporeddy
75cf4a286c
Better doc for jnp.reciprocal
2024-10-04 04:21:25 +05:30
jax authors
dda4712da0
Merge pull request #24088 from rajasekharporeddy:testbranch1
...
PiperOrigin-RevId: 681965331
2024-10-03 11:41:30 -07:00
rajasekharporeddy
ff8f76aedb
Improve docs for jax.numpy: isfinite, isinf, isposinf, isneginf and isnan
2024-10-03 23:25:49 +05:30
Jake VanderPlas
635e29a0b9
Implement jax.numpy.spacing
...
Somehow we've missed this numpy API up until now.
2024-10-03 10:40:39 -07:00
Sergei Lebedev
4cf33c0239
Added scatter_sub_p
...
The new primitive is used for in-place subtract and update.
Closes #23933
PiperOrigin-RevId: 681754037
2024-10-03 00:27:31 -07:00
Jake VanderPlas
4495daee11
Better docs for jnp.issubdtype & jnp.result_type
2024-10-02 05:46:16 -07:00
Blake Hechtman
ce21a12a07
[JAX] Make a one hot mode of take along axis.
...
PiperOrigin-RevId: 681139055
2024-10-01 13:16:26 -07:00
jax authors
9ad7e2eb42
Merge pull request #24048 from jakevdp:packbits-doc
...
PiperOrigin-RevId: 681093674
2024-10-01 11:15:19 -07:00
jax authors
28098bef93
Merge pull request #24034 from jakevdp:ldexp-doc
...
PiperOrigin-RevId: 681058537
2024-10-01 09:47:56 -07:00
jax authors
9ba90741a8
Merge pull request #23984 from jakevdp:mask-indices-doc
...
PiperOrigin-RevId: 681053740
2024-10-01 09:35:30 -07:00
Jake VanderPlas
ae374e0096
Document jnp.packbits & jnp.unpackbits
2024-10-01 05:33:25 -07:00
Jake VanderPlas
22906b06ed
Improve docs for jnp.ldexp and jnp.frexp
2024-09-30 16:38:21 -07:00
jax authors
31cb3fd36e
Merge pull request #23923 from carlosgmartin:ldexp_custom_jvp
...
PiperOrigin-RevId: 680757259
2024-09-30 16:21:57 -07:00
carlosgmartin
65a58d622c
Edit implementation of jax.numpy.ldexp to get correct gradient.
2024-09-30 18:27:39 -04:00
Jake VanderPlas
36782e8319
jnp.mask_indices: add docs & tests
2024-09-30 15:13:41 -07:00
jax authors
c557db0bd8
Merge pull request #23995 from jakevdp:trapezoid-doc
...
PiperOrigin-RevId: 680734292
2024-09-30 15:10:16 -07:00
jax authors
cdc72787fc
Merge pull request #24025 from jakevdp:gradient-doc
...
PiperOrigin-RevId: 680703792
2024-09-30 13:48:09 -07:00
Jake VanderPlas
36d6bb9013
Better docs for jnp.gradient
...
Also remove skip_params option from util.implements, as this was its last usage.
2024-09-30 13:07:52 -07:00
rajasekharporeddy
5904fe1563
Better doc for jnp.cbrt
2024-09-30 23:35:57 +05:30
Jake VanderPlas
7cc5ea4c7d
Update docs for jnp.trapezoid
2024-09-29 10:08:27 -07:00
