PyTorch
API: torch.double
Strategy: Plugin (type_methods)
JAX (Core)
API: —
Strategy: Plugin (type_methods)
NumPy
API: numpy.double
Strategy: Plugin (type_methods)
Keras
API: —
Strategy: Plugin (type_methods)
TensorFlow
API: —
Strategy: Plugin (type_methods)
Apple MLX
API: —
Strategy: Plugin (type_methods)
Flax NNX
API: —
Strategy: Plugin (type_methods)
PaxML / Praxis
API: —
Strategy: Plugin (type_methods)