Long checking and analyzing operations when using JAX

Problem
When using  JAX PyLance takes very long to analyze a file and one typically has to wait several seconds until suggestions are available.
Potentially related to #2098.
Environment data
What might be related: I develop on a server via SSH where the directory is mounted via nfs. Though, the network interface shouldn't be a bottleneck nor the compute of the server.

[Info  - 10:35:52 AM] (2403834) No pyproject.toml file found.
[Info  - 10:35:52 AM] (2403834) Setting pythonPath for service "pesnet": "/nfs/staff-ssd/gaoni/miniconda3/envs/pesnet/bin/python"
[Warn  - 10:35:52 AM] (2403834) stubPath /nfs/homedirs/gaoni/repos/pesnet/typings is not a valid directory.
[Info  - 10:35:52 AM] (2403834) Assuming Python version 3.8
[Info  - 10:35:52 AM] (2403834) Assuming Python platform Linux
[Info  - 10:35:52 AM] (2403834) Searching for source files
[Info  - 10:35:56 AM] (2403834) Found 86 source files
[Info  - 10:35:56 AM] (2403834) Background analysis(1) root directory: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.10/dist
[Info  - 10:35:56 AM] (2403834) Background analysis(1) started
[Info  - 10:36:05 AM] (2403834) [BG(1)] Long operation: checking: /nfs/homedirs/gaoni/repos/pesnet/train.py (8252ms)
[Info  - 10:36:05 AM] (2403834) [BG(1)] Long operation: analyzing: /nfs/homedirs/gaoni/repos/pesnet/train.py (8637ms)
[Info  - 10:36:12 AM] (2403834) [BG(1)] Long operation: checking: /nfs/homedirs/gaoni/repos/pesnet/train.py (2195ms)
[Info  - 10:36:12 AM] (2403834) [BG(1)] Long operation: analyzing: /nfs/homedirs/gaoni/repos/pesnet/train.py (2213ms)

thanks for the logs.
can you turn on Trace? see https://github.com/microsoft/pylance-release/blob/main/TROUBLESHOOTING.md#filing-an-issue
thank you.

Here is a more complete log:

(1308596) [IDX(FG)] index libraries /nfs/homedirs/gaoni/repos/planet (refresh) ...
(1308596) [IDX(FG)]   read stdlib indices (43ms)
(1308596) [IDX(FG)] index libraries /nfs/homedirs/gaoni/repos/planet (refresh) [succeed] (44ms)
(1308596) Background analysis message: invalidateAndForceReanalysis
(1308596) Background analysis message: analyze
(1308596) [BG(1)] analyzing: /nfs/homedirs/gaoni/repos/planet/train.py ...
(1308596) [BG(1)]   parsing: /nfs/homedirs/gaoni/repos/planet/train.py (301ms)
(1308596) [BG(1)]   parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/builtins.pyi [fs read 0ms] (105ms)
(1308596) [BG(1)]   binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/builtins.pyi (21ms)
(1308596) [BG(1)]   binding: /nfs/homedirs/gaoni/repos/planet/train.py (3ms)
(1308596) [BG(1)]   checking: /nfs/homedirs/gaoni/repos/planet/train.py ...
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/os/__init__.pyi [fs read 1ms] (43ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/os/__init__.pyi (11ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/collections/abc.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/collections/abc.pyi ...
(1308596) [BG(1)]       parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_collections_abc.pyi [fs read 2ms] (2ms)
(1308596) [BG(1)]       binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_collections_abc.pyi (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/collections/abc.pyi (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/typing.pyi [fs read 2ms] (17ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/typing.pyi (8ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/typing_extensions.pyi [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/typing_extensions.pyi (5ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/abc.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/abc.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/functools.pyi [fs read 0ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/functools.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/logging/__init__.pyi [fs read 1ms] (16ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/logging/__init__.pyi (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/__init__.py [fs read 0ms] (17ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/numpy/__init__.py [fs read 0ms] (14ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/numpy/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs/matplotlib/pyplot.pyi [fs read 2ms] (39ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs/matplotlib/pyplot.pyi ...
(1308596) [BG(1)]       parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs/matplotlib/_typing.pyi [fs read 1ms] (10ms)
(1308596) [BG(1)]       binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs/matplotlib/_typing.pyi (0ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs/matplotlib/pyplot.pyi (15ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/__init__.pyi [fs read 1ms] (111ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/__init__.pyi (29ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/seml/seml/__init__.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml/seml/__init__.py ...
(1308596) [BG(1)]       parsing: /nfs/staff-ssd/gaoni/repos/seml/seml/experiment.py [fs read 2ms] (8ms)
[Info  - 7:31:57 PM] (1308596) Could not import 'torch' in file '/nfs/staff-ssd/gaoni/repos/seml/seml/experiment.py'
[Info  - 7:31:57 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
[Info  - 7:31:57 PM] (1308596) Could not import 'tensorflow' in file '/nfs/staff-ssd/gaoni/repos/seml/seml/experiment.py'
[Info  - 7:31:57 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]       binding: /nfs/staff-ssd/gaoni/repos/seml/seml/experiment.py (1ms)
(1308596) [BG(1)]       parsing: /nfs/staff-ssd/gaoni/repos/seml/seml/observers.py [fs read 2ms] (26ms)
[Info  - 7:31:57 PM] (1308596) Could not import 'neptunecontrib.monitoring.sacred' in file '/nfs/staff-ssd/gaoni/repos/seml/seml/observers.py'
[Info  - 7:31:57 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]       binding: /nfs/staff-ssd/gaoni/repos/seml/seml/observers.py (2ms)
(1308596) [BG(1)]       parsing: /nfs/staff-ssd/gaoni/repos/seml/seml/evaluation.py [fs read 1ms] (8ms)
(1308596) [BG(1)]       binding: /nfs/staff-ssd/gaoni/repos/seml/seml/evaluation.py (0ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml/seml/__init__.py (45ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/__init__.py [fs read 2ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/__init__.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/experiment.py [fs read 1ms] (27ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/experiment.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/ingredient.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/ingredient.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/logger.py [fs read 1ms] (12ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/logger.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/auto.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/auto.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/__init__.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/nn/__init__.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/nn/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/systems/__init__.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/systems/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/systems/collection.py [fs read 0ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/systems/collection.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/systems/constants.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/systems/constants.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/systems/scf.py [fs read 0ms] (16ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/systems/scf.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/trainer.py [fs read 1ms] (19ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/trainer.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/utils/__init__.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/utils/__init__.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/utils/jax_utils.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/utils/jax_utils.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/api.py [fs read 1ms] (52ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/api.py (9ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/__init__.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/traceback_util.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/traceback_util.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_typeshed/__init__.pyi [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_typeshed/__init__.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lib/__init__.py [fs read 0ms] (25ms)
[Info  - 7:31:57 PM] (1308596) Could not import 'jaxlib.pocketfft' in file '/nfs/staff-ssd/gaoni/repos/jax/jax/_src/lib/__init__.py'
[Info  - 7:31:57 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:57 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:57 PM] (1308596)   Resolved import with file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/__init__.py'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:57 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:57 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:31:57 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:31:57 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lib/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_client.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_client.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/__init__.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/__init__.pyi [fs read 0ms] (12ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/__init__.pyi (6ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/jax_jit.pyi [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/jax_jit.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/config.py [fs read 1ms] (23ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/config.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/threading.pyi [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/threading.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/array.pyi [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/array.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/mmap.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/mmap.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/ctypes/__init__.pyi [fs read 0ms] (20ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/ctypes/__init__.pyi (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/pickle.pyi [fs read 0ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/pickle.pyi (7ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/__init__.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/__init__.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/__init__.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_flagvalues.py [fs read 1ms] (42ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_flagvalues.py (14ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_defines.py [fs read 0ms] (22ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_defines.py (24ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_helpers.py [fs read 1ms] (21ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_helpers.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/sys.pyi [fs read 0ms] (16ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/sys.pyi (5ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/re.pyi [fs read 0ms] (15ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/re.pyi (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/enum.pyi [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/enum.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/types.pyi [fs read 0ms] (20ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/types.pyi (6ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_validators.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_validators.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_validators_classes.py [fs read 0ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_validators_classes.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/warnings.pyi [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/warnings.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_warnings.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_warnings.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_flag.py [fs read 0ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_flag.py (18ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_argument_parser.py [fs read 0ms] (15ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/absl/flags/_argument_parser.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/__init__.py [fs read 0ms] (8ms)
[Info  - 7:31:57 PM] (1308596) Could not resolve source for '._ufunc' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/__init__.py'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve relative import
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing'
[Info  - 7:31:57 PM] (1308596)   Resolved import with file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_ufunc.pyi'
[Info  - 7:31:57 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing'
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/overrides.py [fs read 0ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/overrides.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_nbit.py [fs read 0ms] (0ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_nbit.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_char_codes.py [fs read 0ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_char_codes.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_dtype_like.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_dtype_like.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_generic_alias.py [fs read 0ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_generic_alias.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_shape.py [fs read 0ms] (0ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_shape.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/eval.py [fs read 6ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/eval.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/lax_numpy.py [fs read 2ms] (106ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/lax_numpy.py (31ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/ndarray.py [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/ndarray.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/training.py [fs read 1ms] (14ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/training.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/stages.py [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/stages.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/stages.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/stages.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/utils.py [fs read 0ms] (16ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/utils.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/pathlib.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/pathlib.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/host_info.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/host_info.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/commandline_options.py [fs read 2ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/commandline_options.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/__init__.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_dict.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_dict.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_files.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_files.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/os/path.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/os/path.pyi ...
(1308596) [BG(1)]       parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/posixpath.pyi [fs read 2ms] (5ms)
(1308596) [BG(1)]       binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/posixpath.pyi (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/os/path.pyi (6ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/genericpath.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/genericpath.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/serializer.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/serializer.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/__init__.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/__init__.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/decoder.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/decoder.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/__init__.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/ext/pandas.py [fs read 1ms] (15ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/ext/pandas.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/handlers.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/handlers.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/util.py [fs read 1ms] (10ms)
[Info  - 7:31:58 PM] (1308596) Could not import '__builtin__' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/util.py'
[Info  - 7:31:58 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:31:58 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:31:58 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:31:58 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:31:58 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:58 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:31:58 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:31:58 PM] (1308596)   Typeshed path not found
[Info  - 7:31:58 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:31:58 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:31:58 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/util.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/ext/numpy.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/ext/numpy.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/zlib.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/zlib.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/backend.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/backend.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/unpickler.py [fs read 1ms] (10ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/unpickler.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/base64.pyi [fs read 2ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/base64.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/tags.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/tags.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/compat.py [fs read 2ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/compat.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/quopri.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/quopri.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/encoder.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/json/encoder.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/pickler.py [fs read 2ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jsonpickle/pickler.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/decimal.pyi [fs read 0ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/decimal.pyi ...
(1308596) [BG(1)]       parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_decimal.pyi [fs read 0ms] (4ms)
(1308596) [BG(1)]       binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_decimal.pyi (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/decimal.pyi (7ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/io.pyi [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/io.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_scope.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/config_scope.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/tree_util.py [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/tree_util.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/tree_util.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/tree_util.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lib/xla_bridge.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lib/xla_bridge.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/lax/__init__.py [fs read 0ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/lax/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/lax.py [fs read 1ms] (85ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/lax.py (32ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/nn/pesnet.py [fs read 1ms] (18ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/nn/pesnet.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/collections/__init__.pyi [fs read 0ms] (12ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/collections/__init__.pyi (5ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/random.py [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/random.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/random.py [fs read 1ms] (26ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/random.py (9ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/prng.py [fs read 1ms] (29ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/prng.py (10ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/operator.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/operator.pyi ...
(1308596) [BG(1)]       parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_operator.pyi [fs read 1ms] (3ms)
(1308596) [BG(1)]       binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_operator.pyi (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/operator.pyi (5ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/function_base.pyi [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/function_base.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_array_like.py [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_array_like.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_nested_sequence.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_nested_sequence.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/utils.py [fs read 2ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml_logger/seml_logger/utils.py (5ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/seml/seml/utils.py [fs read 2ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/seml/seml/utils.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/datetime.pyi [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/datetime.pyi (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/pretrain.py [fs read 0ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/pretrain.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/util.py [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/util.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/stride_tricks.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/stride_tricks.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/multiarray.pyi [fs read 0ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/multiarray.pyi (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/__init__.py [fs read 1ms] (128ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/combine.py [fs read 2ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/combine.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/base.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/base.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/transform.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/transform.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/wrappers.py [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/wrappers.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/chex/__init__.py [fs read 1ms] (17ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/chex/__init__.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/chex/_src/pytypes.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/chex/_src/pytypes.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/clipping.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/clipping.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/surrogate/training.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/surrogate/training.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/numeric.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/numeric.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/alias.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/optax/_src/alias.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/hamiltonian.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/hamiltonian.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/shape_base.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/shape_base.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/mcmc.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/mcmc.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/nn/dimenet.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/nn/dimenet.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/linen/__init__.py [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/linen/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/nn/initializers.py [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/nn/initializers.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/nn/initializers.py [fs read 0ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/nn/initializers.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/linen/module.py [fs read 0ms] (33ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/linen/module.py (8ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/weakref.pyi [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/weakref.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/dataclasses.pyi [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/dataclasses.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/__init__.py [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/traceback_util.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/traceback_util.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/core/scope.py [fs read 1ms] (13ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/core/scope.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/core/frozen_dict.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/flax/core/frozen_dict.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/core.py [fs read 1ms] (59ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/core.py (26ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/__init__.py [fs read 0ms] (0ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/dtypes.py [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/dtypes.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/utils/optim.py [fs read 1ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/utils/optim.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/__init__.py [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/conditionals.py [fs read 1ms] (16ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/conditionals.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/fromnumeric.pyi [fs read 0ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/core/fromnumeric.pyi (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/util.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/util.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/common.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/control_flow/common.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/pytree.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/jaxlib/xla_extension/pytree.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/__init__.py [fs read 0ms] (0ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/partial_eval.py [fs read 1ms] (54ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/partial_eval.py (27ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/atexit.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/atexit.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/config.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/config.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/api_util.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/api_util.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/api_util.py [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/api_util.py (9ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/linear_util.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/linear_util.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/profiler.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/profiler.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/contextlib.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/contextlib.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/source_info_util.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/source_info_util.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/itertools.pyi [fs read 0ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/itertools.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/inspect.pyi [fs read 0ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/inspect.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/gzip.pyi [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/gzip.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_compression.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_compression.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/device_array.py [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/device_array.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/experimental/array.py [fs read 0ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/experimental/array.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/mlir.py [fs read 0ms] (23ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/mlir.py (10ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/batching.py [fs read 1ms] (13ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/batching.py (9ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/utils.py [fs read 0ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/utils.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/ad.py [fs read 0ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/ad.py (8ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/xla.py [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/xla.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/shape_base.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/shape_base.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/experimental/__init__.py [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/experimental/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/pxla.py [fs read 1ms] (51ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/interpreters/pxla.py (23ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/npyio.pyi [fs read 2ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/npyio.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/custom_derivatives.py [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/custom_derivatives.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/custom_derivatives.py [fs read 1ms] (12ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/custom_derivatives.py (6ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/custom_api_util.py [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/custom_api_util.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/parallel.py [fs read 1ms] (18ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/lax/parallel.py (9ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/reductions.py [fs read 1ms] (14ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/reductions.py (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/repos/planet/planet/vmc/damping.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/repos/planet/planet/vmc/damping.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/asyncio.pyi [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/asyncio.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/std.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/std.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/utils.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/tqdm/tqdm/utils.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_scalars.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_scalars.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/ufuncs.py [fs read 1ms] (9ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/repos/jax/jax/_src/numpy/ufuncs.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_ufunc.pyi [fs read 0ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/_typing/_ufunc.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/textwrap.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/textwrap.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/function_base.pyi [fs read 0ms] (6ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/lib/function_base.pyi (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/typing/__init__.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/numpy/typing/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/copy.pyi [fs read 0ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/copy.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/__init__.py [fs read 2ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/decorators.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/decorators.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/wrappers.py [fs read 1ms] (7ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/wrappers.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/arguments.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/wrapt/arguments.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/captured_function.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/captured_function.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/signature.py [fs read 1ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/config/signature.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/run.py [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/sacred/run.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/merge_args.py [fs read 2ms] (3ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/merge_args.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/__init__.py [fs read 1ms] (13ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/__init__.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/files.py [fs read 2ms] (11ms)
[Info  - 7:32:02 PM] (1308596) Could not import 'mpi4py.MPI' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/files.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/files.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/group.py [fs read 1ms] (13ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/group.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/base.py [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/h5py/_hl/base.py (2ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/__init__.py [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py [fs read 1ms] (17ms)
[Info  - 7:32:02 PM] (1308596) Could not import 'torch' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596) Could not import 'torch' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596) Could not import 'torch.utils.tensorboard._pytorch_graph' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596) Could not import 'torch' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596) Could not import 'caffe2.proto' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596) Could not import 'caffe2.python' in file '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py'
[Info  - 7:32:02 PM] (1308596)   Looking in stubPath '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet/typings'
[Info  - 7:32:02 PM] (1308596)   Looking in root directory of execution environment '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/lib-dynload'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/planet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/jax'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mol_baselines'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/mabenet'
[Info  - 7:32:02 PM] (1308596)   Looking in python search path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/staff-ssd/gaoni/repos/seml_logger'
[Info  - 7:32:02 PM] (1308596)   Looking in bundled stubs path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve stub package using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/bundled/stubs'
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stdlib path
[Info  - 7:32:02 PM] (1308596)   Attempting to resolve using root path '/nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib'
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed third-party path
[Info  - 7:32:02 PM] (1308596)   Looking for typeshed stubs path
[Info  - 7:32:02 PM] (1308596)   Typeshed path not found
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/writer.py (4ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/socket.pyi [fs read 1ms] (8ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/socket.pyi (3ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_socket.pyi [fs read 1ms] (4ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/_socket.pyi (4ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/event_file_writer.py [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/event_file_writer.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/time.pyi [fs read 1ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stdlib/time.pyi (1ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/proto/__init__.py [fs read 2ms] (2ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/proto/__init__.py (0ms)
(1308596) [BG(1)]     parsing: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/proto/event_pb2.py [fs read 2ms] (11ms)
(1308596) [BG(1)]     binding: /nfs/staff-ssd/gaoni/miniconda3/envs/jax0317/lib/python3.9/site-packages/tensorboardX/proto/event_pb2.py (1ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/__init__.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/__init__.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/reflection.pyi [fs read 1ms] (1ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/reflection.pyi (0ms)
(1308596) [BG(1)]     parsing: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/symbol_database.pyi [fs read 1ms] (5ms)
(1308596) [BG(1)]     binding: /nfs/homedirs/gaoni/.vscode-server/extensions/ms-python.vscode-pylance-2022.10.20/dist/typeshed-fallback/stubs/protobuf/google/protobuf/symbol_database.pyi (0ms)