Chris Carmona comments

Repositories
Issues
Comments

Results 2 comments of


                                            Chris Carmona

Not compatible with `tensorflow-macos`

I'm experiencing a similar incompatibility between tensorflow and distrax. In a clean environment, I installed jax, distrax and tensorflow ```bash pip install -U pip pip install -U "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html...

Incompatibility with Haiku

Hi @liangyaning33, Apologies for the delayed response. I observe the same behaviour as @Carbon225, The error still appears in v0.4.1 (as currently installed by `pip install -U orbax-checkpoint`). It works...