search

dpiras / cosmopower-jax

Differentiable cosmological emulators: the JAX version of CosmoPower
GNU General Public License v3.0
30 stars 3 forks source link

Make CosmoPower-JAX work with `TF>= 2.14` #4

Closed dpiras closed 7 months ago

dpiras commented 7 months ago

As reported already, TF stopped supporting pickle with TF>= 2.14, which means that object serialisation is no longer possible and one cannot upload a pre-trained model from CP if using TF>= 2.14. Can we make CPJ work with TF>=2.14 and upload a custom model?

dpiras commented 7 months ago

Should be solved by https://github.com/dpiras/cosmopower-jax/pull/5.