This release allows a user to set PYMBAR_DISABLE_JAX=TRUE to disable JAX acceleration.
This enables disabling JAX at runtime.
Previously, the only way to control JAX was to install or uninstall the jax package, which made it difficult for users to actually disable JAX since some packages depended on the pymbar package which on conda-forge installs the jax package.
See the README.md for more details.
We will be improving accelerator selection in future releases.
conda-forge Package Notes
Previous versions of pymbar pinned numpy < 2.3 due to a change in numpy 2.3 where glibc implementations of some math functions are used where previously numpy's built in implementations were used. See #556 and alchemistry/alchemlyb#431 (review) for some more details.
However, this pin prevents pymbar from working with python 3.14 since there are no numpy <2.3 builds that work with python 3.14. So we have decided to relax the numpy pin. Within our own testing framework we have not observed any differences between numpy version. While there are numerical differences, for our tests, those differences are within our tolerance. You can still install this version of pymbar with an older numpy version, it just won't work with python 3.14. For example conda install -c conda-forge "pymbar==4.2.0" "numpy<2.3" "python=3.13" will work.
Changelog for robots 👇
What's Changed
- switch from master to main by @mikemhenry in #567
- update README to reflect change in default branch name by @mikemhenry in #568
- disable jax if PYMBAR_DISABLE_JAX is set by @mikemhenry in #569
Full Changelog: 4.1.0...4.2.0
