[BUG] Conda installation issues

[junminchen]

Bug summary

After using conda install dmff, which installs JAX version 0.4.28, the issue occurs. (DMFF is supposed to support JAX versions under 0.4.24.)

ImportError: cannot import name 'config' from 'jax.config' (/global/home/users/jmchen24/miniconda3/envs/py39/lib/python3.9/site-packages/jax/config.py)

should it be fixed in conda installlation? or update DMFF settings.py?

DMFF Version

1.0.0

JAX Version

0.4.28

OpenMM Version

8.2.0

How did you download the software?

Others (write below)

Input Files, Running Commands, Error Log, etc.

conda

Steps to Reproduce

conda install dmff

import dmff

Further Information, Files, and Links

No response