config

Contents

config#

import tensorwaves.config

Configure optional computational backends.

configure(*, jax_precision: Literal['float32', 'float64'] | None = None, tensorflow_precision: Literal['float32', 'float64'] | None = None) → None[source]#

Set the precision used by computational backends.

Call this function before creating backend arrays or TensorWaves functions. By default, TensorWaves uses 64-bit precision for JAX and TensorFlow. TensorWaves respects JAX_ENABLE_X64 if jax_precision is not specified.

Contents