Import Codex

[eyyuptasci]

When I import the library as suggested in the README file, I encounter an error when attempting to use any function from the ops.gradient file. The error indicates that the file does not exist. This issue is resolved when I import the library as follows:

pip install 'jax-codex @ git+https://github.com/google/codex'

[ssjhv]

Hi,

Sorry for the inconvenience. It is because we have not officially released newer versions. ops.gradients was probably not there when the initial version was released. We are working towards the next version release. Meanwhile, please import the library from GitHub directly if you wish to import recent changes.