PyTorch
API: torch.rsqrt
Strategy: Direct Mapping
JAX (Core)
API: jax.lax.rsqrt
Strategy: Direct Mapping
NumPy
API: —
Strategy: Macro '1.0 / numpy.sqrt({input})'
Keras
API: keras.ops.rsqrt
Strategy: Direct Mapping
TensorFlow
API: tf.math.rsqrt
Strategy: Direct Mapping
Apple MLX
API: mlx.core.rsqrt
Strategy: Direct Mapping
Flax NNX
API: jax.lax.rsqrt
Strategy: Direct Mapping
PaxML / Praxis
API: jax.lax.rsqrt
Strategy: Direct Mapping