TensorType

Type representation of a Tensor.

PyTorch

API: torch.TensorType
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.Array
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.ndarray
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tensorflow.Tensor
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.array
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.Array
Strategy: Direct Mapping
Official Docs ↗