Merge pull request #380 from Jammy2211/feature/jax_w_tilde_preload

Feature/jax w tilde preload

Author: Jammy2211

autolens/__init__.py

@@ -1,6 +1,7 @@
 from autoconf import jax_wrapper
 from autoconf.dictable import from_dict, from_json, output_to_json, to_dict
 from autoarray import preprocess
+from autoarray.dataset.interferometer.w_tilde import load_curvature_preload_if_compatible
 from autoarray.dataset.imaging.w_tilde import WTildeImaging
 from autoarray.dataset.imaging.dataset import Imaging
 from autoarray.dataset.interferometer.dataset import (