Add ray to the base image

[yhtang]

Installs Ray on the base image to make our containers deployable as Ray workers, so that users can distribute JAX computation to GPU nodes as Ray remote functions.

Replaces #750, which is on Chase's fork and is not able to run CI.

Closes #750.

[nouiz]

What is the extra container size?

[yhtang]

What is the extra container size?

It adds about 70 MB to the compressed size of the image.

Inspecting image: ghcr.io/nvidia/jax-toolbox-internal:11074498143-jax-amd64
amd64 manifest digest: sha256:baff41a793d6fc0312cf8ec8389d2c9ae1fe562e6c49b4219e3dcf7e297d8d1b
Compressed size of image ghcr.io/nvidia/jax-toolbox-internal:11074498143-jax-amd64: 6.49304e+09 bytes
----------------------------------------
Inspecting image: ghcr.io/nvidia/jax-toolbox-internal:11232636645-jax-amd64
amd64 manifest digest: sha256:80e0bdb2560fd1f9c73342b7d2cc1d50d6b30cb056ff0d7ea93a5ad79f920966
Compressed size of image ghcr.io/nvidia/jax-toolbox-internal:11232636645-jax-amd64: 6.42759e+09 bytes

[yhtang]

If no objection, I'll go ahead and merge this PR by the end of this week 😁