jax.numpy.heaviside =================== .. currentmodule:: jax.numpy .. autofunction:: heaviside