Add JAX 0.9.0 to the unittest matrix (#166)

Author: wstcliyu

.github/workflows/test.yml

@@ -24,7 +24,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ['3.11', '3.12', '3.13', '3.14']
-        jax-version: ['0.6.2', '0.7.0', '0.7.1', '0.7.2', '0.8.0', '0.8.1', '0.8.2', 'nightly']
+        jax-version: ['0.6.2', '0.7.0', '0.7.1', '0.7.2', '0.8.0', '0.8.1', '0.8.2', '0.9.0', 'nightly']
 
         exclude:
           - python-version: '3.14'