ControlDependency ================= Insert dependencies between arrays in the graph. **Abstract Signature:** ``ControlDependency(inputs: Union[Tensor, Sequence[Tensor]], dependencies: Union[Tensor, Sequence[Tensor]])`` .. raw:: html

JAX (Core)

API:
Strategy: Plugin (jax_control_dependencies)

TensorFlow

API: tf.control_dependencies
Strategy: Direct Mapping

Apple MLX

API: mlx.core.depends
Strategy: Direct Mapping

Flax NNX

API:
Strategy: Plugin (jax_control_dependencies)

PaxML / Praxis

API:
Strategy: Plugin (jax_control_dependencies)