StreamΒΆ

Context manager for device streams.

Abstract Signature:

Stream(device = None, priority: int = 0)

PyTorch

API: torch.Stream
Strategy: Direct Mapping

JAX (Core)

API: β€”
Strategy: Custom / Partial

NumPy

API: β€”
Strategy: Custom / Partial

Keras

API: β€”
Strategy: Custom / Partial

TensorFlow

API: tf.device
Strategy: Direct Mapping

Apple MLX

API: mlx.core.stream
Strategy: Direct Mapping

Flax NNX

API: β€”
Strategy: Custom / Partial

PaxML / Praxis

API: β€”
Strategy: Custom / Partial