CondΒΆ

Conditional execution (structured control flow).

Abstract Signature:

Cond(pred: Tensor, true_fn: Callable, false_fn: Callable, operands: List[Any] = [])

PyTorch

API: torch.cond
Strategy: Direct Mapping

JAX (Core)

API: jax.lax.cond
Strategy: Direct Mapping

Keras

API: tf.cond
Strategy: Direct Mapping

TensorFlow

API: tf.cond
Strategy: Direct Mapping

Apple MLX

API: β€”
Strategy: Custom / Partial

Flax NNX

API: flax.nnx.cond
Strategy: Direct Mapping

PaxML / Praxis

API: jax.lax.cond
Strategy: Direct Mapping