Mutex issue when constructing gemma model object

I'm running into a mutex issue when constructing the gemma model object:
```
libc++abi: terminating due to uncaught exception of type std::__1::system_error: mutex lock failed: Invalid argument
Abort trap: 6
```

Reproduce:
```
from gemma import gm

model = gm.nn.Gemma3_4B()
params = gm.ckpts.load_params(gm.ckpts.CheckpointPath.GEMMA3_4B_IT)
```

Spec:
- Mac M3 Pro with JAX installed for CPU
- Python 3.13.7

How can I provide additional information to help debug this setup issue?
Thanks!