Jake VanderPlas
093b7032a8
Implement jnp.from* array creation functions
2022-03-29 10:52:47 -07:00
Jake VanderPlas
09105b2d58
Update outdated jax.numpy module docs
2022-03-15 11:02:59 -07:00
Jake VanderPlas
ed2550999f
implement jnp.copy
2022-03-01 11:56:36 -08:00
Peter Hawkins
9ea55468ab
[JAX] Update users of jax.ops.index... functions, which are deprecated.
...
* replace uses of `jax.ops.index[...]` with `jax.numpy.index_exp[...]`, which is a standard NumPy function that does the same thing.
* remove some redundant uses of `jax.ops.index[...]`, where the expression is passed directly to an indexed accessor function like `.at[...]`.
* update some remaining users of `jax.ops.index_update(x, jax.ops.index[idx], y)` to use the `x.at[idx].set(y)` APIs.
PiperOrigin-RevId: 406162068
2021-10-28 09:54:26 -07:00
Jake VanderPlas
c35b2f2485
DOC: move index update API docs to jnp.ndarray.at
...
- Add docstring to abstract property
- Add explicit HTML documentation of this property
- Mark index update functions as deprecated, linking to this documentation
2021-10-01 14:06:08 -07:00
Jake VanderPlas
3e0e47cde6
DOC: add missing jax.numpy functions
2021-09-16 13:29:08 -07:00
Dan Zheng
035b27ffb2
Add jnp.get_printoptions and jnp.printoptions.
...
`jnp.set_printoptions` already exists but other printing utilities do not.
`jnp.get_printoptions` forwards `numpy.get_printoptions`.
`jnp.printoptions` forwards `numpy.printoptions`.
2021-09-13 15:12:52 -07:00
avani17101
c4d300c4ca
Implement np.polyfit
2021-08-31 10:52:45 +05:30
elliotwaite
7392a57b75
DOC: many small fixes
2021-08-04 16:55:13 -07:00
Jake VanderPlas
9be4a3e83f
Fix typo/formatting issue in DeviceArray docs
2021-06-07 13:11:37 -07:00
Jake VanderPlas
80a310f8d8
DOC: add note about array views in numpy docs
2021-05-25 16:54:28 -07:00
Lukas Geiger
b4b02cb7c3
Add jnp.resize
2021-05-21 09:25:49 +01: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
Jake VanderPlas
26f74e64a6
DOC: add documentation of DeviceArray object properties & methods
2021-05-10 12:37:59 -07: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
minoring
4c67dd1f48
Implement jnp.ogrid
...
Related to #5850
2021-04-06 08:37:07 +09:00
minoring
a2209aa509
Implement jnp.mgrid
...
Related to #5850
2021-04-02 12:38:20 +09:00
Jake VanderPlas
737e4796cd
Initial implementation of jnp.delete
2021-03-16 17:05:23 -07:00
jax authors
932e118dd2
Merge pull request #5665 from ashutoshvarma:numpy-setxor1d
...
PiperOrigin-RevId: 359347458
2021-02-24 12:43:14 -08:00
Ashutosh Varma
c223ef9d07
add support for setxor1d
2021-02-24 23:28:42 +05:30
Jonathan Terhorst
4c202ad222
implement np.polyint ( #70 )
2021-02-18 11:08:41 -05:00
Ashutosh Varma
108d078c5f
add support for union1d
...
add union1d in jax.numpy which closely follows
numpy implementaion
2021-02-06 22:19:16 +05:30
Alex Dragan
b7fee456e6
[impl] Add support for setdiff1d
2020-10-21 20:17:12 +03:00
Alex Minnaar
0485fd8736
adding histogram2d implementation
2020-10-10 12:03:46 -04:00
Jake VanderPlas
2fccbbf65b
Implement jnp.choose
2020-10-02 13:13:21 -07:00
jax authors
b609040ce6
Merge pull request #4413 from alexminnaar:master
...
PiperOrigin-RevId: 334946450
2020-10-01 19:26:55 -07:00
Alex Minnaar
0893b08a6f
histogramdd implemenation
2020-10-01 18:02:33 -04:00
Jake VanderPlas
ce1ce9cb27
Implement jnp.array_equiv
2020-09-21 10:37:55 -07:00
Jake Vanderplas
e0af77fbca
Implement jnp.ravel_multi_index() ( #4313 )
2020-09-17 12:42:22 -07:00
Jake Vanderplas
e18a973198
implement jnp.apply_over_axes ( #4225 )
2020-09-16 13:30:08 -07:00
Jake Vanderplas
8a18b10fa4
implement jnp.apply_along_axis ( #4253 )
2020-09-11 08:47:05 -07:00
Alex Riley
708d07d5ff
Add jax.numpy.array_split ( #4197 )
2020-09-02 16:13:17 -07:00
gaurav pathak
29f7fa753a
Add implementation of jax.numpy.trim_zeros ( #4027 )
2020-08-18 13:40:45 -07:00
Jake Vanderplas
d551cec6e8
Add Bessel functions in jax.numpy & jax.scipy.special ( #4007 )
2020-08-10 10:10:59 -07:00
Jake Vanderplas
dd304f62c1
Add jax.numpy.invert ( #3998 )
2020-08-10 08:41:54 -07:00
Justin Lebar
a98103f678
Add missing functions to jax.numpy docs ( #3981 )
2020-08-06 12:57:19 -07:00
Justin Lebar
2277728e8c
Add numpy.vectorize to list of numpy fns in docs ( #3968 )
2020-08-05 18:37:29 -07:00
Jake Vanderplas
d6f131aaee
implement jax.numpy.interp ( #3949 )
2020-08-04 12:39:04 -07:00
Jake Vanderplas
1229f44adf
add jnp.piecewise implementation ( #3943 )
2020-08-03 12:31:52 -07:00
Jake Vanderplas
e0a8d44861
Add jnp.modf() & improve test coverage for related functions ( #3894 )
2020-07-29 12:53:28 -07:00
Vaibhav Srivastav
3aa37d3af4
Replicating sort_complex functionality from np.sort_complex to jax.numpy ( #3870 )
2020-07-27 14:27:36 -07:00
Jake Vanderplas
04a6238d4e
implement jax.numpy.lexsort ( #3812 )
2020-07-22 12:48:49 -07:00
Alex Dragan
0d81e988d8
Implement np.intersect1d ( #3726 )
...
* Implement np.intersect1d
* Add jitable helper to function
* Fix argsort failing tests
* Fix linter errors
2020-07-12 22:32:41 -07:00
Alex Dragan
412b9d5209
hfft and ihfft implementation ( #3664 )
2020-07-10 10:34:59 -07:00
Ayush Shridhar
a44bc0c2c0
Add np.diag_indices_from ( #3500 )
2020-06-24 09:31:16 -07:00
Peter Hawkins
fede6d90f6
Add missing functions to documentation. ( #3538 )
2020-06-24 12:16:56 -04:00
Ayush Shridhar
5ee936fdcf
Add polyder numpy function ( #3403 )
2020-06-17 09:43:50 -07:00
Ayush Shridhar
832431db9e
Add np.triu_indices_from function ( #3346 )
2020-06-10 09:57:35 -07:00
