Jake VanderPlas
e9864c69da
Make logaddexp and logaddexp2 into ufuncs
2024-11-18 09:27:36 -08:00
Jake VanderPlas
6467d03925
Make jnp.subtract a ufunc
2024-10-21 10:11:51 -07:00
Jake VanderPlas
6229511f6a
Make jnp.negative a ufunc & add unary ufunc tests
2024-09-24 05:23:27 -07: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
a3d6cf007e
First pass at ufunc interfaces for several jax.numpy functions
2024-08-30 11:53:02 -07:00
Jake VanderPlas
3833c46d10
jnp.vectorize: respect numpy_rank_promotion config
2024-07-08 09:03:03 -07:00
Jake VanderPlas
f090074d86
Avoid 'from jax import config' imports
...
In some environments this appears to import the config module rather than
the config object.
2024-04-11 13:23:27 -07:00
Jake VanderPlas
e59a0506fe
Deprecate jax.tree_map in favor of jax.tree.map
2024-02-22 11:35:39 -08:00
Sergei Lebedev
cbcaac2756
MAINT Migrate remaining internal/test modules to use state objects
...
The motivation here is to gradually replace all dynamic lookups on `jax.config`
with statically-typed state objects, which are more type checker/IDE friendly.
This is a follow up to #18008 .
2023-10-12 17:32:15 +01:00
Jake VanderPlas
41a7d66686
jnp.frompyfunc: fix .at() edge case
2023-10-09 11:24:25 -07:00
Jake VanderPlas
cb7c7ad942
jnp.ufunc: add fast paths for add/prod reductions
2023-08-28 08:30:23 -07:00
Jake VanderPlas
ac1233b453
jnp.ufunc: support where argument in ufunc.reduce
2023-08-24 15:01:44 -07:00
Jake VanderPlas
61f50bd3b6
jnp.ufunc: minor cleanups & test fixes
2023-08-14 15:19:46 -07:00
Jake VanderPlas
b3a02e1b62
jnp.ufunc: add __hash__ method and jit methods by default
...
This allows the JIT cache to work properly with ufunc methods, because bound
methods are created with a new ID each time.
2023-08-14 13:06:18 -07:00
Jake VanderPlas
ad8e719b82
Add jnp.ufunc and jnp.frompyfunc
2023-08-10 14:58:18 -07:00
