Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions .github/workflows/lint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,11 +27,11 @@ jobs:
with:
path: autolens_profiling

- name: Checkout PyAutoConf
- name: Checkout PyAutoNerves
uses: actions/checkout@v4
with:
repository: PyAutoLabs/PyAutoConf
path: PyAutoConf
repository: PyAutoLabs/PyAutoNerves
path: PyAutoNerves
- name: Checkout PyAutoFit
uses: actions/checkout@v4
with:
Expand Down Expand Up @@ -66,7 +66,7 @@ jobs:
# (pyyaml, astropy, jax, numba, ...) without hand-maintaining a
# package list. PYTHONPATH below still resolves the PyAuto* imports
# from source, so the checkouts stay authoritative.
for r in PyAutoConf PyAutoFit PyAutoArray PyAutoGalaxy PyAutoLens; do
for r in PyAutoNerves PyAutoFit PyAutoArray PyAutoGalaxy PyAutoLens; do
pip install "./$r[optional]"
done

Expand Down Expand Up @@ -111,7 +111,7 @@ jobs:
AUTOLENS_PROFILING_SMOKE: "1"
NUMBA_CACHE_DIR: /tmp/numba_cache
MPLCONFIGDIR: /tmp/matplotlib
PYTHONPATH: ${{ github.workspace }}/PyAutoConf:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
PYTHONPATH: ${{ github.workspace }}/PyAutoNerves:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
run: |
# Each script reads AUTOLENS_PROFILING_SMOKE at module top and
# exits 0 immediately after the import + setup section. Catches
Expand Down
12 changes: 6 additions & 6 deletions .github/workflows/profile.yml
Original file line number Diff line number Diff line change
Expand Up @@ -54,11 +54,11 @@ jobs:
path: autolens_profiling
token: ${{ secrets.GITHUB_TOKEN }}

- name: Checkout PyAutoConf
- name: Checkout PyAutoNerves
uses: actions/checkout@v4
with:
repository: PyAutoLabs/PyAutoConf
path: PyAutoConf
repository: PyAutoLabs/PyAutoNerves
path: PyAutoNerves
- name: Checkout PyAutoFit
uses: actions/checkout@v4
with:
Expand Down Expand Up @@ -99,7 +99,7 @@ jobs:
env:
NUMBA_CACHE_DIR: /tmp/numba_cache
MPLCONFIGDIR: /tmp/matplotlib
PYTHONPATH: ${{ github.workspace }}/PyAutoConf:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
PYTHONPATH: ${{ github.workspace }}/PyAutoNerves:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
run: |
set +e # don't abort the whole step on one script's failure
for script in likelihood_runtime/imaging/*.py likelihood_runtime/interferometer/*.py \
Expand All @@ -116,7 +116,7 @@ jobs:
env:
NUMBA_CACHE_DIR: /tmp/numba_cache
MPLCONFIGDIR: /tmp/matplotlib
PYTHONPATH: ${{ github.workspace }}/PyAutoConf:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
PYTHONPATH: ${{ github.workspace }}/PyAutoNerves:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
run: |
set +e
for script in simulators/imaging.py simulators/interferometer.py \
Expand All @@ -136,7 +136,7 @@ jobs:
env:
NUMBA_CACHE_DIR: /tmp/numba_cache
MPLCONFIGDIR: /tmp/matplotlib
PYTHONPATH: ${{ github.workspace }}/PyAutoConf:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
PYTHONPATH: ${{ github.workspace }}/PyAutoNerves:${{ github.workspace }}/PyAutoFit:${{ github.workspace }}/PyAutoArray:${{ github.workspace }}/PyAutoGalaxy:${{ github.workspace }}/PyAutoLens
run: |
set +e
# Nautilus at n_live=200 on CPU takes O(minutes) to O(hours) per
Expand Down
4 changes: 2 additions & 2 deletions AGENTS.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ auto-simulate-if-missing hook, so per-script boilerplate stays minimal.

The PyAuto* libraries are **not pip-installed** here — they are resolved from sibling source
checkouts via `PYTHONPATH`. On the HPC, `source activate.sh` activates the shared venv (third-party
deps only) and points `PYTHONPATH` at the canonical `PyAutoConf`/`PyAutoFit`/`PyAutoArray`/
deps only) and points `PYTHONPATH` at the canonical `PyAutoNerves`/`PyAutoFit`/`PyAutoArray`/
`PyAutoGalaxy`/`PyAutoLens` checkouts; `HPCPullPyAuto` is then the whole library-update story.

JAX convention (mirrors `autolens_workspace_developer`): pass `xp=jnp` through PyAuto* functions to
Expand Down Expand Up @@ -87,7 +87,7 @@ write from a header skim silently deletes every section below the header.
## Related Repos

- `../PyAutoLens` — the library being profiled (plus `../PyAutoGalaxy`, `../PyAutoArray`,
`../PyAutoFit`, `../PyAutoConf` on `PYTHONPATH`).
`../PyAutoFit`, `../PyAutoNerves` on `PYTHONPATH`).
- `../autolens_workspace` — user-facing science scripts and tutorials.
- `../PyAutoHands` — build/CI tooling.

Expand Down
2 changes: 1 addition & 1 deletion activate.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ BASE=/mnt/ral/jnightin/PyAuto
source "$BASE/PyAuto/bin/activate"

export PYTHONPATH=$BASE:\
$BASE/PyAutoConf:\
$BASE/PyAutoNerves:\
$BASE/PyAutoFit:\
$BASE/PyAutoArray:\
$BASE/PyAutoGalaxy:\
Expand Down
4 changes: 2 additions & 2 deletions jax_compile/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ original tags for provenance.
- ~~Autotuning ruled out (2026-07-15)~~ **downgraded to unproven 2026-07-17**:
the flag never took effect — `autoconf/jax_wrapper.py` overwrote `XLA_FLAGS`
(see Verdict item 3); "identical to the decimal" is exactly what clobbering
produces. Re-test after PyAutoConf#127 if autotune ever matters again.
produces. Re-test after PyAutoNerves#127 if autotune ever matters again.
- Fresh-closure-per-call JIT cache-busting is a known stack trap (cache the
jitted closure on the instance).
- `analysis.print_vram_use()` triggers a full vmapped compile (not a cheap
Expand Down Expand Up @@ -133,7 +133,7 @@ one-time-per-machine cost the cache already removes. **Do not restructure.**
3. **Upstream**: the 7m30 single-fusion compile is XLA-report material.
CORRECTED 2026-07-17: `--xla_dump_to` is not inert — `autoconf/jax_wrapper.py`
was *overwriting* `XLA_FLAGS` at import, silently discarding user/job flags
(fixed in PyAutoConf#127). Two consequences: (a) the HLO dump just needs a
(fixed in PyAutoNerves#127). Two consequences: (a) the HLO dump just needs a
re-run once that fix lands (or `XLA_FLAGS` including the constant_folding
disable so the wrapper leaves it alone); (b) the historical 2026-07-15
"autotuning ruled out" A/B never actually flipped autotune — both runs were
Expand Down
2 changes: 1 addition & 1 deletion jax_compile/trace_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@

LIBRARY_KEYS = (
"autofit",
"autoconf",
"autonerves",
"autoarray",
"autogalaxy",
"autolens",
Expand Down
2 changes: 1 addition & 1 deletion latent/imaging/effective_einstein_radius.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,9 @@
import jax
import jax.numpy as jnp
import numpy as np
from autoconf import conf
from autolens import fixtures
from autolens.analysis.latent import LATENT_FUNCTIONS
from autonerves import conf

if _os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion latent/imaging/magnification.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,9 @@
import jax
import jax.numpy as jnp
import numpy as np
from autoconf import conf
from autolens import fixtures
from autolens.analysis.latent import LATENT_FUNCTIONS
from autonerves import conf

if _os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion latent/imaging/total_lens_flux_mujy.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,9 +29,9 @@
import jax
import jax.numpy as jnp
import numpy as np
from autoconf import conf
from autolens import fixtures
from autolens.analysis.latent import LATENT_FUNCTIONS
from autonerves import conf

if _os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion latent/imaging/total_lensed_source_flux_mujy.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,9 @@
import jax
import jax.numpy as jnp
import numpy as np
from autoconf import conf
from autolens import fixtures
from autolens.analysis.latent import LATENT_FUNCTIONS
from autonerves import conf

if _os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion latent/imaging/total_source_flux_mujy.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,9 +32,9 @@
import jax
import jax.numpy as jnp
import numpy as np
from autoconf import conf
from autolens import fixtures
from autolens.analysis.latent import LATENT_FUNCTIONS
from autonerves import conf

if _os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports OK; exiting.")
Expand Down
4 changes: 2 additions & 2 deletions pipeline_resume/slam_resume.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,8 @@

# jax_wrapper must import before autofit/autolens (it sets the JAX env).
# isort: off
from autoconf import jax_wrapper # noqa: F401, E402
from autoconf.test_mode import test_mode_level, with_test_mode_segment # noqa: E402
from autonerves import jax_wrapper # noqa: F401, E402
from autonerves.test_mode import test_mode_level, with_test_mode_segment # noqa: E402

import autofit as af # noqa: E402
import autolens as al # noqa: E402
Expand Down
2 changes: 1 addition & 1 deletion quick_update/imaging.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
import autofit as af # noqa: E402
import autolens as al # noqa: E402
import numpy as np # noqa: E402
from autoconf import jax_wrapper # noqa: E402
from autonerves import jax_wrapper # noqa: E402

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

Expand Down
2 changes: 1 addition & 1 deletion quick_update/imaging_delaunay.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
import autofit as af # noqa: E402
import autolens as al # noqa: E402
import numpy as np # noqa: E402
from autoconf import jax_wrapper # noqa: E402
from autonerves import jax_wrapper # noqa: E402

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

Expand Down
2 changes: 1 addition & 1 deletion quick_update/interferometer.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
import autofit as af # noqa: E402
import autolens as al # noqa: E402
import numpy as np # noqa: E402
from autoconf import jax_wrapper # noqa: E402
from autonerves import jax_wrapper # noqa: E402

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

Expand Down
2 changes: 1 addition & 1 deletion quick_update/interferometer_delaunay.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
import autofit as af # noqa: E402
import autolens as al # noqa: E402
import numpy as np # noqa: E402
from autoconf import jax_wrapper # noqa: E402
from autonerves import jax_wrapper # noqa: E402

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

Expand Down
4 changes: 2 additions & 2 deletions quick_update/point_source.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@
import autofit as af # noqa: E402
import autolens as al # noqa: E402
import numpy as np # noqa: E402
from autoconf import jax_wrapper # noqa: E402
from autoconf.dictable import from_dict # noqa: E402
from autonerves import jax_wrapper # noqa: E402
from autonerves.dictable import from_dict # noqa: E402

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

Expand Down
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:206: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs/autolens_profiling is compatible with the installed library version (2026.7.6.649): no `version.minimum_library_version` or `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:206: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs/autolens_profiling is compatible with the installed library version (2026.7.6.649): no `version.minimum_library_version` or `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

You can also set the environment variable PYAUTO_SKIP_WORKSPACE_VERSION_CHECK=1 to disable temporarily.
warnings.warn(_missing_version_warning(root, library_version))
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:206: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs/autolens_profiling is compatible with the installed library version (2026.7.6.649): no `version.minimum_library_version` or `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:206: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs/autolens_profiling is compatible with the installed library version (2026.7.6.649): no `version.minimum_library_version` or `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
2026-06-01 11:48:55,318 - jax_plugins.xla_cuda12 - WARNING - cuda_plugin_extension is not found.
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

You can also set the environment variable PYAUTO_SKIP_WORKSPACE_VERSION_CHECK=1 to disable temporarily.
warnings.warn(_missing_version_warning(root, library_version))
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
2026-06-01 11:49:01,448 - jax_plugins.xla_cuda12 - WARNING - cuda_plugin_extension is not found.
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

You can also set the environment variable PYAUTO_SKIP_WORKSPACE_VERSION_CHECK=1 to disable temporarily.
warnings.warn(_missing_version_warning(root, library_version))
/home/jammy/Code/PyAutoLabs/PyAutoConf/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.
/home/jammy/Code/PyAutoLabs/PyAutoNerves/autoconf/workspace.py:173: UserWarning: Cannot verify the workspace at /home/jammy/Code/PyAutoLabs-wt/sparse-vs-dense-profile/autolens_profiling matches the installed library version (2026.5.29.4): no `version.workspace_version` key in config/general.yaml and no version.txt at the workspace root.

If you cloned the workspace from `main` rather than a release tag, set `version.workspace_version_check: False` in config/general.yaml to silence this warning. The `main` branch updates more frequently than library releases, so version mismatches are expected and not actionable for `main`-branch users.

Expand Down
4 changes: 2 additions & 2 deletions searches/sweep.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@

_REPO_ROOT = Path(__file__).resolve().parents[1] # autolens_profiling/
_DEFAULT_OUTPUT_ROOT = _REPO_ROOT / "results" / "searches"
# PyAutoFit's autoconf ``output_path`` defaults to ``<cwd>/output``. The
# PyAutoFit's autonerves ``output_path`` defaults to ``<cwd>/output``. The
# searches package writes search state under ``<output_path>/searches/...``
# via the ``path_prefix`` set in ``_samplers.build_nautilus``. Wiping this
# subtree before a (cell, config) run is what gives honest timing — see the
Expand Down Expand Up @@ -187,7 +187,7 @@ def _parse_args() -> argparse.Namespace:
help=(
f"Where PyAutoFit writes its own per-search output (samples.csv, "
f"search.summary, visualization). Default: "
f"{_DEFAULT_SEARCH_OUTPUT_ROOT}. Must match the autoconf "
f"{_DEFAULT_SEARCH_OUTPUT_ROOT}. Must match the autonerves "
f"``output_path`` + ``searches/`` prefix used by the leaf scripts."
),
)
Expand Down
2 changes: 1 addition & 1 deletion simulators/cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
import jax.numpy as jnp
import matplotlib
import numpy as np
from autoconf import jax_wrapper # noqa: F401 — must be first
from autonerves import jax_wrapper # noqa: F401 — must be first

if _smoke_os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports + module setup OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion simulators/group.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
import jax.numpy as jnp
import matplotlib
import numpy as np
from autoconf import jax_wrapper # noqa: F401 — must be first
from autonerves import jax_wrapper # noqa: F401 — must be first

if _smoke_os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports + module setup OK; exiting.")
Expand Down
2 changes: 1 addition & 1 deletion simulators/imaging.py
Original file line number Diff line number Diff line change
Expand Up @@ -340,7 +340,7 @@ def _image_fn(grid_array):
import os
import sys

from autoconf import jax_wrapper # noqa: F401
from autonerves import jax_wrapper # noqa: F401

if os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
print(f"[smoke] {__file__}: imports + module setup OK; exiting.")
Expand Down
6 changes: 3 additions & 3 deletions simulators/interferometer.py
Original file line number Diff line number Diff line change
Expand Up @@ -357,15 +357,15 @@ def _image_fn(grid_array):


if __name__ == "__main__":
# autoconf.jax_wrapper must be imported first (sets up env before jax
# autonerves.jax_wrapper must be imported first (sets up env before jax
# actually loads). Doing it here rather than at module level keeps the
# module import side-effect-free for likelihood scripts that pull
# ``INSTRUMENTS`` without wanting to trigger the autoconf shim.
# ``INSTRUMENTS`` without wanting to trigger the autonerves shim.
import argparse
import os
import sys

from autoconf import jax_wrapper # noqa: F401
from autonerves import jax_wrapper # noqa: F401

# Smoke gate — only meaningful when running as a script, not on import.
if os.environ.get("AUTOLENS_PROFILING_SMOKE") == "1":
Expand Down
Loading
Loading