scipy==1.11.2
git+https://github.com/ericsuh/dirichlet.git
jax[cpu]==0.4.14
plotly==5.18.0
tensorflow-probability==0.23.0
flax==0.7.2
safetensors==0.4.2