PyTorch
API: —
Strategy: Macro 'torch.tril(torch.ones(({N}, {M} if {M} is not None else {N}), dtype={dtype}), diagonal={k})'
JAX (Core)
API: jax.numpy.tri
Strategy: Direct Mapping
NumPy
API: numpy.tri
Strategy: Direct Mapping
Keras
API: keras.ops.tri
Strategy: Direct Mapping
TensorFlow
API: tf.experimental.numpy.tri
Strategy: Direct Mapping
Apple MLX
API: mlx.core.tri
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.tri
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.tri
Strategy: Direct Mapping