PyTorch
API: —
Strategy: Macro '{input}.long()'
JAX (Core)
API: —
Strategy: Macro '{input}.astype(jax.numpy.int64)'
NumPy
API: —
Strategy: Macro '{input}.astype(numpy.int64)'
Keras
API: —
Strategy: Macro 'keras.ops.cast({input}, 'int64')'
TensorFlow
API: —
Strategy: Macro 'tf.cast({input}, tf.int64)'
Apple MLX
API: —
Strategy: Macro '{input}.astype(mlx.core.int64)'
Flax NNX
API: —
Strategy: Macro '{input}.astype(jax.numpy.int64)'
PaxML / Praxis
API: —
Strategy: Macro '{input}.astype(jax.numpy.int64)'