[math] Update CustomOpByNumba
to support JAX version >= 0.4.24 (#669)
#1254
Loading
CustomOpByNumba
to support JAX version >= 0.4.24 (#669)
#1254