TensorType

Auto-generated from jax_code_defs

PyTorchJAX (Core)NumPyTensorFlowApple MLXFlax NNX

PyTorch

API: torch.Tensor

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 ↗