PyTorch
API: torch.ShortTensor
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'
NumPy
API: —
Strategy: Macro 'np.array({data}, dtype=np.int16)'
Keras
API: —
Strategy: Macro 'keras.ops.cast({data}, 'int16')'
TensorFlow
API: —
Strategy: Macro 'tf.cast({data}, tf.int16)'
Apple MLX
API: —
Strategy: Macro 'mx.array({data}, dtype=mx.int16)'
Flax NNX
API: —
Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'
PaxML / Praxis
API: —
Strategy: Macro 'jnp.array({data}, dtype=jnp.int16)'