INNER CODE UNIT · Python

Backend

bayesflow-org/bayesflow · bayesflow/__init__.py:22

        class Backend:
            def __init__(self, display_name, package_name, env_name, install_url, priority):
                self.display_name = display_name
                self.package_name = package_name
                self.env_name = env_name
                self.install_url = install_url
                self.priority = priority

        backends = [
            Backend("JAX", "jax", "jax", "https://docs.jax.dev/en/latest/quickstart.html#installation", 0),
            Backend("PyTorch", "torch", "torch", "https://pytorch.org/get-started/locally/", 1),
            Backend("TensorFlow", "tensorflow", "tensorflow", "https://www.tensorflow.org/install", 2),
        ]

        found_backends = []
        for backend in backends:
            if find_spec(backend.package_name) is not None:
                found_backends.append(backend)

View source record →