BFloat16Tensor

Constructs a Tensor with bfloat16 data type.

Abstract Signature:

BFloat16Tensor(data)

PyTorch

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

JAX (Core)

API: jax.numpy.array
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.convert_to_tensor
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.convert_to_tensor
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

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

Flax NNX

API: jax.numpy.array
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.array
Strategy: Direct Mapping
Official Docs ↗