meridian.backend.config.set_backend

Sets the backend for Meridian.

meridian.backend.config.set_backend(
    backend: Union[meridian.backend.config.Backend, str]
) -> None

Warning: This function should ideally be called at the beginning of your program, before any other Meridian modules are imported or used.

Changing the backend after Meridian's functions or classes have been imported can lead to unpredictable behavior. This is because already-imported modules will not reflect the backend change.

Changing the backend at runtime requires reloading the meridian.backend module for the changes to take effect globally.

backend The backend to use, must be a member of the Backend enum or a valid string ('tensorflow', 'jax').

ValueError If the provided backend is not valid.