JAX operations are for ArrayLike objects but we are passing Numeric typed objects

[Qazalbash]

Description

Numeric and ArrayLike are union types defined as,

Numeric = Union[Array, np.ndarray, np.bool_, np.number, bool, int, float]
Union[
  Array,  # JAX array type
  np.ndarray,  # NumPy array type
  np.bool_, np.number,  # NumPy scalar types
  bool, int, float, complex,  # Python scalar types
]

Numeric is only short of complex, other than that everything is the same. We have to come up with a method to allow Numeric typed objects for JAX operation.

/media/gradf/Academic/project/jaxampler/jaxampler/_rvs/pareto.py:62:68 - error: Operator "-" not supported for type "Numeric" when expected type is "ArrayLike" (reportGeneralTypeIssues)

What jaxampler version are you using?

nightly

Which accelerator(s) are you using?

CPU

Additional system info?

Linux

NVIDIA GPU info

No response