Skip to content

Dev #40

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
bgctw merged 28 commits into from
Aug 19, 2026
Merged

Dev #40

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
288892b
improve penalty tutorial
bgctw May 12, 2026
12e4ce0
computing cost on test_data in callback
bgctw May 12, 2026
4e666bd
chaning n_site for train + test data
bgctw May 12, 2026
436a5ca
allow test_data to have more entries than the five used by HVI
bgctw May 13, 2026
a7676d3
utility functions
bgctw May 13, 2026
d65637a
docu
bgctw Jun 9, 2026
0348667
start to work on random effects for sites
bgctw Jun 9, 2026
903407f
random effects in cost functions
bgctw Jun 12, 2026
22b9329
start sampling random effect during prediction without known site
bgctw Jun 13, 2026
b226d54
random effects computer constructed in Solver
bgctw Jun 15, 2026
1c9a2b4
implement get_choleskyΣ_ranef(RandomEffect)
bgctw Jun 17, 2026
5011821
remove priors on mean theta
bgctw Jun 18, 2026
33b752e
factor out compute_entropyζ_from_σ from elbo
bgctw Jun 18, 2026
3d371cb
make predict_hvi return also logdetjac of transformations
bgctw Jun 18, 2026
60c64d1
take care of frac_cluster in sample_posterior
bgctw Jun 19, 2026
7568471
implement predict_hvi_and_compute_elbo_components
bgctw Jun 19, 2026
e7c8036
make logdensity function return a vector per site
bgctw Jun 22, 2026
f716f39
assemble sitespecific elbo components
bgctw Jun 23, 2026
5124277
test columwise logden_normal
bgctw Jun 23, 2026
03329e4
simplify neg_logden_indep_normal with multiplying by is_finite_obs
bgctw Jun 24, 2026
011bda4
implmenet WeightedDataLoader
bgctw Jun 24, 2026
4b35e72
rename i_sites to i_sites_train
bgctw Jun 25, 2026
4beb020
update tutorials
bgctw Jun 25, 2026
0e3a115
fix frac_cluster on i_sites_train[1] == 0, i.e. untrained site
bgctw Jun 25, 2026
a29c034
rename i_sites_train to itrain_sites
bgctw Jul 4, 2026
2586271
import CommonSolve solve
bgctw Aug 19, 2026
e8f4653
typos
bgctw Aug 19, 2026
d3614ce
add DocStrings for HybridPointSolver and HybridPosteriorSolver
bgctw Aug 19, 2026
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
1 change: 1 addition & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,3 +27,4 @@ docs/src/tutorials/intermediate/clustered_sites_probo_allsites.jld2
projects/clustered_sites/intermediate/*

build/*
intermediate/*
9 changes: 7 additions & 2 deletions Project.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ Distances = "b4f34e82-e78d-54a5-968a-f98e89d6e8f7"
DistributionFits = "45214091-1ed4-4409-9bcf-fdb48a05e921"
Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
Functors = "d9f16b24-f501-4c13-a1f2-28368ffc5196"
GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
IterTools = "c8e1da08-722c-5040-9ed9-7db0dc04731e"
Expand All @@ -29,7 +30,9 @@ NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
Optimization = "7f7a1694-90dd-40f0-9382-eb1efda571ba"
PDMats = "90014a1f-27ba-587c-ab20-58faa44d9150"
PreallocationTools = "d236fae5-4411-538c-8e31-a6e3d9e00b46"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
Expand All @@ -40,11 +43,10 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"

[weakdeps]
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
SimpleChains = "de6bee2f-e2f4-4ec7-b6ed-219cc6f6e9e5"
DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"


[extensions]
HybridVariationalInferenceCUDAExt = "CUDA"
Expand All @@ -69,6 +71,7 @@ DistributionFits = "0.3.9"
Distributions = "0.25.117"
FillArrays = "1.13.0"
Flux = "0.14, 0.15, 0.16"
ForwardDiff = "1"
Functors = "0.4, 0.5"
GPUArraysCore = "0.1, 0.2"
IterTools = "1.10.0"
Expand All @@ -83,7 +86,9 @@ NaNMath = "1.1.3"
Optimisers = "0.4.6"
Optimization = "5.3"
PDMats = "0.11.37"
PreallocationTools = "1.2.1"
Random = "1.10.0"
RecursiveArrayTools = "3.54.0"
SimpleChains = "0.4.7"
StableRNGs = "1.0.2"
StaticArrays = "1.9.13"
Expand Down
Loading
Loading