docker-image-jax-torch-sphinx