AsyncEval

Asynchronously evaluate an array or tree of arrays.

Abstract Signature:

AsyncEval(tensors: Tensor)

PyTorchJAX (Core)Apple MLX

PyTorch

API: torch.cuda.synchronize

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '[t.block_until_ready() for t in {tensors}]'

Apple MLX

API: mlx.core.async_eval

Strategy: Direct Mapping

Official Docs ↗