Jake VanderPlas
|
30b687c486
|
[x64] make jnp.poly* functions work under strict dtype promotion
|
2022-05-31 14:22:49 -07:00 |
|
Jiajie Li
|
128e51c638
|
Add polydiv to jax.numpy
Fix code style, fix tests
Add warning when use polydiv with trim_leading_zeros
Update warning for polydiv
Co-authored-by: Jake Vanderplas <jakevdp@gmail.com>
Enable type check in _CompileAndCheck
Fix cutoff
Fix cut-off in polydiv
Add trim_zeros_tol, remove redundant code in polydiv
Remove unused import
Fix trim_zero_tol usage in polydiv
|
2022-04-13 18:31:27 +00:00 |
|
Jake VanderPlas
|
603bb3c5ca
|
lax_numpy: move poly functions into numpy.polynomial
|
2022-03-17 13:28:54 -07:00 |
|
Peter Hawkins
|
4e21922055
|
Use imports relative to the jax package consistently, rather than .-relative imports.
This is more consistent, since currently we use a mix of both styles. It may also help pytype yield more accurate types.
PiperOrigin-RevId: 412057514
|
2021-11-24 07:48:29 -08:00 |
|
Peter Hawkins
|
a84426cb8f
|
Switch internal users of jax.ops.index_... to use x.at[x].set() APIs.
|
2021-09-13 19:48:29 -04:00 |
|
Jake VanderPlas
|
8e789c7380
|
Run doctest on all source files except jax2tf
|
2021-04-05 10:39:59 -07:00 |
|
Peter Hawkins
|
aa107cf1f4
|
Move jax.numpy internals into jax._src.numpy.
|
2020-10-16 20:35:19 -04:00 |
|