PyTorch
API: torch.heaviside
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.heaviside
Strategy: Direct Mapping
NumPy
API: numpy.heaviside
Strategy: Direct Mapping
Keras
API: keras.ops.numpy.heaviside
Strategy: Direct Mapping
TensorFlow
API: tf.math.heaviside
Strategy: Direct Mapping
Apple MLX
API: —
Strategy: Custom / Partial
Flax NNX
API: jax.numpy.heaviside
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.heaviside
Strategy: Direct Mapping