Expand scico.numpy ufunc list

[Michael-T-McCann]

Universal functions (ufuncs) are functions that are applied elementwise on arrays. These should be very easy to make work on BlockArrays, but currently only the following are supported in scico.numpy:

scico/scico/numpy/__init__.py

Lines 90 to 105 in 912bb9e

	_ufunc_functions = [
	("abs", jnp.abs),
	jnp.maximum,
	jnp.sign,
	jnp.where,
	jnp.true_divide,
	jnp.floor_divide,
	jnp.real,
	jnp.imag,
	jnp.conjugate,
	jnp.angle,
	jnp.exp,
	jnp.sqrt,
	jnp.log,
	jnp.log10,
	]

Eventually, we'd like to have every ufunc from numpy that is currently implemented in jax. (I think) this is only requires cross-checking these lists and adding to _ufunc_functions in numpy/__init__.py.

[bwohlberg]

@Michael-T-McCann: Was this perhaps resolved in #259?

[Michael-T-McCann]

@Michael-T-McCann: Was this perhaps resolved in #259?

#259 added significantly to this list. I don't think every possible (or even reasonable) jax.numpy function is wrapped, but most are.

[bwohlberg]

OK. Closing for now.