Jax to Main!#1155

Merged

Jammy2211 merged 95 commits intomainfrom

feature/jax_wrapper

Oct 6, 2025

Commits on May 30, 2025

Merge branch 'feature/github_actions'

Jammy2211
authored and
Jammy2211
committed
Merge branch 'main' of github.com:rhayes777/PyAutoFit

Jammy2211
authored and
Jammy2211
committed
Merge branch 'main' of github.com:rhayes777/PyAutoFit

Jammy2211
authored and
Jammy2211
committed

Commits on May 31, 2025

git ch0;10;1cMerge branch 'main' of github.com:rhayes777/PyAutoFit

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 19, 2025

switchign use jax to config enables tests to pass

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 20, 2025

simplify use jax import and make jax requirmeent

Jammy2211
authored and
Jammy2211
committed
disable warning when jax is used

Jammy2211
authored and
Jammy2211
committed
simplify factor jax use

Jammy2211
authored and
Jammy2211
committed
simplify jacobians

Jammy2211
authored and
Jammy2211
committed
move more jax imports

Jammy2211
authored and
Jammy2211
committed
fix pool and erfinv imports for cleaner JAX support

Jammy2211
authored and
Jammy2211
committed
all unit tests pass due to JAX fixes

Jammy2211
authored and
Jammy2211
committed
docstring of normal message complete

Jammy2211
authored and
Jammy2211
committed
beta.py docstringed

Jammy2211
authored and
Jammy2211
committed
GaussianPrior updated

Jammy2211
authored and
Jammy2211
committed
removed lower_limit and upper_limit from GaussianPrior

Jammy2211
authored and
Jammy2211
committed
remove lower_limit from all messages

Jammy2211
authored and
Jammy2211
committed
remove limits from test_limits

Jammy2211
authored and
Jammy2211
committed
remove all use of PriorLimitException

Jammy2211
authored and
Jammy2211
committed
remove assert_within_limits

Jammy2211
authored and
Jammy2211
committed
remove ignore_prior_limits

Jammy2211
authored and
Jammy2211
committed
limits removed rom mapping

Jammy2211
authored and
Jammy2211
committed
cleaning up some tests

Jammy2211
authored and
Jammy2211
committed
more removal of stuff trying to ix unit tests

Jammy2211
authored and
Jammy2211
committed
remove limits from more priors

Jammy2211
authored and
Jammy2211
committed
identifier tests pass

Jammy2211
authored and
Jammy2211
committed
aggregator tests pass

Jammy2211
authored and
Jammy2211
committed
remove lower limits of gaussians from more unit test sertup

Jammy2211
authored and
Jammy2211
committed
reinstate __database_args__ in UniformPrior and LogUniformPrior

Jammy2211
authored and
Jammy2211
committed
remove tests on gaussiaPrior limits being copied

Jammy2211
authored and
Jammy2211
committed
fix random prior test

Jammy2211
authored and
Jammy2211
committed
add dict to UniformPrior and LogUniformPRior fixint tests lost via inherritance

Jammy2211
authored and
Jammy2211
committed
all prior tests pass

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 23, 2025

fix non linear test regression by putting assertion skip back in

Jammy2211
authored and
Jammy2211
committed
readd limits to priors for prior passing

Jammy2211
authored and
Jammy2211
committed
fix mapper model test regression

Jammy2211
authored and
Jammy2211
committed
pytest test_autofit/mapper/model/test_model_mapper.py

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/mapper/functionality/test_take_attributes.py

Jammy2211
authored and
Jammy2211
committed
test_autofit/jax/test_pytrees.py

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/graphical/test_unification.py

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/graphical/stochastic/test_regression.py

Jammy2211
authored and
Jammy2211
committed
test_autofit/graphical/regression/test_identifier.py

Jammy2211
authored and
Jammy2211
committed
fixtest_autofit/graphical/global/test_hierarchical.py

Jammy2211
authored and
Jammy2211
committed
test_autofit/graphical/gaussian/test_other_priors.py

Jammy2211
authored and
Jammy2211
committed
test_autofit/database/identifier/test_identifiers.py

Jammy2211
authored and
Jammy2211
committed
fix pytest test_autofit/aggregator/test_reference.py

Jammy2211
authored and
Jammy2211
committed
TruncatedNormal setup

Jammy2211
authored and
Jammy2211
committed
value for on truncated gaussian

Jammy2211
authored and
Jammy2211
committed
TruncatedGaussianPrior full implemented, removes assert within limits for JAX

Jammy2211
authored and
Jammy2211
committed
gaussian liimts -> limits

Jammy2211
authored and
Jammy2211
committed
git checkout fea0;10;1cMerge branch 'main' of github.com:rhayes777/PyAutoFit

Jammy2211
authored and
Jammy2211
committed
Merge branch 'main' into feature/truncated_gaussian

Jammy2211
authored and
Jammy2211
committed
integration tests pass

Jammy2211
authored and
Jammy2211
committed
model absolute -> model centred absolute

Jammy2211
authored and
Jammy2211
committed
model relative -> model centred relative

Jammy2211
authored and
Jammy2211
committed
model_bounded -> model_centred_max_lh_bounded

Jammy2211
authored and
Jammy2211
committed
default prior passing now just reuses original prior

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/non_linear/search/test_abstract_search.py

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/non_linear/result/test_result.py

Jammy2211
authored and
Jammy2211
committed
fix test_autofit/mapper/functionality/test_by_path.py

Jammy2211
authored and
Jammy2211
committed
fix

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 24, 2025

Merge pull request #1151 from rhayes777/feature/model_centered
Jammy2211
authored
clean up some imports

Jammy2211
authored and
Jammy2211
committed

Commits on Jun 25, 2025

reudce import time by going internal

Jammy2211
authored and
Jammy2211
committed

Commits on Jul 12, 2025

Merge pull request #1152 from rhayes777/feature/jax_speed_up_general
Jammy2211
authored

Commits on Jul 24, 2025

stuff added

Jammy2211
authored and
Jammy2211
committed

Commits on Jul 30, 2025

fix some unit tests and truncated gaussian log prior

Jammy2211
authored and
Jammy2211
committed
nautilus now uses batch with JAX

Jammy2211
authored and
Jammy2211
committed
logging

Jammy2211
authored and
Jammy2211
committed

Commits on Aug 1, 2025

spawn parallel

Jammy2211
authored and
Jammy2211
committed

Commits on Aug 7, 2025

haphazard

Jammy2211
authored and
Jammy2211
committed
PriorVectorize implemented

Jammy2211
authored and
Jammy2211
committed
added unit test of vectorized with UniformPrior

Jammy2211
authored and
Jammy2211
committed
added Gaussian, TruncatedGaussian, LogUniform to Vectorized

Jammy2211
authored and
Jammy2211
committed
documentation

Jammy2211
authored and
Jammy2211
committed

Commits on Aug 11, 2025

dask stuff seems to work

Jammy2211
authored and
Jammy2211
committed
change to actors futrues

Jammy2211
authored and
Jammy2211
committed

Commits on Aug 18, 2025

backup

Jammy2211
authored and
Jammy2211
committed

Commits on Sep 25, 2025

update sampling with jax

Jammy2211
authored and
Jammy2211
committed

Commits on Oct 1, 2025

update nautilus

Jammy2211
authored and
Jammy2211
committed

Commits on Oct 2, 2025

cast to float to prevent JAX arrays breaking output

Jammy2211
authored and
Jammy2211
committed
latent variables now compute in JAX

Jammy2211
authored and
Jammy2211
committed
added samples_drawn_randomly_via_pdf_from

Jammy2211
authored and
Jammy2211
committed

Commits on Oct 3, 2025

fix one latent variable test

Jammy2211
authored and
Jammy2211
committed
fix all latent tests

Jammy2211
authored and
Jammy2211
committed
implementation fully tested

Jammy2211
authored and
Jammy2211
committed
fix latent varaible thing

Jammy2211
authored and
Jammy2211
committed

Commits on Oct 6, 2025

Merge pull request #1154 from rhayes777/feature/jax_latent
Jammy2211
authored
fix text util

Jammy2211
authored and
Jammy2211
committed
fix test_truncated_gaussian_prior_fields

Jammy2211
authored and
Jammy2211
committed
fix beta message

Jammy2211
authored and
Jammy2211
committed
test_truncated_gaussian_prior

Jammy2211
authored and
Jammy2211
committed
fix weight

Jammy2211
authored and
Jammy2211
committed
all unit tests pass

Jammy2211
authored and
Jammy2211
committed
git push

Jammy2211
authored and
Jammy2211
committed