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