JAX (Core)
API: jax.scipy.special.xdivy
Strategy: Direct Mapping
NumPy
API: —
Strategy: Macro 'np.where({x} == 0, 0, {x} / {y})'
TensorFlow
API: tf.math.xdivy
Strategy: Direct Mapping
Flax NNX
API: jax.scipy.special.xdivy
Strategy: Direct Mapping
PaxML / Praxis
API: jax.scipy.special.xdivy
Strategy: Direct Mapping