search

dpiras / cosmopower-jax

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

Make `TF>=2.14` possible #5

Closed dpiras closed 6 months ago

dpiras commented 6 months ago

If you need to use TF>=2.14, you can now run the convert_tf214.py script once on your .pkl files from e.g. CosmoPower runs, run pip install . and you should be good to go. This is because the .pkl files get converted to .npz dictionaries and therefore do not rely on TensorFlow anymore.

dpiras commented 6 months ago

Also, solves https://github.com/dpiras/cosmopower-jax/issues/4.