HardSiluΒΆ

Hard SiLU (swish) activation function.

Abstract Signature:

HardSilu(x: Array)

PyTorch

API: torch.nn.functional.hardswish
Strategy: Direct Mapping

JAX (Core)

API: jax.nn.hard_swish
Strategy: Direct Mapping

Keras

API: keras.ops.hard_swish
Strategy: Direct Mapping

TensorFlow

API: tf.nn.hard_swish
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.hard_swish
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.hard_silu
Strategy: Direct Mapping

PaxML / Praxis

API: jax.nn.hard_swish
Strategy: Direct Mapping