Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* As of jax and jaxlib v0.4.20 accessing jax.config from the jax.config submodule is deprecated and it should be accessed from the jax top level API instead. ``` >>> from jax.config import config <stdin>:1: DeprecationWarning: Accessing jax.config via the jax.config submodule is deprecated ```
- Loading branch information