Deterministics also take a ton of RAM, and it doesn’t show up until after sampling when using JAX (they are computed in the “transforming” step). I had a model that would crash the kernel after an hour of sampling because I was saving a lot of intermediate computations.