This reverts commit 57427f84fe5fdda71aef4be257ed28d7b4f55d05. For some reason mlir-nvidia CI is failing to import `float8_e3m4` from `ml_dtypes`. See https://lab.llvm.org/buildbot/#/builders/138/builds/27095.
This reverts commit 57427f84fe5fdda71aef4be257ed28d7b4f55d05. For some reason mlir-nvidia CI is failing to import `float8_e3m4` from `ml_dtypes`. See https://lab.llvm.org/buildbot/#/builders/138/builds/27095.