A declarative framework for probabalistic model specification with jax.
trellis aims to provide a natural api for declarative model specification.
By separating the concerns of model definition, state information, and parameter values, trellis enables transparent and flexible model training / conditioning.
- Clone Repo
git clone https://github.com/JWSchaefer/trellis.git- Play
uv run examples/fixed_prior.py