FROM reax13-base
RUN pip install --no-cache-dir 'ninja==1.13.2' 'nvidia-cuda-cccl==13.0.85'
RUN CAUSAL_CONV1D_FORCE_BUILD=TRUE MAX_JOBS=4 CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST=12.0 pip install --no-cache-dir --no-build-isolation --no-deps 'causal-conv1d==1.7.0'
RUN python -c 'import causal_conv1d, torch; print(causal_conv1d.__version__, torch.__version__)'

RUN pip install --no-cache-dir 'httpx==0.28.1'

COPY . /engine
ENV PYTHONPATH=/engine
WORKDIR /engine