Merge branch 'chase/ray_install' of https://github.com/chaserileyroberts/JAX-Toolbox into add-ray

Co-authored-by: Chase Riley Roberts <[email protected]>

Author: yhtang

.github/container/Dockerfile.base

@@ -127,6 +127,12 @@ git commit -m 'Adds JAX_TOOLBOX_VCS_EQUIVALENCY as a trigger to treat all github
 EOF
 RUN pip install --upgrade --no-cache-dir -e /opt/pip pip-tools && rm -rf ~/.cache/*
 
+###############################################################################
+## Install Ray
+###############################################################################
+RUN echo "ray==2.37.0" >> /opt/pip-tools.d/requirements-ray.in
+EXPOSE 6379 8265 10001-10004
+
 ###############################################################################
 ## Install TCPx
 ###############################################################################