jax

aaff39a7 - Change Linux docker wheel build to build separate wheels for each minor Python version.

Change Linux docker wheel build to build separate wheels for each minor Python version. * Use pyenv to build specific Python releases against which we build wheels. Otherwise we are limited to only those Python releases present in the OS repository, which only correspond to certain major versions of Python. * Build with docker instead of nvidia-docker. We need a CUDA image to build JAX, but we don't require a GPU or nvidia-docker unless we want to actually run things. * Various other cleanups.