Feature/jax latent#1154

Merged

Jammy2211 merged 6 commits intofeature/jax_wrapperfrom

feature/jax_latent

Oct 6, 2025

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