PyTorch
API: torch.short
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.astype
Strategy: Macro '{input}.astype(jnp.int16)'
NumPy
API: numpy.astype
Strategy: Macro '{input}.astype(np.int16)'
Keras
API: keras.ops.cast
Strategy: Macro 'keras.ops.cast({input}, 'int16')'
TensorFlow
API: tf.cast
Strategy: Macro 'tf.cast({input}, tf.int16)'
Apple MLX
API: mlx.core.astype
Strategy: Macro '{input}.astype(mx.int16)'
Flax NNX
API: jax.numpy.astype
Strategy: Macro '{input}.astype(jnp.int16)'
PaxML / Praxis
API: jax.numpy.astype
Strategy: Macro '{input}.astype(jnp.int16)'