What is max

Asked on 2024-08-01

1 search

The maximum floating-point precision supported in the JAX Metal framework is the Pfloat 16 data type. This data type represents a wide dynamic range of floating point values and is suitable for use cases like mixed precision training. You can use the new data type just like any other data type in JAX.

For more details, you can refer to the session Train your machine learning and AI models on Apple GPUs.