FROM mambaorg/micromamba:2.0.5
RUN micromamba install -y -n base -c conda-forge --strict-channel-priority "conda-forge::jax-md=0.2.29" \
 && micromamba clean -afy \
 && find /opt/conda -name '*.a' -delete
COPY --chmod=755 run_lj.py /opt/driver/run_lj.py
ENV PATH=/opt/conda/bin:$PATH LANG=C.UTF-8 JAX_PLATFORMS=cpu JAX_ENABLE_X64=1 PYTHONWARNINGS=ignore
USER 65534:65534
WORKDIR /work/out
ENTRYPOINT []
