Jake VanderPlas
81e627d5bd
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
Peter Hawkins
f51a05a889
Remove jax.ops.index... functions.
...
These functions have been deprecated and have issued a DeprecationWarning since jax 0.2.22 in October 2021.
2022-02-24 09:36:28 -05:00
Jake VanderPlas
928087ada0
DOC: add info about repeated indices to jax.ops docs
2022-01-27 10:22:25 -08:00
Peter Hawkins
2a6f836d30
Fix rendering of lax.gather docs.
...
Remove spurious extra argument to `get()` in documentation.
2021-09-24 09:59:10 -04:00
Peter Hawkins
867068821e
Drop out-of-bounds indexes in gather.
2021-09-23 10:35:03 -04:00
Stephan Hoyer
d204325c1f
Don't refer to deprecated jax.ops.index_update in error messages
...
I've also updated the docs for ``jax.ops`` to note that ``at[].set()``
is guaranteed to be performed in-place under JIT. Someone who knows XLA
well should double check that fact!
2021-09-01 20:43:13 -07:00
Peter Hawkins
d005e38f78
Promote the x.at[idx].set(y) operators as the preferred way to do indexed updates.
...
Mark the index_update() etc. operators as deprecated in the documentation.
Add new .divide and .power operators. Fixes #2694 .
Add .multiply as an alias for .mul. To be more numpy-like we should probably prefer the longer names.
2021-05-10 20:32:00 -04:00
Jake VanderPlas
cbb7052379
Implement segment_prod, segment_max, segment_min
2021-04-09 12:06:51 -07:00
Jake VanderPlas
0796bfe6e7
errors: add NonConcreteBooleanIndexError & debugging tips
2021-03-23 11:23:20 -07:00
Daniel Johnson
835949a36a
Fix typo
2020-04-17 01:43:00 +00:00
Daniel Johnson
a5efe84256
Update names and documentation.
2020-04-17 01:33:08 +00:00
Peter Hawkins
714b276b9a
Implement jax.ops.index_mul. ( #2696 )
...
* Implement jax.ops.index_mul.
* Add index_mul to documentation.
* Fix RHS JVP rule for scatter_mul, fix test bug that meant it was not tested.
* Fix typo in docstring.
2020-04-13 16:16:34 -04:00
Peter Hawkins
4eb1820ae2
Add documentation to JAX modules.
2019-07-21 15:55:47 -04:00
Justin Lebar
3da993802b
Update docs to include index_{min,max}.
2019-06-23 15:32:43 -07:00
Peter Hawkins
e6082d203d
Add segment_sum to the docs and fix its rendering.
...
Minor doc fixes.
2019-05-08 21:18:17 -04:00
Peter Hawkins
eefc0a7378
Add jax.ops to documentation.
2019-03-04 20:12:22 -05:00
