Matthew R. Becker
Matthew R. Becker
Also to be clear, we don't want to use autodiff in this kernel, which I realize is outside of the A part of JAX. ;p
Sure here you go! ```python import time import numpy as np from numba import njit @njit def count_weighted_pairs_3d_cpu( x1, y1, z1, w1, x2, y2, z2, w2, rbins_squared, result): n1 =...
I get ```bash (anl) clarence:Desktop beckermr$ python numba_cou_jit.py numba: 2.117280626296997 seconds ```
I have not benchmarked the gpu version against Jax. Note however that this the simplest of a class of kernels. Idk if the version you posted will generalize to those.
What is the ABI compatibility? Will these be like pypy with a migration job and a separate Abi build?
@jaimergp shall we merge this?
This is what we found too. I am going to close this issue for now.
I just want to comment that it cannot possibly be correct behavior for “mamba update —all” to make updates to the packages that violate constraints from the downloaded repodata.
Note for future ppl. Only wp(rp) remains on this issue.