pandas-bokeh
jax[cuda11_cudnn82]

--find-links https://storage.googleapis.com/jax-releases/jax_releases.html

fedjax
