runtime error
ackages/jax/_src/core.py", line 816, in process_primitive return primitive.impl(*tracers, **params) File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/dispatch.py", line 117, in apply_primitive compiled_fun = xla_primitive_callable(prim, *unsafe_map(arg_spec, args), File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/util.py", line 253, in wrapper return cached(config._trace_context(), *args, **kwargs) File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/util.py", line 246, in cached return f(*args, **kwargs) File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/dispatch.py", line 208, in xla_primitive_callable compiled = _xla_callable_uncached(lu.wrap_init(prim_fun), prim.name, File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/dispatch.py", line 254, in _xla_callable_uncached return computation.compile(_allow_propagation_to_outputs=allow_prop).unsafe_call File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/interpreters/pxla.py", line 2836, in compile self._executable = UnloadedMeshExecutable.from_hlo( File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/interpreters/pxla.py", line 3048, in from_hlo xla_executable = dispatch.compile_or_get_cached( File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/dispatch.py", line 526, in compile_or_get_cached return backend_compile(backend, serialized_computation, compile_options, File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/profiler.py", line 314, in wrapper return func(*args, **kwargs) File "/home/user/.pyenv/versions/3.8.9/lib/python3.8/site-packages/jax/_src/dispatch.py", line 471, in backend_compile return backend.compile(built_c, compile_options=options) jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed. Look at the errors above for more details.
Container logs:
Fetching error logs...