mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
17,013 Commits 503 Branches 263 Tags
Commit Graph

2 Commits

Author SHA1 Message Date
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