What's Changed
- Adds edge to plot_vars if input is used in value while var has distribution by @wiep in #231
- Allow transformation of weak vars by @jobrachem in #235
- Update optim_flat by @jobrachem in #239
- Implement Var.plot_vars and Model.plot_vars by @jobrachem in #230
Maintenance Changes
- Update distreg.py by @jobrachem in #225
- Weaken jax version pin to <0.5.0, closes #214 by @jobrachem in #232
- unpin jax by @wiep in #238
- Fix test warnings by @jobrachem in #233
Full Changelog: v0.3.1...v0.3.2