jax-ml / ml_dtypes

A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Apache License 2.0
191 stars 25 forks source link

Add FP8 support to the exhaustive tests #177

Open copybara-service[bot] opened 2 weeks ago

copybara-service[bot] commented 2 weeks ago

Add FP8 support to the exhaustive tests

Adds new tests for two FP8 variants to both unary and binary exhaustive tests.