Does JAX support Ada / Hopper GPUs?

[JyK-Furiosa]

Hello,

As the title of the discussion says, I would like to know whether JAX fully supports new Ada Lovelace / Hopper GPUs.

More specifically, I would like to know whether JAX fully supports cuda 11.8 or above, which is required for GPU architectures mentioned above.

Thank you