FEAT: configure backend precision - #578
Conversation
* BEHAVIOR: stop enabling TensorFlow numpy behavior in `TFUniformRealNumberGenerator` * DOC: describe when backend precision takes effect * FIX: configure precision for `tuple` and `dict` backends * FIX: read the value of `JAX_ENABLE_X64` instead of testing for the key * FIX: validate all `configure()` arguments before applying any of them * MAINT: annotate `raise_missing_module_error()` as `NoReturn` * MAINT: confine benchmark precision configuration to the JAX fixtures * MAINT: move configuration tests to `tests/test_config.py`
|
Verified downstream with TensorWaves at 2c97d7c: the default is still 64-bit. JAX initially reports I also ran six affected |
✨ New features
tensorwaves.configure()API for independently selecting 32-bit or 64-bit precision in JAX and TensorFlow without eagerly importing optional backends.JAX_ENABLE_X64unless an explicit JAX precision is configured.🐛 Bug fixes
📝 Documentation
Squash commit messages