Speed up gibbs sampler using jax library#336

ljgray wants to merge 1 commit intomasterradiocosmology/draco:masterfrom

ljg/jax-pspecradiocosmology/draco:ljg/jax-pspecCopy head branch name to clipboard