Enable f8.
Enable f8.

I decided to port the conversion function from ml_dtypes (with one fix
that's not yet submitted there) instead of the legacy XLA logic, since
the latter isn't generic and quite a bit more code.

Tested with convert_test_gpu, float8_test_gpu, JAX test suite.
