INNER CODE UNIT · Python
__init__
bayesflow-org/bayesflow · bayesflow/__init__.py:23
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)