shuhand0

It is reproducible. The jitted module is incorrectly optimized and we will look into the fix.

chlo.lgmma is not supported. We will look into add the support.

Based on the packages, it is AMD GPU? Could you try a venv with python=3.9?

Haven't been able to reproduce the issue. The below config shows an installation and verification result: ProductName: macOS ProductVersion: 14.4 ``` The following NEW packages will be INSTALLED: ca-certificates pkgs/main/osx-64::ca-certificates-2023.12.12-hecd8cb5_0...

The issue is not reproducible. Do you still see the same problem with the latest OS 14.4 and jax-metal 0.0.6?

While we are looking into the issue, FP64 support will NOT be there for sometime.

The issue should be fixed in jax-metal 0.0.7. Pls reopen it if otherwise.

Conversion pattern for chlo.lgmma has been added into jax-metal 0.0.7. Reopen the issue if otherwise.

Bfloat16 support has been added since jax-metal 0.0.7.

We are aware of the issue and working on a fix.